Merge lp:~tangent-org/libmemcached/1.0-build/ Build: jenkins-Libmemcached-187
[m6w6/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 #include <netdb.h>
43 #include <netinet/in.h>
44 #include <string>
45 #include <unistd.h>
46 #include <vector>
47
48 namespace libtest {
49
50 class server_startup_st
51 {
52 private:
53 uint64_t _magic;
54 std::string server_list;
55 bool _socket;
56 bool _sasl;
57 std::string _username;
58 std::string _password;
59
60 public:
61
62 uint8_t udp;
63 std::vector<Server *> servers;
64
65 server_startup_st();
66 ~server_startup_st();
67
68 bool validate();
69
70 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);
71 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);
72
73 uint32_t count() const
74 {
75 return uint32_t(servers.size());
76 }
77
78 void restart();
79
80 std::string option_string() const;
81
82 const std::string& password() const
83 {
84 return _password;
85 }
86
87 const std::string& username() const
88 {
89 return _username;
90 }
91
92 bool socket()
93 {
94 return _socket;
95 }
96
97 bool sasl()
98 {
99 return _sasl;
100 }
101
102 void set_socket()
103 {
104 _socket= true;
105 }
106
107 void set_sasl(const std::string& username_arg, const std::string& password_arg)
108 {
109 _sasl= true;
110 _username= username_arg;
111 _password= password_arg;
112 }
113
114
115 // Just remove everything after shutdown
116 void clear();
117
118 bool shutdown();
119 bool shutdown(uint32_t number_of_host);
120
121 bool check() const;
122
123 void push_server(Server *);
124 Server* last();
125 Server *pop_server();
126
127 Server* create(const std::string& server_type, in_port_t try_port, const bool is_socket);
128
129 unsigned long int servers_to_run() const
130 {
131 return _servers_to_run;
132 }
133
134 void set_servers_to_run(unsigned long int arg)
135 {
136 _servers_to_run= arg;
137 }
138
139 private:
140 bool _start_server(const bool is_socket,
141 const std::string& server_type,
142 const in_port_t try_port,
143 int argc, const char *argv[],
144 const bool opt_startup_message);
145
146 private:
147 unsigned long int _servers_to_run;
148 };
149
150 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);
151
152 } // namespace libtest