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     break;
  35   case MEMCACHED_BEHAVIOR_TCP_NODELAY:
  36     set_behavior_flag(ptr, MEM_TCP_NODELAY, data);
  37     break;
  38   case MEMCACHED_BEHAVIOR_DISTRIBUTION:
  39     ptr->distribution= *(memcached_server_distribution *)(data);
  40     break;
  41   case MEMCACHED_BEHAVIOR_HASH:
  42     ptr->hash= *(memcached_hash *)(data);
  43     break;
  44   case MEMCACHED_BEHAVIOR_CACHE_LOOKUPS:
  45     set_behavior_flag(ptr, MEM_USE_CACHE_LOOKUPS, data);
  46     break;
  47   case MEMCACHED_BEHAVIOR_KETAMA:
  48     set_behavior_flag(ptr, MEM_USE_KETAMA, data);
  49     break;
  50   case MEMCACHED_BEHAVIOR_POLL_TIMEOUT:
  51     {
  52       unsigned int timeout= (*((unsigned int *)data));
  53
  54       ptr->poll_timeout= timeout;
  55       break;
  56     }
  57   case MEMCACHED_BEHAVIOR_SOCKET_SEND_SIZE:
  58     {
  59       ptr->send_size= (*((int *)data));
  60       memcached_quit(ptr);
  61       break;
  62     }
  63   case MEMCACHED_BEHAVIOR_SOCKET_RECV_SIZE:
  64     {
  65       ptr->recv_size= (*((int *)data));
  66       memcached_quit(ptr);
  67       break;
  68     }
  69
  70
  71   }
  72
  73   return MEMCACHED_SUCCESS;
  74 }
  75
  76 unsigned long long memcached_behavior_get(memcached_st *ptr,
  77                                           memcached_behavior flag)
  78 {
  79   memcached_flags temp_flag= 0;
  80
  81   switch (flag)
  82   {
  83   case MEMCACHED_BEHAVIOR_SUPPORT_CAS:
  84     temp_flag= MEM_SUPPORT_CAS;
  85     break;
  86   case MEMCACHED_BEHAVIOR_CACHE_LOOKUPS:
  87     temp_flag= MEM_USE_CACHE_LOOKUPS;
  88     break;
  89   case MEMCACHED_BEHAVIOR_NO_BLOCK:
  90     temp_flag= MEM_NO_BLOCK;
  91     break;
  92   case MEMCACHED_BEHAVIOR_TCP_NODELAY:
  93     temp_flag= MEM_TCP_NODELAY;
  94     break;
  95   case MEMCACHED_BEHAVIOR_DISTRIBUTION:
  96     return ptr->distribution;
  97   case MEMCACHED_BEHAVIOR_HASH:
  98     return ptr->hash;
  99   case MEMCACHED_BEHAVIOR_KETAMA:
 100     temp_flag= MEM_USE_KETAMA;
 101     break;
 102   case MEMCACHED_BEHAVIOR_POLL_TIMEOUT:
 103     {
 104       return (unsigned long long)ptr->poll_timeout;
 105     }
 106   case MEMCACHED_BEHAVIOR_SOCKET_SEND_SIZE:
 107     {
 108       int sock_size;
 109       socklen_t sock_length= sizeof(int);
 110
 111       /* We just try the first host, and if it is down we return zero */
 112       if ((memcached_connect(ptr, 0)) != MEMCACHED_SUCCESS)
 113         return 0;
 114
 115       if (getsockopt(ptr->hosts[0].fd, SOL_SOCKET,
 116                      SO_SNDBUF, &sock_size, &sock_length))
 117         return 0; /* Zero means error */
 118
 119       return sock_size;
 120     }
 121   case MEMCACHED_BEHAVIOR_SOCKET_RECV_SIZE:
 122     {
 123       int sock_size;
 124       socklen_t sock_length= sizeof(int);
 125
 126       /* We just try the first host, and if it is down we return zero */
 127       if ((memcached_connect(ptr, 0)) != MEMCACHED_SUCCESS)
 128         return 0;
 129
 130       if (getsockopt(ptr->hosts[0].fd, SOL_SOCKET,
 131                      SO_RCVBUF, &sock_size, &sock_length))
 132         return 0; /* Zero means error */
 133
 134       return sock_size;
 135     }
 136   }
 137
 138   WATCHPOINT_ASSERT(temp_flag); /* Programming mistake if it gets this far */
 139   if (ptr->flags & temp_flag)
 140     return 1;
 141   else
 142     return 0;
 143
 144   return MEMCACHED_SUCCESS;
 145 }