4 #include <sys/socket.h>
5 #include <netinet/tcp.h>
8 This function is used to modify the behabior of running client.
10 We quit all connections so we can reset the sockets.
13 static void set_behavior_flag(memcached_st
*ptr
, memcached_flags temp_flag
, uint64_t data
)
16 ptr
->flags
|= temp_flag
;
18 ptr
->flags
&= ~temp_flag
;
21 memcached_return
memcached_behavior_set(memcached_st
*ptr
,
22 memcached_behavior flag
,
27 case MEMCACHED_BEHAVIOR_IO_MSG_WATERMARK
:
28 ptr
->io_msg_watermark
= (int32_t)data
;
30 case MEMCACHED_BEHAVIOR_IO_BYTES_WATERMARK
:
31 ptr
->io_bytes_watermark
= (int32_t)data
;
33 case MEMCACHED_BEHAVIOR_SND_TIMEOUT
:
34 ptr
->snd_timeout
= (int32_t)data
;
36 case MEMCACHED_BEHAVIOR_RCV_TIMEOUT
:
37 ptr
->rcv_timeout
= (int32_t)data
;
39 case MEMCACHED_BEHAVIOR_SERVER_FAILURE_LIMIT
:
40 ptr
->server_failure_limit
= (uint32_t)data
;
42 case MEMCACHED_BEHAVIOR_BINARY_PROTOCOL
:
43 set_behavior_flag(ptr
, MEM_BINARY_PROTOCOL
, data
);
45 case MEMCACHED_BEHAVIOR_SUPPORT_CAS
:
46 set_behavior_flag(ptr
, MEM_SUPPORT_CAS
, data
);
48 case MEMCACHED_BEHAVIOR_NO_BLOCK
:
49 set_behavior_flag(ptr
, MEM_NO_BLOCK
, data
);
51 case MEMCACHED_BEHAVIOR_BUFFER_REQUESTS
:
52 set_behavior_flag(ptr
, MEM_BUFFER_REQUESTS
, data
);
55 case MEMCACHED_BEHAVIOR_TCP_NODELAY
:
56 set_behavior_flag(ptr
, MEM_TCP_NODELAY
, data
);
59 case MEMCACHED_BEHAVIOR_DISTRIBUTION
:
61 ptr
->distribution
= (memcached_server_distribution
)(data
);
62 if (ptr
->distribution
== MEMCACHED_DISTRIBUTION_RANDOM
)
66 run_distribution(ptr
);
69 case MEMCACHED_BEHAVIOR_KETAMA
:
73 ptr
->hash
= MEMCACHED_HASH_MD5
;
74 ptr
->distribution
= MEMCACHED_DISTRIBUTION_CONSISTENT_KETAMA
;
81 run_distribution(ptr
);
84 case MEMCACHED_BEHAVIOR_KETAMA_WEIGHTED
:
86 ptr
->hash
= MEMCACHED_HASH_MD5
;
87 ptr
->distribution
= MEMCACHED_DISTRIBUTION_CONSISTENT_KETAMA
;
88 set_behavior_flag(ptr
, MEM_KETAMA_WEIGHTED
, data
);
89 run_distribution(ptr
);
92 case MEMCACHED_BEHAVIOR_HASH
:
93 ptr
->hash
= (memcached_hash
)(data
);
95 case MEMCACHED_BEHAVIOR_KETAMA_HASH
:
96 ptr
->hash_continuum
= (memcached_hash
)(data
);
97 run_distribution(ptr
);
99 case MEMCACHED_BEHAVIOR_CACHE_LOOKUPS
:
100 set_behavior_flag(ptr
, MEM_USE_CACHE_LOOKUPS
, data
);
103 case MEMCACHED_BEHAVIOR_VERIFY_KEY
:
104 set_behavior_flag(ptr
, MEM_VERIFY_KEY
, data
);
106 case MEMCACHED_BEHAVIOR_SORT_HOSTS
:
108 set_behavior_flag(ptr
, MEM_USE_SORT_HOSTS
, data
);
109 run_distribution(ptr
);
113 case MEMCACHED_BEHAVIOR_POLL_TIMEOUT
:
114 ptr
->poll_timeout
= (int32_t)data
;
116 case MEMCACHED_BEHAVIOR_CONNECT_TIMEOUT
:
117 ptr
->connect_timeout
= (int32_t)data
;
119 case MEMCACHED_BEHAVIOR_RETRY_TIMEOUT
:
120 ptr
->retry_timeout
= (int32_t)data
;
122 case MEMCACHED_BEHAVIOR_SOCKET_SEND_SIZE
:
123 ptr
->send_size
= (int32_t)data
;
126 case MEMCACHED_BEHAVIOR_SOCKET_RECV_SIZE
:
127 ptr
->recv_size
= (int32_t)data
;
130 case MEMCACHED_BEHAVIOR_USER_DATA
:
131 return MEMCACHED_FAILURE
;
132 case MEMCACHED_BEHAVIOR_HASH_WITH_PREFIX_KEY
:
133 set_behavior_flag(ptr
, MEM_HASH_WITH_PREFIX_KEY
, data
);
137 return MEMCACHED_SUCCESS
;
140 uint64_t memcached_behavior_get(memcached_st
*ptr
,
141 memcached_behavior flag
)
143 memcached_flags temp_flag
= 0;
147 case MEMCACHED_BEHAVIOR_IO_MSG_WATERMARK
:
148 temp_flag
= ptr
->io_msg_watermark
;
150 case MEMCACHED_BEHAVIOR_IO_BYTES_WATERMARK
:
151 temp_flag
= ptr
->io_bytes_watermark
;
153 case MEMCACHED_BEHAVIOR_BINARY_PROTOCOL
:
154 temp_flag
= MEM_BINARY_PROTOCOL
;
156 case MEMCACHED_BEHAVIOR_SUPPORT_CAS
:
157 temp_flag
= MEM_SUPPORT_CAS
;
159 case MEMCACHED_BEHAVIOR_CACHE_LOOKUPS
:
160 temp_flag
= MEM_USE_CACHE_LOOKUPS
;
162 case MEMCACHED_BEHAVIOR_NO_BLOCK
:
163 temp_flag
= MEM_NO_BLOCK
;
165 case MEMCACHED_BEHAVIOR_BUFFER_REQUESTS
:
166 temp_flag
= MEM_BUFFER_REQUESTS
;
168 case MEMCACHED_BEHAVIOR_TCP_NODELAY
:
169 temp_flag
= MEM_TCP_NODELAY
;
171 case MEMCACHED_BEHAVIOR_VERIFY_KEY
:
172 temp_flag
= MEM_VERIFY_KEY
;
174 case MEMCACHED_BEHAVIOR_KETAMA_WEIGHTED
:
175 temp_flag
= MEM_KETAMA_WEIGHTED
;
177 case MEMCACHED_BEHAVIOR_DISTRIBUTION
:
178 return ptr
->distribution
;
179 case MEMCACHED_BEHAVIOR_KETAMA
:
180 return (ptr
->distribution
== MEMCACHED_DISTRIBUTION_CONSISTENT_KETAMA
) ? 1 : 0;
181 case MEMCACHED_BEHAVIOR_HASH
:
183 case MEMCACHED_BEHAVIOR_KETAMA_HASH
:
184 return ptr
->hash_continuum
;
185 case MEMCACHED_BEHAVIOR_SERVER_FAILURE_LIMIT
:
186 return ptr
->server_failure_limit
;
187 case MEMCACHED_BEHAVIOR_SORT_HOSTS
:
188 temp_flag
= MEM_USE_SORT_HOSTS
;
190 case MEMCACHED_BEHAVIOR_POLL_TIMEOUT
:
192 return (uint64_t)ptr
->poll_timeout
;
194 case MEMCACHED_BEHAVIOR_CONNECT_TIMEOUT
:
196 return (uint64_t)ptr
->connect_timeout
;
198 case MEMCACHED_BEHAVIOR_RETRY_TIMEOUT
:
200 return (uint64_t)ptr
->retry_timeout
;
202 case MEMCACHED_BEHAVIOR_SND_TIMEOUT
:
203 return (uint64_t)ptr
->snd_timeout
;
204 case MEMCACHED_BEHAVIOR_RCV_TIMEOUT
:
205 return (uint64_t)ptr
->rcv_timeout
;
206 case MEMCACHED_BEHAVIOR_SOCKET_SEND_SIZE
:
209 socklen_t sock_length
= sizeof(int);
212 /* We just try the first host, and if it is down we return zero */
213 if ((memcached_connect(&ptr
->hosts
[0])) != MEMCACHED_SUCCESS
)
216 if (getsockopt(ptr
->hosts
[0].fd
, SOL_SOCKET
,
217 SO_SNDBUF
, &sock_size
, &sock_length
))
218 return 0; /* Zero means error */
222 case MEMCACHED_BEHAVIOR_SOCKET_RECV_SIZE
:
225 socklen_t sock_length
= sizeof(int);
228 /* We just try the first host, and if it is down we return zero */
229 if ((memcached_connect(&ptr
->hosts
[0])) != MEMCACHED_SUCCESS
)
232 if (getsockopt(ptr
->hosts
[0].fd
, SOL_SOCKET
,
233 SO_RCVBUF
, &sock_size
, &sock_length
))
234 return 0; /* Zero means error */
238 case MEMCACHED_BEHAVIOR_USER_DATA
:
239 return MEMCACHED_FAILURE
;
240 case MEMCACHED_BEHAVIOR_HASH_WITH_PREFIX_KEY
:
241 temp_flag
= MEM_HASH_WITH_PREFIX_KEY
;
245 WATCHPOINT_ASSERT(temp_flag
); /* Programming mistake if it gets this far */
246 if (ptr
->flags
& temp_flag
)
251 return MEMCACHED_SUCCESS
;