4 #include <sys/socket.h>
5 #include <netinet/tcp.h>
8 This function is used to modify the behavior 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_IO_KEY_PREFETCH
:
34 ptr
->io_key_prefetch
= (int32_t)data
;
36 case MEMCACHED_BEHAVIOR_SND_TIMEOUT
:
37 ptr
->snd_timeout
= (int32_t)data
;
39 case MEMCACHED_BEHAVIOR_RCV_TIMEOUT
:
40 ptr
->rcv_timeout
= (int32_t)data
;
42 case MEMCACHED_BEHAVIOR_SERVER_FAILURE_LIMIT
:
43 ptr
->server_failure_limit
= (uint32_t)data
;
45 case MEMCACHED_BEHAVIOR_BINARY_PROTOCOL
:
47 set_behavior_flag(ptr
, MEM_VERIFY_KEY
, 0);
48 set_behavior_flag(ptr
, MEM_BINARY_PROTOCOL
, data
);
50 case MEMCACHED_BEHAVIOR_SUPPORT_CAS
:
51 set_behavior_flag(ptr
, MEM_SUPPORT_CAS
, data
);
53 case MEMCACHED_BEHAVIOR_NO_BLOCK
:
54 set_behavior_flag(ptr
, MEM_NO_BLOCK
, data
);
57 case MEMCACHED_BEHAVIOR_BUFFER_REQUESTS
:
58 set_behavior_flag(ptr
, MEM_BUFFER_REQUESTS
, data
);
61 case MEMCACHED_BEHAVIOR_USE_UDP
:
62 if (ptr
->number_of_hosts
)
63 return MEMCACHED_FAILURE
;
64 set_behavior_flag(ptr
, MEM_USE_UDP
, data
);
66 set_behavior_flag(ptr
,MEM_NOREPLY
,data
);
68 case MEMCACHED_BEHAVIOR_TCP_NODELAY
:
69 set_behavior_flag(ptr
, MEM_TCP_NODELAY
, data
);
72 case MEMCACHED_BEHAVIOR_DISTRIBUTION
:
74 ptr
->distribution
= (memcached_server_distribution
)(data
);
75 if (ptr
->distribution
== MEMCACHED_DISTRIBUTION_RANDOM
)
79 run_distribution(ptr
);
82 case MEMCACHED_BEHAVIOR_KETAMA
:
86 ptr
->hash
= MEMCACHED_HASH_MD5
;
87 ptr
->distribution
= MEMCACHED_DISTRIBUTION_CONSISTENT_KETAMA
;
94 run_distribution(ptr
);
97 case MEMCACHED_BEHAVIOR_KETAMA_WEIGHTED
:
99 ptr
->hash
= MEMCACHED_HASH_MD5
;
100 ptr
->distribution
= MEMCACHED_DISTRIBUTION_CONSISTENT_KETAMA
;
101 set_behavior_flag(ptr
, MEM_KETAMA_WEIGHTED
, data
);
102 run_distribution(ptr
);
105 case MEMCACHED_BEHAVIOR_HASH
:
106 #ifndef HAVE_HSIEH_HASH
107 if ((memcached_hash
)(data
) == MEMCACHED_HASH_HSIEH
)
108 return MEMCACHED_FAILURE
;
110 ptr
->hash
= (memcached_hash
)(data
);
112 case MEMCACHED_BEHAVIOR_KETAMA_HASH
:
113 ptr
->hash_continuum
= (memcached_hash
)(data
);
114 run_distribution(ptr
);
116 case MEMCACHED_BEHAVIOR_CACHE_LOOKUPS
:
117 set_behavior_flag(ptr
, MEM_USE_CACHE_LOOKUPS
, data
);
120 case MEMCACHED_BEHAVIOR_VERIFY_KEY
:
121 if (ptr
->flags
& MEM_BINARY_PROTOCOL
)
123 set_behavior_flag(ptr
, MEM_VERIFY_KEY
, data
);
125 case MEMCACHED_BEHAVIOR_SORT_HOSTS
:
127 set_behavior_flag(ptr
, MEM_USE_SORT_HOSTS
, data
);
128 run_distribution(ptr
);
132 case MEMCACHED_BEHAVIOR_POLL_TIMEOUT
:
133 ptr
->poll_timeout
= (int32_t)data
;
135 case MEMCACHED_BEHAVIOR_CONNECT_TIMEOUT
:
136 ptr
->connect_timeout
= (int32_t)data
;
138 case MEMCACHED_BEHAVIOR_RETRY_TIMEOUT
:
139 ptr
->retry_timeout
= (int32_t)data
;
141 case MEMCACHED_BEHAVIOR_SOCKET_SEND_SIZE
:
142 ptr
->send_size
= (int32_t)data
;
145 case MEMCACHED_BEHAVIOR_SOCKET_RECV_SIZE
:
146 ptr
->recv_size
= (int32_t)data
;
149 case MEMCACHED_BEHAVIOR_USER_DATA
:
150 return MEMCACHED_FAILURE
;
151 case MEMCACHED_BEHAVIOR_HASH_WITH_PREFIX_KEY
:
152 set_behavior_flag(ptr
, MEM_HASH_WITH_PREFIX_KEY
, data
);
154 case MEMCACHED_BEHAVIOR_NOREPLY
:
155 set_behavior_flag(ptr
, MEM_NOREPLY
, data
);
157 case MEMCACHED_BEHAVIOR_AUTO_EJECT_HOSTS
:
158 set_behavior_flag(ptr
, MEM_AUTO_EJECT_HOSTS
, data
);
162 return MEMCACHED_SUCCESS
;
165 uint64_t memcached_behavior_get(memcached_st
*ptr
,
166 memcached_behavior flag
)
168 memcached_flags temp_flag
= 0;
172 case MEMCACHED_BEHAVIOR_IO_MSG_WATERMARK
:
173 return ptr
->io_msg_watermark
;
174 case MEMCACHED_BEHAVIOR_IO_BYTES_WATERMARK
:
175 return ptr
->io_bytes_watermark
;
176 case MEMCACHED_BEHAVIOR_IO_KEY_PREFETCH
:
177 return ptr
->io_key_prefetch
;
178 case MEMCACHED_BEHAVIOR_BINARY_PROTOCOL
:
179 temp_flag
= MEM_BINARY_PROTOCOL
;
181 case MEMCACHED_BEHAVIOR_SUPPORT_CAS
:
182 temp_flag
= MEM_SUPPORT_CAS
;
184 case MEMCACHED_BEHAVIOR_CACHE_LOOKUPS
:
185 temp_flag
= MEM_USE_CACHE_LOOKUPS
;
187 case MEMCACHED_BEHAVIOR_NO_BLOCK
:
188 temp_flag
= MEM_NO_BLOCK
;
190 case MEMCACHED_BEHAVIOR_BUFFER_REQUESTS
:
191 temp_flag
= MEM_BUFFER_REQUESTS
;
193 case MEMCACHED_BEHAVIOR_USE_UDP
:
194 temp_flag
= MEM_USE_UDP
;
196 case MEMCACHED_BEHAVIOR_TCP_NODELAY
:
197 temp_flag
= MEM_TCP_NODELAY
;
199 case MEMCACHED_BEHAVIOR_VERIFY_KEY
:
200 temp_flag
= MEM_VERIFY_KEY
;
202 case MEMCACHED_BEHAVIOR_KETAMA_WEIGHTED
:
203 temp_flag
= MEM_KETAMA_WEIGHTED
;
205 case MEMCACHED_BEHAVIOR_DISTRIBUTION
:
206 return ptr
->distribution
;
207 case MEMCACHED_BEHAVIOR_KETAMA
:
208 return (ptr
->distribution
== MEMCACHED_DISTRIBUTION_CONSISTENT_KETAMA
) ? 1 : 0;
209 case MEMCACHED_BEHAVIOR_HASH
:
211 case MEMCACHED_BEHAVIOR_KETAMA_HASH
:
212 return ptr
->hash_continuum
;
213 case MEMCACHED_BEHAVIOR_SERVER_FAILURE_LIMIT
:
214 return ptr
->server_failure_limit
;
215 case MEMCACHED_BEHAVIOR_SORT_HOSTS
:
216 temp_flag
= MEM_USE_SORT_HOSTS
;
218 case MEMCACHED_BEHAVIOR_POLL_TIMEOUT
:
220 return (uint64_t)ptr
->poll_timeout
;
222 case MEMCACHED_BEHAVIOR_CONNECT_TIMEOUT
:
224 return (uint64_t)ptr
->connect_timeout
;
226 case MEMCACHED_BEHAVIOR_RETRY_TIMEOUT
:
228 return (uint64_t)ptr
->retry_timeout
;
230 case MEMCACHED_BEHAVIOR_SND_TIMEOUT
:
231 return (uint64_t)ptr
->snd_timeout
;
232 case MEMCACHED_BEHAVIOR_RCV_TIMEOUT
:
233 return (uint64_t)ptr
->rcv_timeout
;
234 case MEMCACHED_BEHAVIOR_SOCKET_SEND_SIZE
:
237 socklen_t sock_length
= sizeof(int);
240 /* We just try the first host, and if it is down we return zero */
241 if ((memcached_connect(&ptr
->hosts
[0])) != MEMCACHED_SUCCESS
)
244 if (getsockopt(ptr
->hosts
[0].fd
, SOL_SOCKET
,
245 SO_SNDBUF
, &sock_size
, &sock_length
))
246 return 0; /* Zero means error */
250 case MEMCACHED_BEHAVIOR_SOCKET_RECV_SIZE
:
253 socklen_t sock_length
= sizeof(int);
256 /* We just try the first host, and if it is down we return zero */
257 if ((memcached_connect(&ptr
->hosts
[0])) != MEMCACHED_SUCCESS
)
260 if (getsockopt(ptr
->hosts
[0].fd
, SOL_SOCKET
,
261 SO_RCVBUF
, &sock_size
, &sock_length
))
262 return 0; /* Zero means error */
266 case MEMCACHED_BEHAVIOR_USER_DATA
:
267 return MEMCACHED_FAILURE
;
268 case MEMCACHED_BEHAVIOR_HASH_WITH_PREFIX_KEY
:
269 temp_flag
= MEM_HASH_WITH_PREFIX_KEY
;
271 case MEMCACHED_BEHAVIOR_NOREPLY
:
272 temp_flag
= MEM_NOREPLY
;
274 case MEMCACHED_BEHAVIOR_AUTO_EJECT_HOSTS
:
275 temp_flag
= MEM_AUTO_EJECT_HOSTS
;
279 WATCHPOINT_ASSERT(temp_flag
); /* Programming mistake if it gets this far */
280 if (ptr
->flags
& temp_flag
)