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
, void *data
)
17 truefalse
= *(unsigned int *)data
;
22 ptr
->flags
|= temp_flag
;
24 ptr
->flags
+= temp_flag
;
27 memcached_return
memcached_behavior_set(memcached_st
*ptr
,
28 memcached_behavior flag
,
33 case MEMCACHED_BEHAVIOR_REPLICAS
:
35 uint8_t number_of_replicas
= (uint8_t)data
;
37 if (number_of_replicas
> ptr
->number_of_hosts
|| number_of_replicas
== 0 || number_of_replicas
> MEMCACHED_MAX_REPLICAS
)
38 return MEMCACHED_FAILURE
;
40 ptr
->number_of_replicas
= number_of_replicas
;
43 case MEMCACHED_BEHAVIOR_SUPPORT_CAS
:
44 set_behavior_flag(ptr
, MEM_SUPPORT_CAS
, data
);
46 case MEMCACHED_BEHAVIOR_NO_BLOCK
:
47 set_behavior_flag(ptr
, MEM_NO_BLOCK
, data
);
49 case MEMCACHED_BEHAVIOR_BUFFER_REQUESTS
:
50 set_behavior_flag(ptr
, MEM_BUFFER_REQUESTS
, data
);
53 case MEMCACHED_BEHAVIOR_TCP_NODELAY
:
54 set_behavior_flag(ptr
, MEM_TCP_NODELAY
, data
);
57 case MEMCACHED_BEHAVIOR_DISTRIBUTION
:
58 ptr
->distribution
= *(memcached_server_distribution
*)(data
);
60 case MEMCACHED_BEHAVIOR_HASH
:
61 ptr
->hash
= *(memcached_hash
*)(data
);
63 case MEMCACHED_BEHAVIOR_CACHE_LOOKUPS
:
64 set_behavior_flag(ptr
, MEM_USE_CACHE_LOOKUPS
, data
);
67 case MEMCACHED_BEHAVIOR_VERIFY_KEY
:
68 set_behavior_flag(ptr
, MEM_VERIFY_KEY
, data
);
70 case MEMCACHED_BEHAVIOR_KETAMA
:
71 set_behavior_flag(ptr
, MEM_USE_KETAMA
, data
);
73 case MEMCACHED_BEHAVIOR_SORT_HOSTS
:
74 set_behavior_flag(ptr
, MEM_USE_SORT_HOSTS
, data
);
76 case MEMCACHED_BEHAVIOR_USER_DATA
:
79 case MEMCACHED_BEHAVIOR_POLL_TIMEOUT
:
81 int32_t timeout
= (*((int32_t *)data
));
83 ptr
->poll_timeout
= timeout
;
86 case MEMCACHED_BEHAVIOR_CONNECT_TIMEOUT
:
88 int32_t timeout
= (*((int32_t *)data
));
90 ptr
->connect_timeout
= timeout
;
93 case MEMCACHED_BEHAVIOR_SOCKET_SEND_SIZE
:
95 ptr
->send_size
= (*((int *)data
));
99 case MEMCACHED_BEHAVIOR_SOCKET_RECV_SIZE
:
101 ptr
->recv_size
= (*((int *)data
));
109 return MEMCACHED_SUCCESS
;
112 unsigned long long memcached_behavior_get(memcached_st
*ptr
,
113 memcached_behavior flag
)
115 memcached_flags temp_flag
= 0;
119 case MEMCACHED_BEHAVIOR_REPLICAS
:
120 return (unsigned long long)ptr
->number_of_replicas
;
121 case MEMCACHED_BEHAVIOR_SUPPORT_CAS
:
122 temp_flag
= MEM_SUPPORT_CAS
;
124 case MEMCACHED_BEHAVIOR_CACHE_LOOKUPS
:
125 temp_flag
= MEM_USE_CACHE_LOOKUPS
;
127 case MEMCACHED_BEHAVIOR_NO_BLOCK
:
128 temp_flag
= MEM_NO_BLOCK
;
130 case MEMCACHED_BEHAVIOR_BUFFER_REQUESTS
:
131 temp_flag
= MEM_BUFFER_REQUESTS
;
133 case MEMCACHED_BEHAVIOR_TCP_NODELAY
:
134 temp_flag
= MEM_TCP_NODELAY
;
136 case MEMCACHED_BEHAVIOR_VERIFY_KEY
:
137 temp_flag
= MEM_VERIFY_KEY
;
139 case MEMCACHED_BEHAVIOR_DISTRIBUTION
:
140 return ptr
->distribution
;
141 case MEMCACHED_BEHAVIOR_HASH
:
143 case MEMCACHED_BEHAVIOR_KETAMA
:
144 temp_flag
= MEM_USE_KETAMA
;
146 case MEMCACHED_BEHAVIOR_SORT_HOSTS
:
147 temp_flag
= MEM_USE_SORT_HOSTS
;
149 case MEMCACHED_BEHAVIOR_USER_DATA
:
151 case MEMCACHED_BEHAVIOR_POLL_TIMEOUT
:
153 return (unsigned long long)ptr
->poll_timeout
;
155 case MEMCACHED_BEHAVIOR_CONNECT_TIMEOUT
:
157 return (unsigned long long)ptr
->connect_timeout
;
159 case MEMCACHED_BEHAVIOR_SOCKET_SEND_SIZE
:
162 socklen_t sock_length
= sizeof(int);
165 /* We just try the first host, and if it is down we return zero */
166 if ((memcached_connect(&ptr
->hosts
[0])) != MEMCACHED_SUCCESS
)
169 if (getsockopt(ptr
->hosts
[0].fd
, SOL_SOCKET
,
170 SO_SNDBUF
, &sock_size
, &sock_length
))
171 return 0; /* Zero means error */
175 case MEMCACHED_BEHAVIOR_SOCKET_RECV_SIZE
:
178 socklen_t sock_length
= sizeof(int);
181 /* We just try the first host, and if it is down we return zero */
182 if ((memcached_connect(&ptr
->hosts
[0])) != MEMCACHED_SUCCESS
)
185 if (getsockopt(ptr
->hosts
[0].fd
, SOL_SOCKET
,
186 SO_RCVBUF
, &sock_size
, &sock_length
))
187 return 0; /* Zero means error */
193 WATCHPOINT_ASSERT(temp_flag
); /* Programming mistake if it gets this far */
194 if (ptr
->flags
& temp_flag
)
199 return MEMCACHED_SUCCESS
;