Merge lp:~tangent-org/libmemcached/1.0-build/ Build: jenkins-Libmemcached-192
[awesomized/libmemcached] / libtest / server_container.h
1 /* vim:expandtab:shiftwidth=2:tabstop=2:smarttab:
2 *
3 * Data Differential YATL (i.e. libtest) library
4 *
5 * Copyright (C) 2012 Data Differential, http://datadifferential.com/
6 *
7 * Redistribution and use in source and binary forms, with or without
8 * modification, are permitted provided that the following conditions are
9 * met:
10 *
11 * * Redistributions of source code must retain the above copyright
12 * notice, this list of conditions and the following disclaimer.
13 *
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
17 * distribution.
18 *
19 * * The names of its contributors may not be used to endorse or
20 * promote products derived from this software without specific prior
21 * written permission.
22 *
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.
34 *
35 */
36
37 #pragma once
38
39 #include <cassert>
40 #include <cstdio>
41 #include <cstring>
42
43 #include <netdb.h>
44 #include <netinet/in.h>
45
46 #include <string>
47 #include <unistd.h>
48 #include <vector>
49
50 namespace libtest {
51
52 class server_startup_st
53 {
54 private:
55 uint64_t _magic;
56 std::string server_list;
57 bool _socket;
58 bool _sasl;
59 std::string _username;
60 std::string _password;
61
62 public:
63
64 uint8_t udp;
65 std::vector<Server *> servers;
66
67 server_startup_st();
68 ~server_startup_st();
69
70 bool validate();
71
72 bool start_socket_server(const std::string& server_type, const in_port_t try_port, int argc, const char *argv[], const bool opt_startup_message= true);
73 bool start_server(const std::string& server_type, const in_port_t try_port, int argc, const char *argv[], const bool opt_startup_message= true);
74
75 uint32_t count() const
76 {
77 return uint32_t(servers.size());
78 }
79
80 void restart();
81
82 std::string option_string() const;
83
84 const std::string& password() const
85 {
86 return _password;
87 }
88
89 const std::string& username() const
90 {
91 return _username;
92 }
93
94 bool socket()
95 {
96 return _socket;
97 }
98
99 bool sasl()
100 {
101 return _sasl;
102 }
103
104 void set_socket()
105 {
106 _socket= true;
107 }
108
109 void set_sasl(const std::string& username_arg, const std::string& password_arg)
110 {
111 _sasl= true;
112 _username= username_arg;
113 _password= password_arg;
114 }
115
116
117 // Just remove everything after shutdown
118 void clear();
119
120 bool shutdown();
121 bool shutdown(uint32_t number_of_host);
122
123 bool check() const;
124
125 void push_server(Server *);
126 Server* last();
127 Server *pop_server();
128
129 Server* create(const std::string& server_type, in_port_t try_port, const bool is_socket);
130
131 unsigned long int servers_to_run() const
132 {
133 return _servers_to_run;
134 }
135
136 void set_servers_to_run(unsigned long int arg)
137 {
138 _servers_to_run= arg;
139 }
140
141 private:
142 bool _start_server(const bool is_socket,
143 const std::string& server_type,
144 const in_port_t try_port,
145 int argc, const char *argv[],
146 const bool opt_startup_message);
147
148 private:
149 unsigned long int _servers_to_run;
150 };
151
152 bool server_startup(server_startup_st&, const std::string&, in_port_t try_port, int argc, const char *argv[], const bool opt_startup_message= true);
153
154 } // namespace libtest