1 /* vim:expandtab:shiftwidth=2:tabstop=2:smarttab:
3 * Data Differential YATL (i.e. libtest) library
5 * Copyright (C) 2012 Data Differential, http://datadifferential.com/
7 * Redistribution and use in source and binary forms, with or without
8 * modification, are permitted provided that the following conditions are
11 * * Redistributions of source code must retain the above copyright
12 * notice, this list of conditions and the following disclaimer.
14 * * Redistributions in binary form must reproduce the above
15 * copyright notice, this list of conditions and the following disclaimer
16 * in the documentation and/or other materials provided with the
19 * * The names of its contributors may not be used to endorse or
20 * promote products derived from this software without specific prior
23 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
24 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
25 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
26 * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
27 * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
28 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
29 * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
30 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
31 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
32 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
33 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
38 #include <libtest/common.h>
40 #include <libtest/gearmand.h>
42 #include "util/instance.hpp"
43 #include "util/operation.hpp"
45 using namespace datadifferential
;
46 using namespace libtest
;
56 #include <sys/types.h>
60 #include <libgearman/gearman.h>
62 #ifndef __INTEL_COMPILER
63 #pragma GCC diagnostic ignored "-Wold-style-cast"
66 using namespace libtest
;
68 class Gearmand
: public libtest::Server
72 Gearmand(const std::string
& host_arg
, in_port_t port_arg
) :
73 libtest::Server(host_arg
, port_arg
, GEARMAND_BINARY
, true)
80 gearman_client_st
*client
= gearman_client_create(NULL
);
83 Error
<< "Could not allocate memory for gearman_client_create()";
86 gearman_client_set_timeout(client
, 3000);
88 if (gearman_success(gearman_client_add_server(client
, hostname().c_str(), port())))
90 gearman_return_t rc
= gearman_client_echo(client
, test_literal_param("This is my echo test"));
92 if (gearman_success(rc
))
94 gearman_client_free(client
);
98 Error
<< hostname().c_str() << ":" << port() << " was " << gearman_strerror(rc
) << " extended: " << gearman_client_error(client
);
103 Error
<< "gearman_client_add_server() " << gearman_client_error(client
);
106 gearman_client_free(client
);
116 void log_file_option(Application
& app
, const std::string
& arg
)
118 if (arg
.empty() == false)
120 std::string
buffer("--log-file=");
122 app
.add_option("--verbose=DEBUG");
123 app
.add_option(buffer
);
127 bool has_log_file_option() const
137 bool has_syslog() const
139 return false; // --syslog.errmsg-enable
142 bool has_port_option() const
147 bool build(size_t argc
, const char *argv
[]);
150 bool Gearmand::build(size_t argc
, const char *argv
[])
152 if (getuid() == 0 or geteuid() == 0)
154 add_option("-u", "root");
157 add_option("--listen=localhost");
159 for (size_t x
= 0 ; x
< argc
; x
++)
169 libtest::Server
*build_gearmand(const char *hostname
, in_port_t try_port
)
171 return new Gearmand(hostname
, try_port
);