Merge in updates for server failure testing.
[m6w6/libmemcached] / libmemcached / server.hpp
1 /* vim:expandtab:shiftwidth=2:tabstop=2:smarttab:
2 *
3 * Libmemcached library
4 *
5 * Copyright (C) 2011 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
38 #pragma once
39
40 #include <libmemcached/basic_string.h>
41
42 #ifdef HAVE_SYS_TIME_H
43 #include <sys/time.h>
44 #endif
45
46 #include <cassert>
47
48 static inline bool memcached_is_valid_servername(const memcached_string_t& arg)
49 {
50 return arg.size > 0 or arg.size < NI_MAXHOST;
51 }
52
53 static inline void memcached_mark_server_as_clean(memcached_server_write_instance_st server)
54 {
55 server->server_failure_counter= 0;
56 server->next_retry= 0;
57 }
58
59
60 static inline void set_last_disconnected_host(memcached_server_write_instance_st self)
61 {
62 assert(self->root);
63 if (self->root == NULL)
64 {
65 return;
66 }
67
68 if (self->root->last_disconnected_server and self->root->last_disconnected_server->version == self->version)
69 {
70 return;
71 }
72
73 // const_cast
74 memcached_st *root= (memcached_st *)self->root;
75
76 memcached_server_free(root->last_disconnected_server);
77 root->last_disconnected_server= memcached_server_clone(NULL, self);
78 root->last_disconnected_server->version= self->version;
79 }
80
81 static inline void memcached_mark_server_for_timeout(memcached_server_write_instance_st server)
82 {
83 if (server->state != MEMCACHED_SERVER_STATE_IN_TIMEOUT)
84 {
85 struct timeval next_time;
86 if (gettimeofday(&next_time, NULL) == 0)
87 {
88 server->next_retry= next_time.tv_sec +server->root->retry_timeout;
89 }
90 else
91 {
92 server->next_retry= 1; // Setting the value to 1 causes the timeout to occur immediatly
93 }
94
95 server->state= MEMCACHED_SERVER_STATE_IN_TIMEOUT;
96 server->server_failure_counter++;
97 set_last_disconnected_host(server);
98 }
99 }
100
101 LIBMEMCACHED_LOCAL
102 memcached_server_st *__server_create_with(memcached_st *memc,
103 memcached_server_write_instance_st host,
104 const memcached_string_t& hostname,
105 const in_port_t port,
106 uint32_t weight,
107 const memcached_connection_t type);