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_SND_TIMEOUT
:
27 ptr
->snd_timeout
= (int32_t)data
;
29 case MEMCACHED_BEHAVIOR_RCV_TIMEOUT
:
30 ptr
->rcv_timeout
= (int32_t)data
;
32 case MEMCACHED_BEHAVIOR_SERVER_FAILURE_LIMIT
:
33 ptr
->server_failure_limit
= (uint32_t)data
;
35 case MEMCACHED_BEHAVIOR_BINARY_PROTOCOL
:
36 set_behavior_flag(ptr
, MEM_BINARY_PROTOCOL
, data
);
38 case MEMCACHED_BEHAVIOR_SUPPORT_CAS
:
39 set_behavior_flag(ptr
, MEM_SUPPORT_CAS
, data
);
41 case MEMCACHED_BEHAVIOR_NO_BLOCK
:
42 set_behavior_flag(ptr
, MEM_NO_BLOCK
, data
);
44 case MEMCACHED_BEHAVIOR_BUFFER_REQUESTS
:
45 set_behavior_flag(ptr
, MEM_BUFFER_REQUESTS
, data
);
48 case MEMCACHED_BEHAVIOR_TCP_NODELAY
:
49 set_behavior_flag(ptr
, MEM_TCP_NODELAY
, data
);
52 case MEMCACHED_BEHAVIOR_DISTRIBUTION
:
54 ptr
->distribution
= (memcached_server_distribution
)(data
);
55 run_distribution(ptr
);
58 case MEMCACHED_BEHAVIOR_KETAMA
:
62 ptr
->hash
= MEMCACHED_HASH_MD5
;
63 ptr
->distribution
= MEMCACHED_DISTRIBUTION_CONSISTENT_KETAMA
;
70 run_distribution(ptr
);
73 case MEMCACHED_BEHAVIOR_KETAMA_WEIGHTED
:
75 ptr
->hash
= MEMCACHED_HASH_MD5
;
76 ptr
->distribution
= MEMCACHED_DISTRIBUTION_CONSISTENT_KETAMA
;
77 set_behavior_flag(ptr
, MEM_KETAMA_WEIGHTED
, data
);
78 run_distribution(ptr
);
81 case MEMCACHED_BEHAVIOR_HASH
:
82 ptr
->hash
= (memcached_hash
)(data
);
84 case MEMCACHED_BEHAVIOR_KETAMA_HASH
:
85 ptr
->hash_continuum
= (memcached_hash
)(data
);
86 run_distribution(ptr
);
88 case MEMCACHED_BEHAVIOR_CACHE_LOOKUPS
:
89 set_behavior_flag(ptr
, MEM_USE_CACHE_LOOKUPS
, data
);
92 case MEMCACHED_BEHAVIOR_VERIFY_KEY
:
93 set_behavior_flag(ptr
, MEM_VERIFY_KEY
, data
);
95 case MEMCACHED_BEHAVIOR_SORT_HOSTS
:
97 set_behavior_flag(ptr
, MEM_USE_SORT_HOSTS
, data
);
98 run_distribution(ptr
);
102 case MEMCACHED_BEHAVIOR_POLL_TIMEOUT
:
103 ptr
->poll_timeout
= (int32_t)data
;
105 case MEMCACHED_BEHAVIOR_CONNECT_TIMEOUT
:
106 ptr
->connect_timeout
= (int32_t)data
;
108 case MEMCACHED_BEHAVIOR_RETRY_TIMEOUT
:
109 ptr
->retry_timeout
= (int32_t)data
;
111 case MEMCACHED_BEHAVIOR_SOCKET_SEND_SIZE
:
112 ptr
->send_size
= (int32_t)data
;
115 case MEMCACHED_BEHAVIOR_SOCKET_RECV_SIZE
:
116 ptr
->recv_size
= (int32_t)data
;
119 case MEMCACHED_BEHAVIOR_USER_DATA
:
120 return MEMCACHED_FAILURE
;
123 return MEMCACHED_SUCCESS
;
126 uint64_t memcached_behavior_get(memcached_st
*ptr
,
127 memcached_behavior flag
)
129 memcached_flags temp_flag
= 0;
133 case MEMCACHED_BEHAVIOR_BINARY_PROTOCOL
:
134 temp_flag
= MEM_BINARY_PROTOCOL
;
136 case MEMCACHED_BEHAVIOR_SUPPORT_CAS
:
137 temp_flag
= MEM_SUPPORT_CAS
;
139 case MEMCACHED_BEHAVIOR_CACHE_LOOKUPS
:
140 temp_flag
= MEM_USE_CACHE_LOOKUPS
;
142 case MEMCACHED_BEHAVIOR_NO_BLOCK
:
143 temp_flag
= MEM_NO_BLOCK
;
145 case MEMCACHED_BEHAVIOR_BUFFER_REQUESTS
:
146 temp_flag
= MEM_BUFFER_REQUESTS
;
148 case MEMCACHED_BEHAVIOR_TCP_NODELAY
:
149 temp_flag
= MEM_TCP_NODELAY
;
151 case MEMCACHED_BEHAVIOR_VERIFY_KEY
:
152 temp_flag
= MEM_VERIFY_KEY
;
154 case MEMCACHED_BEHAVIOR_KETAMA_WEIGHTED
:
155 temp_flag
= MEM_KETAMA_WEIGHTED
;
157 case MEMCACHED_BEHAVIOR_DISTRIBUTION
:
158 return ptr
->distribution
;
159 case MEMCACHED_BEHAVIOR_KETAMA
:
160 return (ptr
->distribution
== MEMCACHED_DISTRIBUTION_CONSISTENT_KETAMA
) ? 1 : 0;
161 case MEMCACHED_BEHAVIOR_HASH
:
163 case MEMCACHED_BEHAVIOR_KETAMA_HASH
:
164 return ptr
->hash_continuum
;
165 case MEMCACHED_BEHAVIOR_SERVER_FAILURE_LIMIT
:
166 return ptr
->server_failure_limit
;
167 case MEMCACHED_BEHAVIOR_SORT_HOSTS
:
168 temp_flag
= MEM_USE_SORT_HOSTS
;
170 case MEMCACHED_BEHAVIOR_POLL_TIMEOUT
:
172 return (unsigned long long)ptr
->poll_timeout
;
174 case MEMCACHED_BEHAVIOR_CONNECT_TIMEOUT
:
176 return (unsigned long long)ptr
->connect_timeout
;
178 case MEMCACHED_BEHAVIOR_RETRY_TIMEOUT
:
180 return (unsigned long long)ptr
->retry_timeout
;
182 case MEMCACHED_BEHAVIOR_SOCKET_SEND_SIZE
:
185 socklen_t sock_length
= sizeof(int);
188 /* We just try the first host, and if it is down we return zero */
189 if ((memcached_connect(&ptr
->hosts
[0])) != MEMCACHED_SUCCESS
)
192 if (getsockopt(ptr
->hosts
[0].fd
, SOL_SOCKET
,
193 SO_SNDBUF
, &sock_size
, &sock_length
))
194 return 0; /* Zero means error */
198 case MEMCACHED_BEHAVIOR_SOCKET_RECV_SIZE
:
201 socklen_t sock_length
= sizeof(int);
204 /* We just try the first host, and if it is down we return zero */
205 if ((memcached_connect(&ptr
->hosts
[0])) != MEMCACHED_SUCCESS
)
208 if (getsockopt(ptr
->hosts
[0].fd
, SOL_SOCKET
,
209 SO_RCVBUF
, &sock_size
, &sock_length
))
210 return 0; /* Zero means error */
214 case MEMCACHED_BEHAVIOR_USER_DATA
:
215 return MEMCACHED_FAILURE
;
218 WATCHPOINT_ASSERT(temp_flag
); /* Programming mistake if it gets this far */
219 if (ptr
->flags
& temp_flag
)
224 return MEMCACHED_SUCCESS
;