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   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_SOCKET_SEND_SIZE:
  77     {
  78       ptr->send_size= (*((int *)data));
  79       memcached_quit(ptr);
  80       break;
  81     }
  82   case MEMCACHED_BEHAVIOR_SOCKET_RECV_SIZE:
  83     {
  84       ptr->recv_size= (*((int *)data));
  85       memcached_quit(ptr);
  86       break;
  87     }
  88
  89
  90   }
  91
  92   return MEMCACHED_SUCCESS;
  93 }
  94
  95 unsigned long long memcached_behavior_get(memcached_st *ptr,
  96                                           memcached_behavior flag)
  97 {
  98   memcached_flags temp_flag= 0;
  99
 100   switch (flag)
 101   {
 102   case MEMCACHED_BEHAVIOR_SUPPORT_CAS:
 103     temp_flag= MEM_SUPPORT_CAS;
 104     break;
 105   case MEMCACHED_BEHAVIOR_CACHE_LOOKUPS:
 106     temp_flag= MEM_USE_CACHE_LOOKUPS;
 107     break;
 108   case MEMCACHED_BEHAVIOR_NO_BLOCK:
 109     temp_flag= MEM_NO_BLOCK;
 110     break;
 111   case MEMCACHED_BEHAVIOR_BUFFER_REQUESTS:
 112     temp_flag= MEM_BUFFER_REQUESTS;
 113     break;
 114   case MEMCACHED_BEHAVIOR_TCP_NODELAY:
 115     temp_flag= MEM_TCP_NODELAY;
 116     break;
 117   case MEMCACHED_BEHAVIOR_VERIFY_KEY:
 118     temp_flag= MEM_VERIFY_KEY;
 119     break;
 120   case MEMCACHED_BEHAVIOR_DISTRIBUTION:
 121     return ptr->distribution;
 122   case MEMCACHED_BEHAVIOR_HASH:
 123     return ptr->hash;
 124   case MEMCACHED_BEHAVIOR_KETAMA:
 125     temp_flag= MEM_USE_KETAMA;
 126     break;
 127   case MEMCACHED_BEHAVIOR_SORT_HOSTS:
 128     temp_flag= MEM_USE_SORT_HOSTS;
 129     break;
 130   case MEMCACHED_BEHAVIOR_USER_DATA:
 131     return 0;
 132   case MEMCACHED_BEHAVIOR_POLL_TIMEOUT:
 133     {
 134       return (unsigned long long)ptr->poll_timeout;
 135     }
 136   case MEMCACHED_BEHAVIOR_SOCKET_SEND_SIZE:
 137     {
 138       int sock_size;
 139       socklen_t sock_length= sizeof(int);
 140
 141       /* REFACTOR */
 142       /* We just try the first host, and if it is down we return zero */
 143       if ((memcached_connect(&ptr->hosts[0])) != MEMCACHED_SUCCESS)
 144         return 0;
 145
 146       if (getsockopt(ptr->hosts[0].fd, SOL_SOCKET,
 147                      SO_SNDBUF, &sock_size, &sock_length))
 148         return 0; /* Zero means error */
 149
 150       return sock_size;
 151     }
 152   case MEMCACHED_BEHAVIOR_SOCKET_RECV_SIZE:
 153     {
 154       int sock_size;
 155       socklen_t sock_length= sizeof(int);
 156
 157       /* REFACTOR */
 158       /* We just try the first host, and if it is down we return zero */
 159       if ((memcached_connect(&ptr->hosts[0])) != MEMCACHED_SUCCESS)
 160         return 0;
 161
 162       if (getsockopt(ptr->hosts[0].fd, SOL_SOCKET,
 163                      SO_RCVBUF, &sock_size, &sock_length))
 164         return 0; /* Zero means error */
 165
 166       return sock_size;
 167     }
 168   }
 169
 170   WATCHPOINT_ASSERT(temp_flag); /* Programming mistake if it gets this far */
 171   if (ptr->flags & temp_flag)
 172     return 1;
 173   else
 174     return 0;
 175
 176   return MEMCACHED_SUCCESS;
 177 }