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
)
14 unsigned int *truefalse
= (unsigned int *)data
;
18 ptr
->flags
|= temp_flag
;
20 ptr
->flags
+= temp_flag
;
23 memcached_return
memcached_behavior_set(memcached_st
*ptr
,
24 memcached_behavior flag
,
29 case MEMCACHED_BEHAVIOR_SUPPORT_CAS
:
30 set_behavior_flag(ptr
, MEM_SUPPORT_CAS
, data
);
32 case MEMCACHED_BEHAVIOR_NO_BLOCK
:
33 set_behavior_flag(ptr
, MEM_NO_BLOCK
, data
);
34 case MEMCACHED_BEHAVIOR_BUFFER_REQUESTS
:
35 set_behavior_flag(ptr
, MEM_BUFFER_REQUESTS
, data
);
37 case MEMCACHED_BEHAVIOR_TCP_NODELAY
:
38 set_behavior_flag(ptr
, MEM_TCP_NODELAY
, data
);
40 case MEMCACHED_BEHAVIOR_DISTRIBUTION
:
41 ptr
->distribution
= *(memcached_server_distribution
*)(data
);
43 case MEMCACHED_BEHAVIOR_HASH
:
44 ptr
->hash
= *(memcached_hash
*)(data
);
46 case MEMCACHED_BEHAVIOR_CACHE_LOOKUPS
:
47 set_behavior_flag(ptr
, MEM_USE_CACHE_LOOKUPS
, data
);
49 case MEMCACHED_BEHAVIOR_KETAMA
:
50 set_behavior_flag(ptr
, MEM_USE_KETAMA
, data
);
52 case MEMCACHED_BEHAVIOR_SORT_HOSTS
:
53 set_behavior_flag(ptr
, MEM_USE_SORT_HOSTS
, data
);
55 case MEMCACHED_BEHAVIOR_USER_DATA
:
58 case MEMCACHED_BEHAVIOR_POLL_TIMEOUT
:
60 int32_t timeout
= (*((int32_t *)data
));
62 ptr
->poll_timeout
= timeout
;
65 case MEMCACHED_BEHAVIOR_SOCKET_SEND_SIZE
:
67 ptr
->send_size
= (*((int *)data
));
71 case MEMCACHED_BEHAVIOR_SOCKET_RECV_SIZE
:
73 ptr
->recv_size
= (*((int *)data
));
81 return MEMCACHED_SUCCESS
;
84 unsigned long long memcached_behavior_get(memcached_st
*ptr
,
85 memcached_behavior flag
)
87 memcached_flags temp_flag
= 0;
91 case MEMCACHED_BEHAVIOR_SUPPORT_CAS
:
92 temp_flag
= MEM_SUPPORT_CAS
;
94 case MEMCACHED_BEHAVIOR_CACHE_LOOKUPS
:
95 temp_flag
= MEM_USE_CACHE_LOOKUPS
;
97 case MEMCACHED_BEHAVIOR_NO_BLOCK
:
98 temp_flag
= MEM_NO_BLOCK
;
100 case MEMCACHED_BEHAVIOR_BUFFER_REQUESTS
:
101 temp_flag
= MEM_BUFFER_REQUESTS
;
103 case MEMCACHED_BEHAVIOR_TCP_NODELAY
:
104 temp_flag
= MEM_TCP_NODELAY
;
106 case MEMCACHED_BEHAVIOR_DISTRIBUTION
:
107 return ptr
->distribution
;
108 case MEMCACHED_BEHAVIOR_HASH
:
110 case MEMCACHED_BEHAVIOR_KETAMA
:
111 temp_flag
= MEM_USE_KETAMA
;
113 case MEMCACHED_BEHAVIOR_SORT_HOSTS
:
114 temp_flag
= MEM_USE_SORT_HOSTS
;
116 case MEMCACHED_BEHAVIOR_USER_DATA
:
118 //return (unsigned long long)ptr->user_data;
119 case MEMCACHED_BEHAVIOR_POLL_TIMEOUT
:
121 return (unsigned long long)ptr
->poll_timeout
;
123 case MEMCACHED_BEHAVIOR_SOCKET_SEND_SIZE
:
126 socklen_t sock_length
= sizeof(int);
129 /* We just try the first host, and if it is down we return zero */
130 if ((memcached_connect(&ptr
->hosts
[0])) != MEMCACHED_SUCCESS
)
133 if (getsockopt(ptr
->hosts
[0].fd
, SOL_SOCKET
,
134 SO_SNDBUF
, &sock_size
, &sock_length
))
135 return 0; /* Zero means error */
139 case MEMCACHED_BEHAVIOR_SOCKET_RECV_SIZE
:
142 socklen_t sock_length
= sizeof(int);
145 /* We just try the first host, and if it is down we return zero */
146 if ((memcached_connect(&ptr
->hosts
[0])) != MEMCACHED_SUCCESS
)
149 if (getsockopt(ptr
->hosts
[0].fd
, SOL_SOCKET
,
150 SO_RCVBUF
, &sock_size
, &sock_length
))
151 return 0; /* Zero means error */
157 WATCHPOINT_ASSERT(temp_flag
); /* Programming mistake if it gets this far */
158 if (ptr
->flags
& temp_flag
)
163 return MEMCACHED_SUCCESS
;