We quit all connections so we can reset the sockets.
*/
+void set_behavior_flag(memcached_st *ptr, memcached_flags temp_flag, void *data)
+{
+ unsigned int *truefalse= (unsigned int *)data;
+
+ memcached_quit(ptr);
+ if (truefalse)
+ ptr->flags|= temp_flag;
+ else
+ ptr->flags+= temp_flag;
+}
+
memcached_return memcached_behavior_set(memcached_st *ptr,
memcached_behavior flag,
void *data)
{
- memcached_flags temp_flag;
-
switch (flag)
{
case MEMCACHED_BEHAVIOR_NO_BLOCK:
- temp_flag= MEM_NO_BLOCK;
+ set_behavior_flag(ptr, MEM_NO_BLOCK, data);
+ break;
case MEMCACHED_BEHAVIOR_TCP_NODELAY:
- temp_flag= MEM_TCP_NODELAY;
+ set_behavior_flag(ptr, MEM_TCP_NODELAY, data);
+ break;
case MEMCACHED_BEHAVIOR_MD5_HASHING:
- temp_flag= MEM_USE_MD5;
+ set_behavior_flag(ptr, MEM_USE_MD5, data);
+ break;
case MEMCACHED_BEHAVIOR_KETAMA:
- temp_flag= MEM_USE_KETAMA;
+ set_behavior_flag(ptr, MEM_USE_KETAMA, data);
+ break;
+ case MEMCACHED_BEHAVIOR_SOCKET_SEND_SIZE:
+ {
+ ptr->send_size= (*((int *)data));
+ memcached_quit(ptr);
+ break;
+ }
+ case MEMCACHED_BEHAVIOR_SOCKET_RECV_SIZE:
{
- unsigned int *truefalse= (unsigned int *)data;
+ ptr->recv_size= (*((int *)data));
memcached_quit(ptr);
- if (truefalse)
- ptr->flags|= temp_flag;
- else
- ptr->flags+= temp_flag;
break;
}
+
+
}
return MEMCACHED_SUCCESS;
unsigned long long memcached_behavior_get(memcached_st *ptr,
memcached_behavior flag)
{
- memcached_flags temp_flag;
+ memcached_flags temp_flag= 0;
switch (flag)
{
case MEMCACHED_BEHAVIOR_NO_BLOCK:
temp_flag= MEM_NO_BLOCK;
+ break;
case MEMCACHED_BEHAVIOR_TCP_NODELAY:
temp_flag= MEM_TCP_NODELAY;
+ break;
case MEMCACHED_BEHAVIOR_MD5_HASHING:
temp_flag= MEM_USE_MD5;
+ break;
case MEMCACHED_BEHAVIOR_KETAMA:
temp_flag= MEM_USE_KETAMA;
- if (ptr->flags & temp_flag)
- return 1;
- else
- return 0;
+ break;
+ case MEMCACHED_BEHAVIOR_SOCKET_SEND_SIZE:
+ {
+ int sock_size;
+ socklen_t sock_length= sizeof(int);
+
+ /* We just try the first host, and if it is down we return zero */
+ if ((memcached_connect(ptr, 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);
+
+ /* We just try the first host, and if it is down we return zero */
+ if ((memcached_connect(ptr, 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
+ return 0;
+
return MEMCACHED_SUCCESS;
}