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_SOCKET_SEND_SIZE:
  84     {
  85       ptr->send_size= (*((int *)data));
  86       memcached_quit(ptr);
  87       break;
  88     }
  89   case MEMCACHED_BEHAVIOR_SOCKET_RECV_SIZE:
  90     {
  91       ptr->recv_size= (*((int *)data));
  92       memcached_quit(ptr);
  93       break;
  94     }
  95
  96
  97   }
  98
  99   return MEMCACHED_SUCCESS;
 100 }
 101
 102 unsigned long long 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_HASH:
 130     return ptr->hash;
 131   case MEMCACHED_BEHAVIOR_KETAMA:
 132     temp_flag= MEM_USE_KETAMA;
 133     break;
 134   case MEMCACHED_BEHAVIOR_SORT_HOSTS:
 135     temp_flag= MEM_USE_SORT_HOSTS;
 136     break;
 137   case MEMCACHED_BEHAVIOR_USER_DATA:
 138     return 0;
 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 }