X-Git-Url: https://git.m6w6.name/?a=blobdiff_plain;f=libtest%2Fserver_container.cc;h=25da72c6ff2e410f420a119959f6c26c60e6f18c;hb=24ce539ecb81145bad8e57e6f9c288adfbd71fac;hp=6973937bc6f8749de1ef7da815c027336fd1ba0f;hpb=facfe08e21d2c3135c508ff098ff603ddaacef3e;p=awesomized%2Flibmemcached diff --git a/libtest/server_container.cc b/libtest/server_container.cc index 6973937b..25da72c6 100644 --- a/libtest/server_container.cc +++ b/libtest/server_container.cc @@ -34,10 +34,10 @@ * */ -#include -#include +#include "libtest/yatlcon.h" + +#include "libtest/common.h" -#include #include #include #include @@ -75,18 +75,18 @@ void server_startup_st::push_server(Server *arg) } else { - char port_str[NI_MAXSERV]; - snprintf(port_str, sizeof(port_str), "%u", int(arg->port())); + libtest::vchar_t port_str; + port_str.resize(NI_MAXSERV); + snprintf(&port_str[0], port_str.size(), "%u", int(arg->port())); server_config_string+= "--server="; server_config_string+= arg->hostname(); server_config_string+= ":"; - server_config_string+= port_str; + server_config_string+= &port_str[0]; server_config_string+= " "; } server_list+= server_config_string; - } Server* server_startup_st::pop_server() @@ -116,10 +116,7 @@ bool server_startup_st::shutdown(uint32_t host_to_shutdown) void server_startup_st::clear() { - for (std::vector::iterator iter= servers.begin(); iter != servers.end(); ++iter) - { - delete *iter; - } + std::for_each(servers.begin(), servers.end(), DeleteFromVector()); servers.clear(); } @@ -142,7 +139,7 @@ bool server_startup_st::shutdown() bool success= true; for (std::vector::iterator iter= servers.begin(); iter != servers.end(); ++iter) { - if ((*iter) and (*iter)->has_pid() and (*iter)->kill() == false) + if ((*iter)->has_pid() and (*iter)->kill() == false) { Error << "Unable to kill:" << *(*iter); success= false; @@ -156,7 +153,12 @@ void server_startup_st::restart() { for (std::vector::iterator iter= servers.begin(); iter != servers.end(); ++iter) { - (*iter)->start(); + Server *server = *iter; + + if (server->check()) { + server->kill(); + } + server->start(); } } @@ -165,7 +167,6 @@ server_startup_st::server_startup_st() : _magic(MAGIC_MEMORY), _socket(false), _sasl(false), - _count(0), udp(0), _servers_to_run(5) { } @@ -180,198 +181,128 @@ bool server_startup_st::validate() return _magic == MAGIC_MEMORY; } -bool server_startup(server_startup_st& construct, const std::string& server_type, in_port_t try_port, int argc, const char *argv[], const bool opt_startup_message) +bool server_startup(server_startup_st& construct, const std::string& server_type, in_port_t try_port, const char *argv[]) { - construct.start_server(server_type, try_port, argc, argv, opt_startup_message); + return construct.start_server(server_type, try_port, argv); } -bool server_startup_st::start_server(const std::string& server_type, in_port_t try_port, int argc, const char *argv[], const bool opt_startup_message) +libtest::Server* server_startup_st::create(const std::string& server_type, in_port_t try_port, const bool is_socket) { - if (try_port <= 0) - { - throw libtest::fatal(LIBYATL_DEFAULT_PARAM, "was passed the invalid port number %d", int(try_port)); - } - libtest::Server *server= NULL; - try { - if (0) - { } - else if (server_type.compare("gearmand") == 0) + + if (is_socket == false) + { + if (try_port <= 0) { - if (GEARMAND_BINARY) - { - if (HAVE_LIBGEARMAN) - { - server= build_gearmand("localhost", try_port); - } - } + throw libtest::fatal(LIBYATL_DEFAULT_PARAM, "was passed the invalid port number %d", int(try_port)); } - else if (server_type.compare("drizzled") == 0) + } + + if (is_socket) + { + if (server_type.compare("memcached") == 0) { - if (DRIZZLED_BINARY) - { - if (HAVE_LIBDRIZZLE) - { - server= build_drizzled("localhost", try_port); - } - } + server= build_memcached_socket("localhost", try_port); } - else if (server_type.compare("blobslap_worker") == 0) + else { - if (GEARMAND_BINARY) - { - if (GEARMAND_BLOBSLAP_WORKER) - { - if (HAVE_LIBGEARMAN) - { - server= build_blobslap_worker(try_port); - } - } - } + Error << "Socket is not support for server: " << server_type; + return NULL; } - else if (server_type.compare("memcached-sasl") == 0) + } + else if (server_type.compare("gearmand") == 0) + { + server= build_gearmand("localhost", try_port); + } + else if (server_type.compare("hostile-gearmand") == 0) + { + server= build_gearmand("localhost", try_port, "gearmand/hostile_gearmand"); + } + else if (server_type.compare("drizzled") == 0) + { + if (has_drizzled()) { - if (MEMCACHED_SASL_BINARY) + if (has_libdrizzle()) { - if (HAVE_LIBMEMCACHED) - { - server= build_memcached_sasl("localhost", try_port, username(), password()); - } + server= build_drizzled("localhost", try_port); } } - else if (server_type.compare("memcached") == 0) + } + else if (server_type.compare("blobslap_worker") == 0) + { + if (has_gearmand()) { - if (HAVE_MEMCACHED_BINARY) +#ifdef GEARMAND_BLOBSLAP_WORKER + if (GEARMAND_BLOBSLAP_WORKER) { - if (HAVE_LIBMEMCACHED) + if (HAVE_LIBGEARMAN) { - server= build_memcached("localhost", try_port); + server= build_blobslap_worker(try_port); } } +#endif // GEARMAND_BLOBSLAP_WORKER } - else if (server_type.compare("memcached-light") == 0) + } + else if (server_type.compare("memcached") == 0) + { + if (has_memcached()) { - if (MEMCACHED_LIGHT_BINARY) - { - if (HAVE_LIBMEMCACHED) - { - server= build_memcached_light("localhost", try_port); - } - } + server= build_memcached("localhost", try_port); } + } - if (server == NULL) - { - throw libtest::fatal(LIBYATL_DEFAULT_PARAM, "Launching of an unknown server was attempted: %s", server_type.c_str()); - } + return server; +} - /* - We will now cycle the server we have created. - */ - if (server->cycle() == false) - { - Error << "Could not start up server " << *server; - delete server; - return false; - } +class ServerPtr { +public: + ServerPtr(libtest::Server* server_): + _server(server_) + { } - server->build(argc, argv); + ~ServerPtr() + { + delete _server; + } - if (false) - { - Out << "Pausing for startup, hit return when ready."; - std::string gdb_command= server->base_command(); - std::string options; -#if 0 - Out << "run " << server->args(options); -#endif - getchar(); - } - else if (server->start() == false) - { - delete server; - return false; - } - else - { - if (opt_startup_message) - { - Outn(); - Out << "STARTING SERVER(pid:" << server->pid() << "): " << server->running(); - Outn(); - } - } + void reset() + { + delete _server; + _server= NULL; } - catch (...) + + libtest::Server* release(libtest::Server* server_= NULL) { - delete server; - throw; + libtest::Server* tmp= _server; + _server= server_; + return tmp; } - push_server(server); + libtest::Server* operator->() const + { + return _server; + } - return true; -} + libtest::Server* operator&() const + { + return _server; + } -bool server_startup_st::start_socket_server(const std::string& server_type, const in_port_t try_port, int argc, - const char *argv[], - const bool opt_startup_message) -{ - (void)try_port; - Outn(); +private: + libtest::Server* _server; +}; - Server *server= NULL; +bool server_startup_st::_start_server(const bool is_socket, + const std::string& server_type, + in_port_t try_port, + const char *argv[]) +{ try { - if (0) - { } - else if (server_type.compare("gearmand") == 0) - { - Error << "Socket files are not supported for gearmand yet"; - } - else if (server_type.compare("memcached-sasl") == 0) - { - if (MEMCACHED_SASL_BINARY) - { - if (HAVE_LIBMEMCACHED) - { - server= build_memcached_sasl_socket("localhost", try_port, username(), password()); - } - else - { - Error << "Libmemcached was not found"; - } - } - else - { - Error << "No memcached binary is available"; - } - } - else if (server_type.compare("memcached") == 0) - { - if (MEMCACHED_BINARY) - { - if (HAVE_LIBMEMCACHED) - { - server= build_memcached_socket("localhost", try_port); - } - else - { - Error << "Libmemcached was not found"; - } - } - else - { - Error << "No memcached binary is available"; - } - } - else - { - Error << "Failed to start " << server_type << ", no support was found to be compiled in for it."; - } + ServerPtr server(create(server_type, try_port, is_socket)); - if (server == NULL) + if (&server == NULL) { - Error << "Failure occured while creating server: " << server_type; + Error << "Could not allocate server: " << server_type; return false; } @@ -380,52 +311,81 @@ bool server_startup_st::start_socket_server(const std::string& server_type, cons */ if (server->cycle() == false) { - Error << "Could not start up server " << *server; - delete server; + Error << "Could not start up server " << &server; return false; } - server->build(argc, argv); + server->init(argv); +#if 0 if (false) { Out << "Pausing for startup, hit return when ready."; std::string gdb_command= server->base_command(); - std::string options; -#if 0 - Out << "run " << server->args(options); -#endif getchar(); } - else if (server->start() == false) - { - Error << "Failed to start " << *server; - delete server; - return false; - } else - { - if (opt_startup_message) +#endif + + if (server->start() == false) { - Outn(); - Out << "STARTING SERVER(pid:" << server->pid() << "): " << server->running(); - Outn(); + return false; + } + else + { + { +#ifdef DEBUG + if (DEBUG) + { + Outn(); + Out << "STARTING SERVER(pid:" << server->pid() << "): " << server->running(); + } +#endif + } } + + push_server(server.release()); + + if (is_socket and &server) + { + set_default_socket(server->socket().c_str()); + } + } + catch (const libtest::disconnected& err) + { + if (fatal::is_disabled() == false and try_port != LIBTEST_FAIL_PORT) + { + stream::cerr(err.file(), err.line(), err.func()) << err.what(); + return false; } } + catch (const libtest::__test_result& err) + { + stream::cerr(err.file(), err.line(), err.func()) << err.what(); + return false; + } + catch (const std::exception& err) + { + Error << err.what(); + return false; + } catch (...) { - delete server; - throw; + Error << "error occured while creating server: " << server_type; + return false; } - push_server(server); - - set_default_socket(server->socket().c_str()); + return true; +} - Outn(); +bool server_startup_st::start_server(const std::string& server_type, in_port_t try_port, const char *argv[]) +{ + return _start_server(false, server_type, try_port, argv); +} - return true; +bool server_startup_st::start_socket_server(const std::string& server_type, const in_port_t try_port, const char *argv[]) +{ + return _start_server(true, server_type, try_port, argv); } std::string server_startup_st::option_string() const