X-Git-Url: https://git.m6w6.name/?a=blobdiff_plain;f=lib%2Fmemcached_behavior.c;h=67a52fa8ac212c18bc9a37cbced1f1abe728aa06;hb=8d0cd77fc7bafd1f353bc7c50910b751766f931a;hp=1ce495afe336ee27e0d28ef78325afe82e6450f3;hpb=afcf6c1a1c42ed21d50fb17fc1ba6589fddb5030;p=m6w6%2Flibmemcached diff --git a/lib/memcached_behavior.c b/lib/memcached_behavior.c index 1ce495af..67a52fa8 100644 --- a/lib/memcached_behavior.c +++ b/lib/memcached_behavior.c @@ -32,12 +32,29 @@ memcached_return memcached_behavior_set(memcached_st *ptr, case MEMCACHED_BEHAVIOR_TCP_NODELAY: set_behavior_flag(ptr, MEM_TCP_NODELAY, data); break; - case MEMCACHED_BEHAVIOR_MD5_HASHING: - set_behavior_flag(ptr, MEM_USE_MD5, data); + case MEMCACHED_BEHAVIOR_HASH: + ptr->hash= *(memcached_hash *)(data); + break; + case MEMCACHED_BEHAVIOR_CACHE_LOOKUPS: + set_behavior_flag(ptr, MEM_USE_CACHE_LOOKUPS, data); break; case MEMCACHED_BEHAVIOR_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: + { + ptr->recv_size= (*((int *)data)); + memcached_quit(ptr); + break; + } + + } return MEMCACHED_SUCCESS; @@ -46,24 +63,57 @@ memcached_return memcached_behavior_set(memcached_st *ptr, 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_CACHE_LOOKUPS: + temp_flag= MEM_USE_CACHE_LOOKUPS; + break; 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_HASH: + return ptr->hash; case MEMCACHED_BEHAVIOR_KETAMA: temp_flag= MEM_USE_KETAMA; 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