git.m6w6.name Git - awesomized/libmemcached/blob - libmemcached/memcached_behavior.c

   1 #include "common.h"
   2 #include <sys/types.h>
   3 #include <sys/socket.h>
   4 #include <netinet/tcp.h>
   5
   6 /*
   7   This function is used to modify the behabior of running client.
   8
   9   We quit all connections so we can reset the sockets.
  10 */
  11
  12 void set_behavior_flag(memcached_st *ptr, memcached_flags temp_flag, uint64_t data)
  13 {
  14   if (data)
  15     ptr->flags|= temp_flag;
  16   else
  17     ptr->flags&= ~temp_flag;
  18 }
  19
  20 memcached_return memcached_behavior_set(memcached_st *ptr,
  21                                         memcached_behavior flag,
  22                                         uint64_t data)
  23 {
  24   switch (flag)
  25   {
  26   case MEMCACHED_BEHAVIOR_SUPPORT_CAS:
  27     set_behavior_flag(ptr, MEM_SUPPORT_CAS, data);
  28     break;
  29   case MEMCACHED_BEHAVIOR_NO_BLOCK:
  30     set_behavior_flag(ptr, MEM_NO_BLOCK, data);
  31     memcached_quit(ptr);
  32   case MEMCACHED_BEHAVIOR_BUFFER_REQUESTS:
  33     set_behavior_flag(ptr, MEM_BUFFER_REQUESTS, data);
  34     memcached_quit(ptr);
  35     break;
  36   case MEMCACHED_BEHAVIOR_TCP_NODELAY:
  37     set_behavior_flag(ptr, MEM_TCP_NODELAY, data);
  38     memcached_quit(ptr);
  39     break;
  40   case MEMCACHED_BEHAVIOR_DISTRIBUTION:
  41     {
  42       ptr->distribution= (memcached_server_distribution)(data);
  43       run_distribution(ptr);
  44       break;
  45     }
  46   case MEMCACHED_BEHAVIOR_KETAMA:
  47     if (data)
  48     {
  49       ptr->hash= MEMCACHED_HASH_MD5;
  50       ptr->distribution= MEMCACHED_DISTRIBUTION_CONSISTENT_KETAMA;
  51     }
  52     else
  53     {
  54       ptr->hash= 0;
  55       ptr->distribution= 0;
  56     }
  57     break;
  58   case MEMCACHED_BEHAVIOR_HASH:
  59     ptr->hash= (memcached_hash)(data);
  60     break;
  61   case MEMCACHED_BEHAVIOR_CACHE_LOOKUPS:
  62     set_behavior_flag(ptr, MEM_USE_CACHE_LOOKUPS, data);
  63     memcached_quit(ptr);
  64     break;
  65   case MEMCACHED_BEHAVIOR_VERIFY_KEY:
  66     set_behavior_flag(ptr, MEM_VERIFY_KEY, data);
  67     break;
  68   case MEMCACHED_BEHAVIOR_SORT_HOSTS:
  69     {
  70       set_behavior_flag(ptr, MEM_USE_SORT_HOSTS, data);
  71       run_distribution(ptr);
  72
  73       break;
  74     }
  75   case MEMCACHED_BEHAVIOR_POLL_TIMEOUT:
  76     ptr->poll_timeout= (int32_t)data;
  77     break;
  78   case MEMCACHED_BEHAVIOR_CONNECT_TIMEOUT:
  79     ptr->connect_timeout= (int32_t)data;
  80     break;
  81   case MEMCACHED_BEHAVIOR_RETRY_TIMEOUT:
  82     ptr->retry_timeout= (int32_t)data;
  83     break;
  84   case MEMCACHED_BEHAVIOR_SOCKET_SEND_SIZE:
  85     ptr->send_size= (int32_t)data;
  86     memcached_quit(ptr);
  87     break;
  88   case MEMCACHED_BEHAVIOR_SOCKET_RECV_SIZE:
  89     ptr->recv_size= (int32_t)data;
  90     memcached_quit(ptr);
  91     break;
  92   case MEMCACHED_BEHAVIOR_USER_DATA:
  93     return MEMCACHED_FAILURE;
  94   }
  95
  96   return MEMCACHED_SUCCESS;
  97 }
  98
  99 uint64_t memcached_behavior_get(memcached_st *ptr,
 100                                 memcached_behavior flag)
 101 {
 102   memcached_flags temp_flag= 0;
 103
 104   switch (flag)
 105   {
 106   case MEMCACHED_BEHAVIOR_SUPPORT_CAS:
 107     temp_flag= MEM_SUPPORT_CAS;
 108     break;
 109   case MEMCACHED_BEHAVIOR_CACHE_LOOKUPS:
 110     temp_flag= MEM_USE_CACHE_LOOKUPS;
 111     break;
 112   case MEMCACHED_BEHAVIOR_NO_BLOCK:
 113     temp_flag= MEM_NO_BLOCK;
 114     break;
 115   case MEMCACHED_BEHAVIOR_BUFFER_REQUESTS:
 116     temp_flag= MEM_BUFFER_REQUESTS;
 117     break;
 118   case MEMCACHED_BEHAVIOR_TCP_NODELAY:
 119     temp_flag= MEM_TCP_NODELAY;
 120     break;
 121   case MEMCACHED_BEHAVIOR_VERIFY_KEY:
 122     temp_flag= MEM_VERIFY_KEY;
 123     break;
 124   case MEMCACHED_BEHAVIOR_DISTRIBUTION:
 125     return ptr->distribution;
 126   case MEMCACHED_BEHAVIOR_KETAMA:
 127     return (ptr->distribution == MEMCACHED_DISTRIBUTION_CONSISTENT_KETAMA  && ptr->hash == MEMCACHED_HASH_MD5 ) ? 1 : 0;
 128   case MEMCACHED_BEHAVIOR_HASH:
 129     return ptr->hash;
 130   case MEMCACHED_BEHAVIOR_SORT_HOSTS:
 131     temp_flag= MEM_USE_SORT_HOSTS;
 132     break;
 133   case MEMCACHED_BEHAVIOR_POLL_TIMEOUT:
 134     {
 135       return (unsigned long long)ptr->poll_timeout;
 136     }
 137   case MEMCACHED_BEHAVIOR_CONNECT_TIMEOUT:
 138     {
 139       return (unsigned long long)ptr->connect_timeout;
 140     }
 141   case MEMCACHED_BEHAVIOR_RETRY_TIMEOUT:
 142     {
 143       return (unsigned long long)ptr->retry_timeout;
 144     }
 145   case MEMCACHED_BEHAVIOR_SOCKET_SEND_SIZE:
 146     {
 147       int sock_size;
 148       socklen_t sock_length= sizeof(int);
 149
 150       /* REFACTOR */
 151       /* We just try the first host, and if it is down we return zero */
 152       if ((memcached_connect(&ptr->hosts[0])) != MEMCACHED_SUCCESS)
 153         return 0;
 154
 155       if (getsockopt(ptr->hosts[0].fd, SOL_SOCKET,
 156                      SO_SNDBUF, &sock_size, &sock_length))
 157         return 0; /* Zero means error */
 158
 159       return sock_size;
 160     }
 161   case MEMCACHED_BEHAVIOR_SOCKET_RECV_SIZE:
 162     {
 163       int sock_size;
 164       socklen_t sock_length= sizeof(int);
 165
 166       /* REFACTOR */
 167       /* We just try the first host, and if it is down we return zero */
 168       if ((memcached_connect(&ptr->hosts[0])) != MEMCACHED_SUCCESS)
 169         return 0;
 170
 171       if (getsockopt(ptr->hosts[0].fd, SOL_SOCKET,
 172                      SO_RCVBUF, &sock_size, &sock_length))
 173         return 0; /* Zero means error */
 174
 175       return sock_size;
 176     }
 177   case MEMCACHED_BEHAVIOR_USER_DATA:
 178     return MEMCACHED_FAILURE;
 179   }
 180
 181   WATCHPOINT_ASSERT(temp_flag); /* Programming mistake if it gets this far */
 182   if (ptr->flags & temp_flag)
 183     return 1;
 184   else
 185     return 0;
 186
 187   return MEMCACHED_SUCCESS;
 188 }