X-Git-Url: https://git.m6w6.name/?a=blobdiff_plain;f=libtest%2Fserver.h;h=b1ca30aa713f04e2d26eacd3fb955910dfeec541;hb=1e8bc3cb173d8b4e206363de7fdabeddb32d0f53;hp=454f69ca81e1266824c6cd7d5d3e58a0f36380b2;hpb=4fe35ea1372eda035618c6e961f421187d3ae837;p=m6w6%2Flibmemcached diff --git a/libtest/server.h b/libtest/server.h index 454f69ca..b1ca30aa 100644 --- a/libtest/server.h +++ b/libtest/server.h @@ -1,42 +1,243 @@ -/* - * Copyright (C) 2011 Data Differential, http://datadifferential.com/ - * Copyright (C) 2006-2009 Brian Aker - * All rights reserved. +/* vim:expandtab:shiftwidth=2:tabstop=2:smarttab: + * + * libtest * - * Use and distribution licensed under the BSD license. See - * the COPYING file in the parent directory for full text. + * Copyright (C) 2011 Data Differential, http://datadifferential.com/ + * + * This library is free software; you can redistribute it and/or + * modify it under the terms of the GNU Lesser General Public + * License as published by the Free Software Foundation; either + * version 3 of the License, or (at your option) any later version. + * + * This library is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU + * Lesser General Public License for more details. + * + * You should have received a copy of the GNU Lesser General Public + * License along with this library; if not, write to the Free Software + * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA */ #pragma once +#include + +#include +#include +#include +#include +#include +#include #include +#include + +namespace libtest { + +struct Server { +private: + typedef std::vector< std::pair > Options; + +private: + uint64_t _magic; + bool _is_socket; + std::string _socket; + std::string _sasl; + std::string _pid_file; + std::string _log_file; + std::string _base_command; // executable command which include libtool, valgrind, gdb, etc + std::string _running; // Current string being used for system() + +protected: + in_port_t _port; + std::string _hostname; + std::string _extra_args; + +public: + Server(const std::string& hostname, const in_port_t port_arg, + const std::string& executable, const bool _is_libtool, + const bool is_socket_arg= false); + + virtual ~Server(); + + virtual const char *name()= 0; + virtual bool is_libtool()= 0; + + virtual bool has_socket_file_option() const + { + return false; + } + + virtual void socket_file_option(Application& app, const std::string& socket_arg) + { + if (socket_arg.empty() == false) + { + std::string buffer("--socket="); + buffer+= socket_arg; + app.add_option(buffer); + } + } + + virtual bool has_log_file_option() const + { + return false; + } + + virtual void log_file_option(Application& app, const std::string& arg) + { + if (arg.empty() == false) + { + std::string buffer("--log-file="); + buffer+= arg; + app.add_option(buffer); + } + } + + virtual void pid_file_option(Application& app, const std::string& arg) + { + if (arg.empty() == false) + { + std::string buffer("--pid-file="); + buffer+= arg; + app.add_option(buffer); + } + } + + virtual bool has_port_option() const + { + return false; + } + + virtual void port_option(Application& app, in_port_t arg) + { + if (arg > 0) + { + char buffer[1024]; + snprintf(buffer, sizeof(buffer), "--port=%d", int(arg)); + app.add_option(buffer); + } + } + + virtual bool broken_socket_cleanup() + { + return false; + } + + virtual bool broken_pid_file() + { + return false; + } + + const std::string& pid_file() const + { + return _pid_file; + } + + const std::string& base_command() const + { + return _base_command; + } + + const std::string& log_file() const + { + return _log_file; + } + + const std::string& hostname() const + { + return _hostname; + } -/* - Server startup and shutdown functions. -*/ -#ifdef __cplusplus -extern "C" { -#endif - -#include - -typedef struct server_startup_st server_startup_st; -#define SERVERS_TO_CREATE 5 - -struct server_startup_st -{ - uint8_t count; - uint8_t udp; - memcached_server_st *servers; - char *server_list; - char pid_file[SERVERS_TO_CREATE][FILENAME_MAX]; - in_port_t port[SERVERS_TO_CREATE]; - int pids[SERVERS_TO_CREATE]; + const std::string& socket() const + { + return _socket; + } + + bool has_socket() const + { + return _is_socket; + } + + bool cycle(); + + virtual bool ping()= 0; + + virtual bool build(size_t argc, const char *argv[])= 0; + + void add_option(const std::string&); + void add_option(const std::string&, const std::string&); + + in_port_t port() const + { + return _port; + } + + bool has_port() const + { + return (_port != 0); + } + + virtual bool has_syslog() const + { + return false; + } + + // Reset a server if another process has killed the server + void reset() + { + _pid_file.clear(); + _log_file.clear(); + } + + bool args(Application&); + + pid_t pid() const; + + bool has_pid() const; + + virtual bool wait_for_pidfile() const; + + bool check_pid(pid_t pid_arg) const + { + return (pid_arg > 1); + } + + bool is_socket() const + { + return _hostname[0] == '/'; + } + + const std::string running() const + { + return _running; + } + + bool check(); + + std::string log_and_pid(); + + bool kill(); + bool start(); + bool command(libtest::Application& app); + + bool validate(); + +protected: + bool set_pid_file(); + Options _options; + Application _app; + +private: + bool is_helgrind() const; + bool is_valgrind() const; + bool is_debug() const; + bool set_log_file(); + bool set_socket_file(); + void reset_pid(); }; -void server_startup(server_startup_st *construct); -void server_shutdown(server_startup_st *construct); +std::ostream& operator<<(std::ostream& output, const libtest::Server &arg); + +} // namespace libtest + -#ifdef __cplusplus -} -#endif