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_REPLICAS
:
28 uint8_t number_of_replicas
= (uint8_t)data
;
30 if (number_of_replicas
> ptr
->number_of_hosts
|| number_of_replicas
== 0 || number_of_replicas
> MEMCACHED_MAX_REPLICAS
)
31 return MEMCACHED_FAILURE
;
33 ptr
->number_of_replicas
= number_of_replicas
;
36 case MEMCACHED_BEHAVIOR_SUPPORT_CAS
:
37 set_behavior_flag(ptr
, MEM_SUPPORT_CAS
, data
);
39 case MEMCACHED_BEHAVIOR_NO_BLOCK
:
40 set_behavior_flag(ptr
, MEM_NO_BLOCK
, data
);
42 case MEMCACHED_BEHAVIOR_BUFFER_REQUESTS
:
43 set_behavior_flag(ptr
, MEM_BUFFER_REQUESTS
, data
);
46 case MEMCACHED_BEHAVIOR_TCP_NODELAY
:
47 set_behavior_flag(ptr
, MEM_TCP_NODELAY
, data
);
50 case MEMCACHED_BEHAVIOR_DISTRIBUTION
:
51 ptr
->distribution
= (memcached_server_distribution
)data
;
53 case MEMCACHED_BEHAVIOR_HASH
:
54 ptr
->hash
= (memcached_hash
)(data
);
56 case MEMCACHED_BEHAVIOR_CACHE_LOOKUPS
:
57 set_behavior_flag(ptr
, MEM_USE_CACHE_LOOKUPS
, data
);
60 case MEMCACHED_BEHAVIOR_VERIFY_KEY
:
61 set_behavior_flag(ptr
, MEM_VERIFY_KEY
, data
);
63 case MEMCACHED_BEHAVIOR_KETAMA
:
64 set_behavior_flag(ptr
, MEM_USE_KETAMA
, data
);
66 case MEMCACHED_BEHAVIOR_SORT_HOSTS
:
67 set_behavior_flag(ptr
, MEM_USE_SORT_HOSTS
, data
);
69 case MEMCACHED_BEHAVIOR_POLL_TIMEOUT
:
71 int32_t timeout
= (int32_t)data
;
73 ptr
->poll_timeout
= timeout
;
76 case MEMCACHED_BEHAVIOR_CONNECT_TIMEOUT
:
78 int32_t timeout
= (int32_t)data
;
80 ptr
->connect_timeout
= timeout
;
83 case MEMCACHED_BEHAVIOR_SOCKET_SEND_SIZE
:
89 case MEMCACHED_BEHAVIOR_SOCKET_RECV_SIZE
:
99 return MEMCACHED_SUCCESS
;
102 uint64_t memcached_behavior_get(memcached_st
*ptr
,
103 memcached_behavior flag
)
105 memcached_flags temp_flag
= 0;
109 case MEMCACHED_BEHAVIOR_REPLICAS
:
110 return (unsigned long long)ptr
->number_of_replicas
;
111 case MEMCACHED_BEHAVIOR_SUPPORT_CAS
:
112 temp_flag
= MEM_SUPPORT_CAS
;
114 case MEMCACHED_BEHAVIOR_CACHE_LOOKUPS
:
115 temp_flag
= MEM_USE_CACHE_LOOKUPS
;
117 case MEMCACHED_BEHAVIOR_NO_BLOCK
:
118 temp_flag
= MEM_NO_BLOCK
;
120 case MEMCACHED_BEHAVIOR_BUFFER_REQUESTS
:
121 temp_flag
= MEM_BUFFER_REQUESTS
;
123 case MEMCACHED_BEHAVIOR_TCP_NODELAY
:
124 temp_flag
= MEM_TCP_NODELAY
;
126 case MEMCACHED_BEHAVIOR_VERIFY_KEY
:
127 temp_flag
= MEM_VERIFY_KEY
;
129 case MEMCACHED_BEHAVIOR_DISTRIBUTION
:
130 return ptr
->distribution
;
131 case MEMCACHED_BEHAVIOR_HASH
:
133 case MEMCACHED_BEHAVIOR_KETAMA
:
134 temp_flag
= MEM_USE_KETAMA
;
136 case MEMCACHED_BEHAVIOR_SORT_HOSTS
:
137 temp_flag
= MEM_USE_SORT_HOSTS
;
139 case MEMCACHED_BEHAVIOR_POLL_TIMEOUT
:
141 return (unsigned long long)ptr
->poll_timeout
;
143 case MEMCACHED_BEHAVIOR_CONNECT_TIMEOUT
:
145 return (unsigned long long)ptr
->connect_timeout
;
147 case MEMCACHED_BEHAVIOR_SOCKET_SEND_SIZE
:
150 socklen_t sock_length
= sizeof(int);
153 /* We just try the first host, and if it is down we return zero */
154 if ((memcached_connect(&ptr
->hosts
[0])) != MEMCACHED_SUCCESS
)
157 if (getsockopt(ptr
->hosts
[0].fd
, SOL_SOCKET
,
158 SO_SNDBUF
, &sock_size
, &sock_length
))
159 return 0; /* Zero means error */
163 case MEMCACHED_BEHAVIOR_SOCKET_RECV_SIZE
:
166 socklen_t sock_length
= sizeof(int);
169 /* We just try the first host, and if it is down we return zero */
170 if ((memcached_connect(&ptr
->hosts
[0])) != MEMCACHED_SUCCESS
)
173 if (getsockopt(ptr
->hosts
[0].fd
, SOL_SOCKET
,
174 SO_RCVBUF
, &sock_size
, &sock_length
))
175 return 0; /* Zero means error */
181 WATCHPOINT_ASSERT(temp_flag
); /* Programming mistake if it gets this far */
182 if (ptr
->flags
& temp_flag
)
187 return MEMCACHED_SUCCESS
;