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     {
  48       if (data)
  49       {
  50         ptr->hash= MEMCACHED_HASH_MD5;
  51         ptr->distribution= MEMCACHED_DISTRIBUTION_CONSISTENT_KETAMA;
  52       }
  53       else
  54       {
  55         ptr->hash= 0;
  56         ptr->distribution= 0;
  57       }
  58       run_distribution(ptr);
  59       break;
  60     }
  61   case MEMCACHED_BEHAVIOR_HASH:
  62     ptr->hash= (memcached_hash)(data);
  63     break;
  64   case MEMCACHED_BEHAVIOR_CACHE_LOOKUPS:
  65     set_behavior_flag(ptr, MEM_USE_CACHE_LOOKUPS, data);
  66     memcached_quit(ptr);
  67     break;
  68   case MEMCACHED_BEHAVIOR_VERIFY_KEY:
  69     set_behavior_flag(ptr, MEM_VERIFY_KEY, data);
  70     break;
  71   case MEMCACHED_BEHAVIOR_SORT_HOSTS:
  72     {
  73       set_behavior_flag(ptr, MEM_USE_SORT_HOSTS, data);
  74       run_distribution(ptr);
  75
  76       break;
  77     }
  78   case MEMCACHED_BEHAVIOR_POLL_TIMEOUT:
  79     ptr->poll_timeout= (int32_t)data;
  80     break;
  81   case MEMCACHED_BEHAVIOR_CONNECT_TIMEOUT:
  82     ptr->connect_timeout= (int32_t)data;
  83     break;
  84   case MEMCACHED_BEHAVIOR_RETRY_TIMEOUT:
  85     ptr->retry_timeout= (int32_t)data;
  86     break;
  87   case MEMCACHED_BEHAVIOR_SOCKET_SEND_SIZE:
  88     ptr->send_size= (int32_t)data;
  89     memcached_quit(ptr);
  90     break;
  91   case MEMCACHED_BEHAVIOR_SOCKET_RECV_SIZE:
  92     ptr->recv_size= (int32_t)data;
  93     memcached_quit(ptr);
  94     break;
  95   case MEMCACHED_BEHAVIOR_USER_DATA:
  96     return MEMCACHED_FAILURE;
  97   }
  98
  99   return MEMCACHED_SUCCESS;
 100 }
 101
 102 uint64_t memcached_behavior_get(memcached_st *ptr,
 103                                 memcached_behavior flag)
 104 {
 105   memcached_flags temp_flag= 0;
 106
 107   switch (flag)
 108   {
 109   case MEMCACHED_BEHAVIOR_SUPPORT_CAS:
 110     temp_flag= MEM_SUPPORT_CAS;
 111     break;
 112   case MEMCACHED_BEHAVIOR_CACHE_LOOKUPS:
 113     temp_flag= MEM_USE_CACHE_LOOKUPS;
 114     break;
 115   case MEMCACHED_BEHAVIOR_NO_BLOCK:
 116     temp_flag= MEM_NO_BLOCK;
 117     break;
 118   case MEMCACHED_BEHAVIOR_BUFFER_REQUESTS:
 119     temp_flag= MEM_BUFFER_REQUESTS;
 120     break;
 121   case MEMCACHED_BEHAVIOR_TCP_NODELAY:
 122     temp_flag= MEM_TCP_NODELAY;
 123     break;
 124   case MEMCACHED_BEHAVIOR_VERIFY_KEY:
 125     temp_flag= MEM_VERIFY_KEY;
 126     break;
 127   case MEMCACHED_BEHAVIOR_DISTRIBUTION:
 128     return ptr->distribution;
 129   case MEMCACHED_BEHAVIOR_KETAMA:
 130     return (ptr->distribution == MEMCACHED_DISTRIBUTION_CONSISTENT_KETAMA  && ptr->hash == MEMCACHED_HASH_MD5 ) ? 1 : 0;
 131   case MEMCACHED_BEHAVIOR_HASH:
 132     return ptr->hash;
 133   case MEMCACHED_BEHAVIOR_SORT_HOSTS:
 134     temp_flag= MEM_USE_SORT_HOSTS;
 135     break;
 136   case MEMCACHED_BEHAVIOR_POLL_TIMEOUT:
 137     {
 138       return (unsigned long long)ptr->poll_timeout;
 139     }
 140   case MEMCACHED_BEHAVIOR_CONNECT_TIMEOUT:
 141     {
 142       return (unsigned long long)ptr->connect_timeout;
 143     }
 144   case MEMCACHED_BEHAVIOR_RETRY_TIMEOUT:
 145     {
 146       return (unsigned long long)ptr->retry_timeout;
 147     }
 148   case MEMCACHED_BEHAVIOR_SOCKET_SEND_SIZE:
 149     {
 150       int sock_size;
 151       socklen_t sock_length= sizeof(int);
 152
 153       /* REFACTOR */
 154       /* We just try the first host, and if it is down we return zero */
 155       if ((memcached_connect(&ptr->hosts[0])) != MEMCACHED_SUCCESS)
 156         return 0;
 157
 158       if (getsockopt(ptr->hosts[0].fd, SOL_SOCKET,
 159                      SO_SNDBUF, &sock_size, &sock_length))
 160         return 0; /* Zero means error */
 161
 162       return sock_size;
 163     }
 164   case MEMCACHED_BEHAVIOR_SOCKET_RECV_SIZE:
 165     {
 166       int sock_size;
 167       socklen_t sock_length= sizeof(int);
 168
 169       /* REFACTOR */
 170       /* We just try the first host, and if it is down we return zero */
 171       if ((memcached_connect(&ptr->hosts[0])) != MEMCACHED_SUCCESS)
 172         return 0;
 173
 174       if (getsockopt(ptr->hosts[0].fd, SOL_SOCKET,
 175                      SO_RCVBUF, &sock_size, &sock_length))
 176         return 0; /* Zero means error */
 177
 178       return sock_size;
 179     }
 180   case MEMCACHED_BEHAVIOR_USER_DATA:
 181     return MEMCACHED_FAILURE;
 182   }
 183
 184   WATCHPOINT_ASSERT(temp_flag); /* Programming mistake if it gets this far */
 185   if (ptr->flags & temp_flag)
 186     return 1;
 187   else
 188     return 0;
 189
 190   return MEMCACHED_SUCCESS;
 191 }