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