git.m6w6.name Git - awesomized/libmemcached/blob - lib/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_REPLICAS:
  27     {
  28       uint8_t number_of_replicas= (uint8_t)data;
  29
  30       if (number_of_replicas > ptr->number_of_hosts || number_of_replicas == 0 || number_of_replicas > MEMCACHED_MAX_REPLICAS)
  31         return MEMCACHED_FAILURE;
  32       else
  33         ptr->number_of_replicas= number_of_replicas;
  34       break;
  35     }
  36   case MEMCACHED_BEHAVIOR_SUPPORT_CAS:
  37     set_behavior_flag(ptr, MEM_SUPPORT_CAS, data);
  38     break;
  39   case MEMCACHED_BEHAVIOR_NO_BLOCK:
  40     set_behavior_flag(ptr, MEM_NO_BLOCK, data);
  41     memcached_quit(ptr);
  42   case MEMCACHED_BEHAVIOR_BUFFER_REQUESTS:
  43     set_behavior_flag(ptr, MEM_BUFFER_REQUESTS, data);
  44     memcached_quit(ptr);
  45     break;
  46   case MEMCACHED_BEHAVIOR_TCP_NODELAY:
  47     set_behavior_flag(ptr, MEM_TCP_NODELAY, data);
  48     memcached_quit(ptr);
  49     break;
  50   case MEMCACHED_BEHAVIOR_DISTRIBUTION:
  51     ptr->distribution= (memcached_server_distribution)data;
  52     break;
  53   case MEMCACHED_BEHAVIOR_HASH:
  54     ptr->hash= (memcached_hash)(data);
  55     break;
  56   case MEMCACHED_BEHAVIOR_CACHE_LOOKUPS:
  57     set_behavior_flag(ptr, MEM_USE_CACHE_LOOKUPS, data);
  58     memcached_quit(ptr);
  59     break;
  60   case MEMCACHED_BEHAVIOR_VERIFY_KEY:
  61     set_behavior_flag(ptr, MEM_VERIFY_KEY, data);
  62     break;
  63   case MEMCACHED_BEHAVIOR_KETAMA:
  64     set_behavior_flag(ptr, MEM_USE_KETAMA, data);
  65     break;
  66   case MEMCACHED_BEHAVIOR_SORT_HOSTS:
  67     set_behavior_flag(ptr, MEM_USE_SORT_HOSTS, data);
  68     break;
  69   case MEMCACHED_BEHAVIOR_POLL_TIMEOUT:
  70     {
  71       int32_t timeout= (int32_t)data;
  72
  73       ptr->poll_timeout= timeout;
  74       break;
  75     }
  76   case MEMCACHED_BEHAVIOR_CONNECT_TIMEOUT:
  77     {
  78       int32_t timeout= (int32_t)data;
  79
  80       ptr->connect_timeout= timeout;
  81       break;
  82     }
  83   case MEMCACHED_BEHAVIOR_SOCKET_SEND_SIZE:
  84     {
  85       ptr->send_size= data;
  86       memcached_quit(ptr);
  87       break;
  88     }
  89   case MEMCACHED_BEHAVIOR_SOCKET_RECV_SIZE:
  90     {
  91       ptr->recv_size= data;
  92       memcached_quit(ptr);
  93       break;
  94     }
  95
  96
  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_REPLICAS:
 110     return (unsigned long long)ptr->number_of_replicas;
 111   case MEMCACHED_BEHAVIOR_SUPPORT_CAS:
 112     temp_flag= MEM_SUPPORT_CAS;
 113     break;
 114   case MEMCACHED_BEHAVIOR_CACHE_LOOKUPS:
 115     temp_flag= MEM_USE_CACHE_LOOKUPS;
 116     break;
 117   case MEMCACHED_BEHAVIOR_NO_BLOCK:
 118     temp_flag= MEM_NO_BLOCK;
 119     break;
 120   case MEMCACHED_BEHAVIOR_BUFFER_REQUESTS:
 121     temp_flag= MEM_BUFFER_REQUESTS;
 122     break;
 123   case MEMCACHED_BEHAVIOR_TCP_NODELAY:
 124     temp_flag= MEM_TCP_NODELAY;
 125     break;
 126   case MEMCACHED_BEHAVIOR_VERIFY_KEY:
 127     temp_flag= MEM_VERIFY_KEY;
 128     break;
 129   case MEMCACHED_BEHAVIOR_DISTRIBUTION:
 130     return ptr->distribution;
 131   case MEMCACHED_BEHAVIOR_HASH:
 132     return ptr->hash;
 133   case MEMCACHED_BEHAVIOR_KETAMA:
 134     temp_flag= MEM_USE_KETAMA;
 135     break;
 136   case MEMCACHED_BEHAVIOR_SORT_HOSTS:
 137     temp_flag= MEM_USE_SORT_HOSTS;
 138     break;
 139   case MEMCACHED_BEHAVIOR_POLL_TIMEOUT:
 140     {
 141       return (unsigned long long)ptr->poll_timeout;
 142     }
 143   case MEMCACHED_BEHAVIOR_CONNECT_TIMEOUT:
 144     {
 145       return (unsigned long long)ptr->connect_timeout;
 146     }
 147   case MEMCACHED_BEHAVIOR_SOCKET_SEND_SIZE:
 148     {
 149       int sock_size;
 150       socklen_t sock_length= sizeof(int);
 151
 152       /* REFACTOR */
 153       /* We just try the first host, and if it is down we return zero */
 154       if ((memcached_connect(&ptr->hosts[0])) != MEMCACHED_SUCCESS)
 155         return 0;
 156
 157       if (getsockopt(ptr->hosts[0].fd, SOL_SOCKET,
 158                      SO_SNDBUF, &sock_size, &sock_length))
 159         return 0; /* Zero means error */
 160
 161       return sock_size;
 162     }
 163   case MEMCACHED_BEHAVIOR_SOCKET_RECV_SIZE:
 164     {
 165       int sock_size;
 166       socklen_t sock_length= sizeof(int);
 167
 168       /* REFACTOR */
 169       /* We just try the first host, and if it is down we return zero */
 170       if ((memcached_connect(&ptr->hosts[0])) != MEMCACHED_SUCCESS)
 171         return 0;
 172
 173       if (getsockopt(ptr->hosts[0].fd, SOL_SOCKET,
 174                      SO_RCVBUF, &sock_size, &sock_length))
 175         return 0; /* Zero means error */
 176
 177       return sock_size;
 178     }
 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 }