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