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, void *data)
  13 {
  14   uint8_t truefalse;
  15
  16   if (data)
  17     truefalse= *(unsigned int *)data;
  18   else
  19     truefalse= 0;
  20
  21   if (truefalse)
  22     ptr->flags|= temp_flag;
  23   else
  24     ptr->flags+= temp_flag;
  25 }
  26
  27 memcached_return memcached_behavior_set(memcached_st *ptr,
  28                                         memcached_behavior flag,
  29                                         void *data)
  30 {
  31   switch (flag)
  32   {
  33   case MEMCACHED_BEHAVIOR_SUPPORT_CAS:
  34     set_behavior_flag(ptr, MEM_SUPPORT_CAS, data);
  35     break;
  36   case MEMCACHED_BEHAVIOR_NO_BLOCK:
  37     set_behavior_flag(ptr, MEM_NO_BLOCK, data);
  38     memcached_quit(ptr);
  39   case MEMCACHED_BEHAVIOR_BUFFER_REQUESTS:
  40     set_behavior_flag(ptr, MEM_BUFFER_REQUESTS, data);
  41     memcached_quit(ptr);
  42     break;
  43   case MEMCACHED_BEHAVIOR_TCP_NODELAY:
  44     set_behavior_flag(ptr, MEM_TCP_NODELAY, data);
  45     memcached_quit(ptr);
  46     break;
  47   case MEMCACHED_BEHAVIOR_DISTRIBUTION:
  48     ptr->distribution= *(memcached_server_distribution *)(data);
  49     break;
  50   case MEMCACHED_BEHAVIOR_HASH:
  51     ptr->hash= *(memcached_hash *)(data);
  52     break;
  53   case MEMCACHED_BEHAVIOR_CACHE_LOOKUPS:
  54     set_behavior_flag(ptr, MEM_USE_CACHE_LOOKUPS, data);
  55     memcached_quit(ptr);
  56     break;
  57   case MEMCACHED_BEHAVIOR_VERIFY_KEY:
  58     set_behavior_flag(ptr, MEM_VERIFY_KEY, data);
  59     break;
  60   case MEMCACHED_BEHAVIOR_KETAMA:
  61     set_behavior_flag(ptr, MEM_USE_KETAMA, data);
  62     break;
  63   case MEMCACHED_BEHAVIOR_SORT_HOSTS:
  64     set_behavior_flag(ptr, MEM_USE_SORT_HOSTS, data);
  65     break;
  66   case MEMCACHED_BEHAVIOR_USER_DATA:
  67     ptr->user_data= 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_RETRY_TIMEOUT:
  84     {
  85       ptr->retry_timeout= (int32_t)data;
  86       break;
  87     }
  88   case MEMCACHED_BEHAVIOR_SOCKET_SEND_SIZE:
  89     {
  90       ptr->send_size= (*((int *)data));
  91       memcached_quit(ptr);
  92       break;
  93     }
  94   case MEMCACHED_BEHAVIOR_SOCKET_RECV_SIZE:
  95     {
  96       ptr->recv_size= (*((int *)data));
  97       memcached_quit(ptr);
  98       break;
  99     }
 100
 101
 102   }
 103
 104   return MEMCACHED_SUCCESS;
 105 }
 106
 107 unsigned long long memcached_behavior_get(memcached_st *ptr,
 108                                           memcached_behavior flag)
 109 {
 110   memcached_flags temp_flag= 0;
 111
 112   switch (flag)
 113   {
 114   case MEMCACHED_BEHAVIOR_SUPPORT_CAS:
 115     temp_flag= MEM_SUPPORT_CAS;
 116     break;
 117   case MEMCACHED_BEHAVIOR_CACHE_LOOKUPS:
 118     temp_flag= MEM_USE_CACHE_LOOKUPS;
 119     break;
 120   case MEMCACHED_BEHAVIOR_NO_BLOCK:
 121     temp_flag= MEM_NO_BLOCK;
 122     break;
 123   case MEMCACHED_BEHAVIOR_BUFFER_REQUESTS:
 124     temp_flag= MEM_BUFFER_REQUESTS;
 125     break;
 126   case MEMCACHED_BEHAVIOR_TCP_NODELAY:
 127     temp_flag= MEM_TCP_NODELAY;
 128     break;
 129   case MEMCACHED_BEHAVIOR_VERIFY_KEY:
 130     temp_flag= MEM_VERIFY_KEY;
 131     break;
 132   case MEMCACHED_BEHAVIOR_DISTRIBUTION:
 133     return ptr->distribution;
 134   case MEMCACHED_BEHAVIOR_HASH:
 135     return ptr->hash;
 136   case MEMCACHED_BEHAVIOR_KETAMA:
 137     temp_flag= MEM_USE_KETAMA;
 138     break;
 139   case MEMCACHED_BEHAVIOR_SORT_HOSTS:
 140     temp_flag= MEM_USE_SORT_HOSTS;
 141     break;
 142   case MEMCACHED_BEHAVIOR_USER_DATA:
 143     return 0;
 144   case MEMCACHED_BEHAVIOR_POLL_TIMEOUT:
 145     {
 146       return (unsigned long long)ptr->poll_timeout;
 147     }
 148   case MEMCACHED_BEHAVIOR_CONNECT_TIMEOUT:
 149     {
 150       return (unsigned long long)ptr->connect_timeout;
 151     }
 152   case MEMCACHED_BEHAVIOR_RETRY_TIMEOUT:
 153     {
 154       return (unsigned long long)ptr->retry_timeout;
 155     }
 156   case MEMCACHED_BEHAVIOR_SOCKET_SEND_SIZE:
 157     {
 158       int sock_size;
 159       socklen_t sock_length= sizeof(int);
 160
 161       /* REFACTOR */
 162       /* We just try the first host, and if it is down we return zero */
 163       if ((memcached_connect(&ptr->hosts[0])) != MEMCACHED_SUCCESS)
 164         return 0;
 165
 166       if (getsockopt(ptr->hosts[0].fd, SOL_SOCKET,
 167                      SO_SNDBUF, &sock_size, &sock_length))
 168         return 0; /* Zero means error */
 169
 170       return sock_size;
 171     }
 172   case MEMCACHED_BEHAVIOR_SOCKET_RECV_SIZE:
 173     {
 174       int sock_size;
 175       socklen_t sock_length= sizeof(int);
 176
 177       /* REFACTOR */
 178       /* We just try the first host, and if it is down we return zero */
 179       if ((memcached_connect(&ptr->hosts[0])) != MEMCACHED_SUCCESS)
 180         return 0;
 181
 182       if (getsockopt(ptr->hosts[0].fd, SOL_SOCKET,
 183                      SO_RCVBUF, &sock_size, &sock_length))
 184         return 0; /* Zero means error */
 185
 186       return sock_size;
 187     }
 188   }
 189
 190   WATCHPOINT_ASSERT(temp_flag); /* Programming mistake if it gets this far */
 191   if (ptr->flags & temp_flag)
 192     return 1;
 193   else
 194     return 0;
 195
 196   return MEMCACHED_SUCCESS;
 197 }