X-Git-Url: https://git.m6w6.name/?a=blobdiff_plain;f=libtest%2Fserver_container.cc;h=8203de9d5b46854268d7fd5f8695fa8db8db24bc;hb=65e75d2db2d217b37c94c36d48607cb8425c8f86;hp=f747dc229b4e47d620f1b9390a214a261c18f4db;hpb=58c279d58e2a44562f729e93e301dfedf42f530b;p=awesomized%2Flibmemcached diff --git a/libtest/server_container.cc b/libtest/server_container.cc index f747dc22..8203de9d 100644 --- a/libtest/server_container.cc +++ b/libtest/server_container.cc @@ -19,6 +19,7 @@ * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA */ +#include #include #include @@ -75,67 +76,94 @@ Server* server_startup_st::pop_server() return tmp; } -void server_startup_st::shutdown(bool remove) +bool server_startup_st::shutdown(uint32_t number_of_host) { - if (remove) + if (servers.size() > number_of_host) { - for (std::vector::iterator iter= servers.begin(); iter != servers.end(); iter++) + Server* tmp= servers[number_of_host]; + + if (tmp and tmp->has_pid() and tmp->kill() == false) + { } + else { - delete *iter; + return true; } - servers.clear(); } - else + + return false; +} + +void server_startup_st::shutdown_and_remove() +{ + for (std::vector::iterator iter= servers.begin(); iter != servers.end(); iter++) + { + delete *iter; + } + servers.clear(); +} + +bool server_startup_st::check() const +{ + bool success= true; + for (std::vector::const_iterator iter= servers.begin(); iter != servers.end(); iter++) { - for (std::vector::iterator iter= servers.begin(); iter != servers.end(); iter++) + if ((*iter)->check() == false) { - if ((*iter)->has_pid() and not (*iter)->kill((*iter)->pid())) - { - Error << "Unable to kill:" << *(*iter); - } + success= false; } } + + return success; } -server_startup_st::~server_startup_st() +bool server_startup_st::shutdown() { - shutdown(true); + bool success= true; + for (std::vector::iterator iter= servers.begin(); iter != servers.end(); iter++) + { + if ((*iter)->has_pid() and (*iter)->kill() == false) + { + Error << "Unable to kill:" << *(*iter); + success= false; + } + } + + return success; } -bool server_startup_st::is_debug() const +void server_startup_st::restart() { - return bool(getenv("LIBTEST_MANUAL_GDB")); + for (std::vector::iterator iter= servers.begin(); iter != servers.end(); iter++) + { + (*iter)->start(); + } } -bool server_startup_st::is_valgrind() const +#define MAGIC_MEMORY 123575 +server_startup_st::server_startup_st() : + _magic(MAGIC_MEMORY), + _socket(false), + _sasl(false), + _count(0), + udp(0), + _servers_to_run(5) +{ } + +server_startup_st::~server_startup_st() { - return bool(getenv("LIBTEST_MANUAL_VALGRIND")); + shutdown_and_remove(); } -bool server_startup_st::is_helgrind() const +bool server_startup_st::validate() { - return bool(getenv("LIBTEST_MANUAL_HELGRIND")); + 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[]) +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) { - Outn(); - (void)try_port; - - // Look to see if we are being provided ports to use + if (try_port <= 0) { - char variable_buffer[1024]; - snprintf(variable_buffer, sizeof(variable_buffer), "LIBTEST_PORT_%lu", (unsigned long)construct.count()); - - char *var; - if ((var= getenv(variable_buffer))) - { - in_port_t tmp= in_port_t(atoi(var)); - - if (tmp > 0) - try_port= tmp; - } + throw libtest::fatal(LIBYATL_DEFAULT_PARAM, "was passed the invalid port number %d", int(try_port)); } libtest::Server *server= NULL; @@ -149,32 +177,19 @@ bool server_startup(server_startup_st& construct, const std::string& server_type { server= build_gearmand("localhost", try_port); } - else - { - Error << "Libgearman was not found"; - } - } - else - { - Error << "No gearmand binary is available"; } } else if (server_type.compare("blobslap_worker") == 0) { - if (GEARMAND_BINARY and GEARMAND_BLOBSLAP_WORKER) + if (GEARMAND_BINARY) { - if (HAVE_LIBGEARMAN) + if (GEARMAND_BLOBSLAP_WORKER) { - server= build_blobslap_worker(try_port); + if (HAVE_LIBGEARMAN) + { + server= build_blobslap_worker(try_port); + } } - else - { - Error << "Libgearman was not found"; - } - } - else - { - Error << "No gearmand binary is available"; } } else if (server_type.compare("memcached-sasl") == 0) @@ -185,14 +200,6 @@ bool server_startup(server_startup_st& construct, const std::string& server_type { server= build_memcached_sasl("localhost", try_port, construct.username(), construct.password()); } - else - { - Error << "Libmemcached was not found"; - } - } - else - { - Error << "No memcached binary that was compiled with sasl is available"; } } else if (server_type.compare("memcached") == 0) @@ -203,31 +210,28 @@ bool server_startup(server_startup_st& construct, const std::string& server_type { server= build_memcached("localhost", try_port); } - else - { - Error << "Libmemcached was not found"; - } - } - else - { - Error << "No memcached binary is available"; } } - else + else if (server_type.compare("memcached-light") == 0) { - Error << "Failed to start " << server_type << ", no support was found to be compiled in for it."; + if (MEMCACHED_LIGHT_BINARY) + { + if (HAVE_LIBMEMCACHED) + { + server= build_memcached_light("localhost", try_port); + } + } } if (server == NULL) { - Error << "Failure occured while creating server: " << server_type; - return false; + fatal_message("Launching of an unknown server was attempted"); } /* We will now cycle the server we have created. */ - if (not server->cycle()) + if (server->cycle() == false) { Error << "Could not start up server " << *server; delete server; @@ -236,35 +240,33 @@ bool server_startup(server_startup_st& construct, const std::string& server_type server->build(argc, argv); - if (construct.is_debug()) + if (gdb_is_caller() and 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 (not server->start()) + else if (server->start() == false) { - Error << "Failed to start " << *server; delete server; return false; } else { - Out << "STARTING SERVER(pid:" << server->pid() << "): " << server->running(); + if (opt_startup_message) + { + Outn(); + Out << "STARTING SERVER(pid:" << server->pid() << "): " << server->running(); + Outn(); + } } construct.push_server(server); - if (default_port() == 0) - { - assert(server->has_port()); - set_default_port(server->port()); - } - - Outn(); - return true; } @@ -330,7 +332,7 @@ bool server_startup_st::start_socket_server(const std::string& server_type, cons /* We will now cycle the server we have created. */ - if (not server->cycle()) + if (server->cycle() == false) { Error << "Could not start up server " << *server; delete server; @@ -339,15 +341,17 @@ bool server_startup_st::start_socket_server(const std::string& server_type, cons server->build(argc, argv); - if (is_debug()) + if (gdb_is_caller() and 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 (not server->start()) + else if (server->start() == false) { Error << "Failed to start " << *server; delete server;