git.m6w6.name Git - awesomized/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 static inline bool memcached_is_valid_servername(const memcached_string_t& arg)
  47 {
  48   return arg.size > 0 or arg.size < NI_MAXHOST;
  49 }
  50
  51 static inline void memcached_mark_server_as_clean(memcached_server_write_instance_st server)
  52 {
  53   server->server_failure_counter= 0;
  54   server->next_retry= 0;
  55 }
  56
  57
  58 static inline void set_last_disconnected_host(memcached_server_write_instance_st self)
  59 {
  60   assert(self->root);
  61   if (self->root == NULL)
  62   {
  63     return;
  64   }
  65
  66   if (self->root->last_disconnected_server and self->root->last_disconnected_server->version == self->version)
  67   {
  68     return;
  69   }
  70
  71   // const_cast
  72   memcached_st *root= (memcached_st *)self->root;
  73
  74   memcached_server_free(root->last_disconnected_server);
  75   root->last_disconnected_server= memcached_server_clone(NULL, self);
  76   root->last_disconnected_server->version= self->version;
  77 }
  78
  79 static inline void memcached_mark_server_for_timeout(memcached_server_write_instance_st server)
  80 {
  81   if (server->state != MEMCACHED_SERVER_STATE_IN_TIMEOUT)
  82   {
  83     struct timeval next_time;
  84     if (gettimeofday(&next_time, NULL) == 0)
  85     {
  86       server->next_retry= next_time.tv_sec +server->root->retry_timeout;
  87     }
  88     else
  89     {
  90       server->next_retry= 1; // Setting the value to 1 causes the timeout to occur immediatly
  91     }
  92
  93     server->state= MEMCACHED_SERVER_STATE_IN_TIMEOUT;
  94     if (server->server_failure_counter_query_id != server->root->query_id)
  95     {
  96       server->server_failure_counter++;
  97       server->server_failure_counter_query_id= server->root->query_id;
  98     }
  99     set_last_disconnected_host(server);
 100   }
 101 }
 102
 103 LIBMEMCACHED_LOCAL
 104   memcached_server_st *__server_create_with(memcached_st *memc,
 105                                             memcached_server_write_instance_st host,
 106                                             const memcached_string_t& hostname,
 107                                             const in_port_t port,
 108                                             uint32_t weight,
 109                                             const memcached_connection_t type);