X-Git-Url: https://git.m6w6.name/?a=blobdiff_plain;ds=sidebyside;f=libtest%2Fserver.h;h=1aa16ca0a19767b95664e5f4fe3055d1608ecf65;hb=675ac17fddca229e391d80eb610959d9c53db6e5;hp=123d1c3255f6d85af7b9b4ee2b26ef35c3b55099;hpb=58c279d58e2a44562f729e93e301dfedf42f530b;p=m6w6%2Flibmemcached diff --git a/libtest/server.h b/libtest/server.h index 123d1c32..1aa16ca0 100644 --- a/libtest/server.h +++ b/libtest/server.h @@ -21,6 +21,8 @@ #pragma once +#include + #include #include #include @@ -34,6 +36,10 @@ namespace libtest { struct Server { private: + typedef std::vector< std::pair > Options; + +private: + uint64_t _magic; bool _is_socket; std::string _socket; std::string _sasl; @@ -49,26 +55,73 @@ protected: std::string _extra_args; public: - Server(const std::string& hostname, const in_port_t port_arg, const bool is_socket_arg= false); + 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 const char *executable()= 0; - virtual const char *port_option()= 0; - virtual const char *pid_file_option()= 0; - virtual const char *daemon_file_option()= 0; - virtual const char *log_file_option()= 0; virtual bool is_libtool()= 0; - virtual bool broken_socket_cleanup() + virtual bool has_socket_file_option() const { return false; } - virtual const char *socket_file_option() const + 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 NULL; + 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() @@ -112,7 +165,10 @@ public: virtual pid_t get_pid(bool error_is_ok= false)= 0; - virtual bool build(int argc, const char *argv[])= 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 { @@ -124,6 +180,11 @@ public: return (_port != 0); } + virtual bool has_syslog() const + { + return false; + } + // Reset a server if another process has killed the server void reset() { @@ -132,9 +193,7 @@ public: _log_file.clear(); } - void set_extra_args(const std::string &arg); - - bool args(std::string& options); + bool args(Application&); pid_t pid(); @@ -143,10 +202,9 @@ public: return _pid; } - bool has_pid() const - { - return (_pid > 1); - } + bool has_pid() const; + + virtual bool wait_for_pidfile() const; bool check_pid(pid_t pid_arg) const { @@ -163,15 +221,20 @@ public: return _running; } + bool check(); + std::string log_and_pid(); - bool kill(pid_t pid_arg); + bool kill(); bool start(); - bool command(std::string& command_arg); + bool command(libtest::Application& app); + + bool validate(); protected: - void nap(); bool set_pid_file(); + Options _options; + Application _app; private: bool is_helgrind() const; @@ -179,7 +242,6 @@ private: bool is_debug() const; bool set_log_file(); bool set_socket_file(); - void rebuild_base_command(); void reset_pid(); };