git.m6w6.name Git - m6w6/libmemcached/blob - 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 #ifdef HAVE_SYS_TIME_H
  41 #include <sys/time.h>
  42 #endif
  43
  44 #include <cassert>
  45
  46 memcached_server_st *__server_create_with(memcached_st *memc,
  47                                           memcached_server_write_instance_st host,
  48                                           const memcached_string_t& hostname,
  49                                           const in_port_t port,
  50                                           uint32_t weight,
  51                                           const memcached_connection_t type);
  52
  53 memcached_server_st *memcached_server_clone(memcached_server_st *destination,
  54                                             memcached_server_st *source);
  55
  56 memcached_return_t memcached_server_add_parsed(memcached_st *ptr,
  57                                                const char *hostname,
  58                                                size_t hostname_length,
  59                                                in_port_t port,
  60                                                uint32_t weight);
  61
  62 void __server_free(memcached_server_st *);
  63
  64 static inline bool memcached_is_valid_servername(const memcached_string_t& arg)
  65 {
  66   return arg.size > 0 or arg.size < NI_MAXHOST;
  67 }
  68
  69 static inline void memcached_mark_server_as_clean(memcached_server_write_instance_st server)
  70 {
  71   server->server_failure_counter= 0;
  72   server->next_retry= 0;
  73 }
  74
  75
  76 static inline void set_last_disconnected_host(memcached_server_write_instance_st self)
  77 {
  78   assert(self->root);
  79   if (self->root == NULL)
  80   {
  81     return;
  82   }
  83
  84   if (self->root->last_disconnected_server and self->root->last_disconnected_server->version == self->version)
  85   {
  86     return;
  87   }
  88
  89   // const_cast
  90   memcached_st *root= (memcached_st *)self->root;
  91
  92   memcached_server_free(root->last_disconnected_server);
  93   root->last_disconnected_server= memcached_server_clone(NULL, self);
  94   root->last_disconnected_server->version= self->version;
  95 }
  96
  97 static inline void memcached_mark_server_for_timeout(memcached_server_write_instance_st server)
  98 {
  99   if (server->state != MEMCACHED_SERVER_STATE_IN_TIMEOUT)
 100   {
 101     struct timeval next_time;
 102     if (gettimeofday(&next_time, NULL) == 0)
 103     {
 104       server->next_retry= next_time.tv_sec +server->root->retry_timeout;
 105     }
 106     else
 107     {
 108       server->next_retry= 1; // Setting the value to 1 causes the timeout to occur immediatly
 109     }
 110
 111     server->state= MEMCACHED_SERVER_STATE_IN_TIMEOUT;
 112     if (server->server_failure_counter_query_id != server->root->query_id)
 113     {
 114       server->server_failure_counter++;
 115       server->server_failure_counter_query_id= server->root->query_id;
 116     }
 117     set_last_disconnected_host(server);
 118   }
 119 }