We quit all connections so we can reset the sockets.
*/
-void set_behavior_flag(memcached_st *ptr, memcached_flags temp_flag, void *data)
+void set_behavior_flag(memcached_st *ptr, memcached_flags temp_flag, uint64_t data)
{
- unsigned int *truefalse= (unsigned int *)data;
-
- memcached_quit(ptr);
- if (truefalse)
+ if (data)
ptr->flags|= temp_flag;
else
- ptr->flags+= temp_flag;
+ ptr->flags&= ~temp_flag;
}
memcached_return memcached_behavior_set(memcached_st *ptr,
memcached_behavior flag,
- void *data)
+ uint64_t data)
{
switch (flag)
{
+ case MEMCACHED_BEHAVIOR_REPLICAS:
+ {
+ uint8_t number_of_replicas= (uint8_t)data;
+
+ if (number_of_replicas > ptr->number_of_hosts || number_of_replicas == 0 || number_of_replicas > MEMCACHED_MAX_REPLICAS)
+ return MEMCACHED_FAILURE;
+ else
+ ptr->number_of_replicas= number_of_replicas;
+ break;
+ }
+ case MEMCACHED_BEHAVIOR_SUPPORT_CAS:
+ set_behavior_flag(ptr, MEM_SUPPORT_CAS, data);
+ break;
case MEMCACHED_BEHAVIOR_NO_BLOCK:
set_behavior_flag(ptr, MEM_NO_BLOCK, data);
+ memcached_quit(ptr);
+ case MEMCACHED_BEHAVIOR_BUFFER_REQUESTS:
+ set_behavior_flag(ptr, MEM_BUFFER_REQUESTS, data);
+ memcached_quit(ptr);
break;
case MEMCACHED_BEHAVIOR_TCP_NODELAY:
set_behavior_flag(ptr, MEM_TCP_NODELAY, data);
+ memcached_quit(ptr);
+ break;
+ case MEMCACHED_BEHAVIOR_DISTRIBUTION:
+ ptr->distribution= (memcached_server_distribution)data;
+ break;
+ case MEMCACHED_BEHAVIOR_HASH:
+ ptr->hash= (memcached_hash)(data);
+ break;
+ case MEMCACHED_BEHAVIOR_CACHE_LOOKUPS:
+ set_behavior_flag(ptr, MEM_USE_CACHE_LOOKUPS, data);
+ memcached_quit(ptr);
break;
- case MEMCACHED_BEHAVIOR_MD5_HASHING:
- set_behavior_flag(ptr, MEM_USE_MD5, data);
+ case MEMCACHED_BEHAVIOR_VERIFY_KEY:
+ set_behavior_flag(ptr, MEM_VERIFY_KEY, data);
break;
case MEMCACHED_BEHAVIOR_KETAMA:
set_behavior_flag(ptr, MEM_USE_KETAMA, data);
break;
+ case MEMCACHED_BEHAVIOR_SORT_HOSTS:
+ set_behavior_flag(ptr, MEM_USE_SORT_HOSTS, data);
+ break;
+ case MEMCACHED_BEHAVIOR_POLL_TIMEOUT:
+ {
+ int32_t timeout= (int32_t)data;
+
+ ptr->poll_timeout= timeout;
+ break;
+ }
+ case MEMCACHED_BEHAVIOR_CONNECT_TIMEOUT:
+ {
+ int32_t timeout= (int32_t)data;
+
+ ptr->connect_timeout= timeout;
+ break;
+ }
+ case MEMCACHED_BEHAVIOR_SOCKET_SEND_SIZE:
+ {
+ ptr->send_size= data;
+ memcached_quit(ptr);
+ break;
+ }
+ case MEMCACHED_BEHAVIOR_SOCKET_RECV_SIZE:
+ {
+ ptr->recv_size= data;
+ memcached_quit(ptr);
+ break;
+ }
+
+
}
return MEMCACHED_SUCCESS;
}
-unsigned long long memcached_behavior_get(memcached_st *ptr,
- memcached_behavior flag)
+uint64_t memcached_behavior_get(memcached_st *ptr,
+ memcached_behavior flag)
{
- memcached_flags temp_flag;
+ memcached_flags temp_flag= 0;
switch (flag)
{
+ case MEMCACHED_BEHAVIOR_REPLICAS:
+ return (unsigned long long)ptr->number_of_replicas;
+ case MEMCACHED_BEHAVIOR_SUPPORT_CAS:
+ temp_flag= MEM_SUPPORT_CAS;
+ break;
+ case MEMCACHED_BEHAVIOR_CACHE_LOOKUPS:
+ temp_flag= MEM_USE_CACHE_LOOKUPS;
+ break;
case MEMCACHED_BEHAVIOR_NO_BLOCK:
temp_flag= MEM_NO_BLOCK;
break;
+ case MEMCACHED_BEHAVIOR_BUFFER_REQUESTS:
+ temp_flag= MEM_BUFFER_REQUESTS;
+ break;
case MEMCACHED_BEHAVIOR_TCP_NODELAY:
temp_flag= MEM_TCP_NODELAY;
break;
- case MEMCACHED_BEHAVIOR_MD5_HASHING:
- temp_flag= MEM_USE_MD5;
+ case MEMCACHED_BEHAVIOR_VERIFY_KEY:
+ temp_flag= MEM_VERIFY_KEY;
break;
+ case MEMCACHED_BEHAVIOR_DISTRIBUTION:
+ return ptr->distribution;
+ case MEMCACHED_BEHAVIOR_HASH:
+ return ptr->hash;
case MEMCACHED_BEHAVIOR_KETAMA:
temp_flag= MEM_USE_KETAMA;
break;
+ case MEMCACHED_BEHAVIOR_SORT_HOSTS:
+ temp_flag= MEM_USE_SORT_HOSTS;
+ break;
+ case MEMCACHED_BEHAVIOR_POLL_TIMEOUT:
+ {
+ return (unsigned long long)ptr->poll_timeout;
+ }
+ case MEMCACHED_BEHAVIOR_CONNECT_TIMEOUT:
+ {
+ return (unsigned long long)ptr->connect_timeout;
+ }
+ case MEMCACHED_BEHAVIOR_SOCKET_SEND_SIZE:
+ {
+ int sock_size;
+ socklen_t sock_length= sizeof(int);
+
+ /* REFACTOR */
+ /* We just try the first host, and if it is down we return zero */
+ if ((memcached_connect(&ptr->hosts[0])) != MEMCACHED_SUCCESS)
+ return 0;
+
+ if (getsockopt(ptr->hosts[0].fd, SOL_SOCKET,
+ SO_SNDBUF, &sock_size, &sock_length))
+ return 0; /* Zero means error */
+
+ return sock_size;
+ }
+ case MEMCACHED_BEHAVIOR_SOCKET_RECV_SIZE:
+ {
+ int sock_size;
+ socklen_t sock_length= sizeof(int);
+
+ /* REFACTOR */
+ /* We just try the first host, and if it is down we return zero */
+ if ((memcached_connect(&ptr->hosts[0])) != MEMCACHED_SUCCESS)
+ return 0;
+
+ if (getsockopt(ptr->hosts[0].fd, SOL_SOCKET,
+ SO_RCVBUF, &sock_size, &sock_length))
+ return 0; /* Zero means error */
+
+ return sock_size;
+ }
}
+ WATCHPOINT_ASSERT(temp_flag); /* Programming mistake if it gets this far */
if (ptr->flags & temp_flag)
return 1;
else