3 #include <sys/socket.h>
4 #include <netinet/tcp.h>
7 This function is used to modify the behabior of running client.
9 We quit all connections so we can reset the sockets.
12 void set_behavior_flag(memcached_st
*ptr
, memcached_flags temp_flag
, uint64_t data
)
15 ptr
->flags
|= temp_flag
;
17 ptr
->flags
&= ~temp_flag
;
20 memcached_return
memcached_behavior_set(memcached_st
*ptr
,
21 memcached_behavior flag
,
26 case MEMCACHED_BEHAVIOR_SUPPORT_CAS
:
27 set_behavior_flag(ptr
, MEM_SUPPORT_CAS
, data
);
29 case MEMCACHED_BEHAVIOR_NO_BLOCK
:
30 set_behavior_flag(ptr
, MEM_NO_BLOCK
, data
);
32 case MEMCACHED_BEHAVIOR_BUFFER_REQUESTS
:
33 set_behavior_flag(ptr
, MEM_BUFFER_REQUESTS
, data
);
36 case MEMCACHED_BEHAVIOR_TCP_NODELAY
:
37 set_behavior_flag(ptr
, MEM_TCP_NODELAY
, data
);
40 case MEMCACHED_BEHAVIOR_DISTRIBUTION
:
42 ptr
->distribution
= (memcached_server_distribution
)(data
);
43 run_distribution(ptr
);
46 case MEMCACHED_BEHAVIOR_KETAMA
:
49 ptr
->hash
= MEMCACHED_HASH_MD5
;
50 ptr
->distribution
= MEMCACHED_DISTRIBUTION_CONSISTENT_KETAMA
;
58 case MEMCACHED_BEHAVIOR_HASH
:
59 ptr
->hash
= (memcached_hash
)(data
);
61 case MEMCACHED_BEHAVIOR_CACHE_LOOKUPS
:
62 set_behavior_flag(ptr
, MEM_USE_CACHE_LOOKUPS
, data
);
65 case MEMCACHED_BEHAVIOR_VERIFY_KEY
:
66 set_behavior_flag(ptr
, MEM_VERIFY_KEY
, data
);
68 case MEMCACHED_BEHAVIOR_SORT_HOSTS
:
70 set_behavior_flag(ptr
, MEM_USE_SORT_HOSTS
, data
);
71 run_distribution(ptr
);
75 case MEMCACHED_BEHAVIOR_POLL_TIMEOUT
:
76 ptr
->poll_timeout
= (int32_t)data
;
78 case MEMCACHED_BEHAVIOR_CONNECT_TIMEOUT
:
79 ptr
->connect_timeout
= (int32_t)data
;
81 case MEMCACHED_BEHAVIOR_RETRY_TIMEOUT
:
82 ptr
->retry_timeout
= (int32_t)data
;
84 case MEMCACHED_BEHAVIOR_SOCKET_SEND_SIZE
:
85 ptr
->send_size
= (int32_t)data
;
88 case MEMCACHED_BEHAVIOR_SOCKET_RECV_SIZE
:
89 ptr
->recv_size
= (int32_t)data
;
92 case MEMCACHED_BEHAVIOR_USER_DATA
:
93 return MEMCACHED_FAILURE
;
96 return MEMCACHED_SUCCESS
;
99 uint64_t memcached_behavior_get(memcached_st
*ptr
,
100 memcached_behavior flag
)
102 memcached_flags temp_flag
= 0;
106 case MEMCACHED_BEHAVIOR_SUPPORT_CAS
:
107 temp_flag
= MEM_SUPPORT_CAS
;
109 case MEMCACHED_BEHAVIOR_CACHE_LOOKUPS
:
110 temp_flag
= MEM_USE_CACHE_LOOKUPS
;
112 case MEMCACHED_BEHAVIOR_NO_BLOCK
:
113 temp_flag
= MEM_NO_BLOCK
;
115 case MEMCACHED_BEHAVIOR_BUFFER_REQUESTS
:
116 temp_flag
= MEM_BUFFER_REQUESTS
;
118 case MEMCACHED_BEHAVIOR_TCP_NODELAY
:
119 temp_flag
= MEM_TCP_NODELAY
;
121 case MEMCACHED_BEHAVIOR_VERIFY_KEY
:
122 temp_flag
= MEM_VERIFY_KEY
;
124 case MEMCACHED_BEHAVIOR_DISTRIBUTION
:
125 return ptr
->distribution
;
126 case MEMCACHED_BEHAVIOR_KETAMA
:
127 return (ptr
->distribution
== MEMCACHED_DISTRIBUTION_CONSISTENT_KETAMA
&& ptr
->hash
== MEMCACHED_HASH_MD5
) ? 1 : 0;
128 case MEMCACHED_BEHAVIOR_HASH
:
130 case MEMCACHED_BEHAVIOR_SORT_HOSTS
:
131 temp_flag
= MEM_USE_SORT_HOSTS
;
133 case MEMCACHED_BEHAVIOR_POLL_TIMEOUT
:
135 return (unsigned long long)ptr
->poll_timeout
;
137 case MEMCACHED_BEHAVIOR_CONNECT_TIMEOUT
:
139 return (unsigned long long)ptr
->connect_timeout
;
141 case MEMCACHED_BEHAVIOR_RETRY_TIMEOUT
:
143 return (unsigned long long)ptr
->retry_timeout
;
145 case MEMCACHED_BEHAVIOR_SOCKET_SEND_SIZE
:
148 socklen_t sock_length
= sizeof(int);
151 /* We just try the first host, and if it is down we return zero */
152 if ((memcached_connect(&ptr
->hosts
[0])) != MEMCACHED_SUCCESS
)
155 if (getsockopt(ptr
->hosts
[0].fd
, SOL_SOCKET
,
156 SO_SNDBUF
, &sock_size
, &sock_length
))
157 return 0; /* Zero means error */
161 case MEMCACHED_BEHAVIOR_SOCKET_RECV_SIZE
:
164 socklen_t sock_length
= sizeof(int);
167 /* We just try the first host, and if it is down we return zero */
168 if ((memcached_connect(&ptr
->hosts
[0])) != MEMCACHED_SUCCESS
)
171 if (getsockopt(ptr
->hosts
[0].fd
, SOL_SOCKET
,
172 SO_RCVBUF
, &sock_size
, &sock_length
))
173 return 0; /* Zero means error */
177 case MEMCACHED_BEHAVIOR_USER_DATA
:
178 return MEMCACHED_FAILURE
;
181 WATCHPOINT_ASSERT(temp_flag
); /* Programming mistake if it gets this far */
182 if (ptr
->flags
& temp_flag
)
187 return MEMCACHED_SUCCESS
;