2 +--------------------------------------------------------------------+
3 | libmemcached - C/C++ Client Library for memcached |
4 +--------------------------------------------------------------------+
5 | Redistribution and use in source and binary forms, with or without |
6 | modification, are permitted under the terms of the BSD license. |
7 | You should have received a copy of the license in a bundled file |
8 | named LICENSE; in case you did not receive a copy you can review |
9 | the terms online at: https://opensource.org/licenses/BSD-3-Clause |
10 +--------------------------------------------------------------------+
11 | Copyright (c) 2006-2014 Brian Aker https://datadifferential.com/ |
12 | Copyright (c) 2020 Michael Wallner <mike@php.net> |
13 +--------------------------------------------------------------------+
16 #include "libmemcached/common.h"
17 #include "libmemcached/options.hpp"
18 #include "libmemcached/virtual_bucket.h"
21 #include <sys/types.h>
23 bool memcached_is_consistent_distribution(const Memcached
*memc
) {
24 switch (memc
->distribution
) {
25 case MEMCACHED_DISTRIBUTION_CONSISTENT
:
26 case MEMCACHED_DISTRIBUTION_CONSISTENT_KETAMA
:
27 case MEMCACHED_DISTRIBUTION_CONSISTENT_KETAMA_SPY
:
28 case MEMCACHED_DISTRIBUTION_CONSISTENT_WEIGHTED
: return true;
30 case MEMCACHED_DISTRIBUTION_MODULA
:
31 case MEMCACHED_DISTRIBUTION_RANDOM
:
32 case MEMCACHED_DISTRIBUTION_VIRTUAL_BUCKET
:
33 case MEMCACHED_DISTRIBUTION_CONSISTENT_MAX
: break;
40 This function is used to modify the behavior of running client.
42 We quit all connections so we can reset the sockets.
45 memcached_return_t
memcached_behavior_set(memcached_st
*shell
, const memcached_behavior_t flag
,
47 Memcached
*ptr
= memcached2Memcached(shell
);
49 return MEMCACHED_INVALID_ARGUMENTS
;
53 case MEMCACHED_BEHAVIOR_NUMBER_OF_REPLICAS
: ptr
->number_of_replicas
= (uint32_t) data
; break;
55 case MEMCACHED_BEHAVIOR_IO_MSG_WATERMARK
: ptr
->io_msg_watermark
= (uint32_t) data
; break;
57 case MEMCACHED_BEHAVIOR_IO_BYTES_WATERMARK
: ptr
->io_bytes_watermark
= (uint32_t) data
; break;
59 case MEMCACHED_BEHAVIOR_IO_KEY_PREFETCH
: ptr
->io_key_prefetch
= (uint32_t) data
; break;
61 case MEMCACHED_BEHAVIOR_SND_TIMEOUT
: ptr
->snd_timeout
= (int32_t) data
; break;
63 case MEMCACHED_BEHAVIOR_RCV_TIMEOUT
: ptr
->rcv_timeout
= (int32_t) data
; break;
65 case MEMCACHED_BEHAVIOR_REMOVE_FAILED_SERVERS
: ptr
->flags
.auto_eject_hosts
= bool(data
); break;
67 case MEMCACHED_BEHAVIOR_SERVER_FAILURE_LIMIT
:
69 return memcached_set_error(
70 *ptr
, MEMCACHED_INVALID_ARGUMENTS
, MEMCACHED_AT
,
71 memcached_literal_param(
72 "MEMCACHED_BEHAVIOR_SERVER_FAILURE_LIMIT requires a value greater then zero."));
74 ptr
->server_failure_limit
= uint32_t(data
);
77 case MEMCACHED_BEHAVIOR_SERVER_TIMEOUT_LIMIT
: ptr
->server_timeout_limit
= uint32_t(data
); break;
79 case MEMCACHED_BEHAVIOR_BINARY_PROTOCOL
:
81 ptr
); // We need t shutdown all of the connections to make sure we do the correct protocol
83 ptr
->flags
.verify_key
= false;
85 ptr
->flags
.binary_protocol
= bool(data
);
88 case MEMCACHED_BEHAVIOR_SUPPORT_CAS
: ptr
->flags
.support_cas
= bool(data
); break;
90 case MEMCACHED_BEHAVIOR_NO_BLOCK
:
91 ptr
->flags
.no_block
= bool(data
);
95 case MEMCACHED_BEHAVIOR_BUFFER_REQUESTS
:
96 if (memcached_is_udp(ptr
)) {
97 return memcached_set_error(
98 *ptr
, MEMCACHED_INVALID_ARGUMENTS
, MEMCACHED_AT
,
99 memcached_literal_param("MEMCACHED_BEHAVIOR_BUFFER_REQUESTS cannot be set while "
100 "MEMCACHED_BEHAVIOR_USE_UDP is enabled."));
102 ptr
->flags
.buffer_requests
= bool(data
);
106 case MEMCACHED_BEHAVIOR_USE_UDP
:
108 ptr
); // We need t shutdown all of the connections to make sure we do the correct protocol
109 ptr
->flags
.use_udp
= bool(data
);
111 ptr
->flags
.reply
= false;
112 ptr
->flags
.buffer_requests
= false;
114 ptr
->flags
.reply
= true;
118 case MEMCACHED_BEHAVIOR_TCP_NODELAY
:
119 ptr
->flags
.tcp_nodelay
= bool(data
);
123 case MEMCACHED_BEHAVIOR_TCP_KEEPALIVE
:
124 ptr
->flags
.tcp_keepalive
= bool(data
);
128 case MEMCACHED_BEHAVIOR_DISTRIBUTION
:
129 return memcached_behavior_set_distribution(ptr
, (memcached_server_distribution_t
) data
);
131 case MEMCACHED_BEHAVIOR_KETAMA
: {
134 return memcached_behavior_set_distribution(ptr
, MEMCACHED_DISTRIBUTION_CONSISTENT_KETAMA
);
137 return memcached_behavior_set_distribution(ptr
, MEMCACHED_DISTRIBUTION_MODULA
);
140 case MEMCACHED_BEHAVIOR_KETAMA_WEIGHTED
: {
141 if (bool(data
) == false) {
142 return memcached_behavior_set(ptr
, MEMCACHED_BEHAVIOR_KETAMA
, true);
145 (void) memcached_behavior_set_key_hash(ptr
, MEMCACHED_HASH_MD5
);
146 (void) memcached_behavior_set_distribution_hash(ptr
, MEMCACHED_HASH_MD5
);
148 @note We try to keep the same distribution going. This should be deprecated and rewritten.
150 return memcached_behavior_set_distribution(ptr
, MEMCACHED_DISTRIBUTION_CONSISTENT_WEIGHTED
);
153 case MEMCACHED_BEHAVIOR_HASH
:
154 return memcached_behavior_set_key_hash(ptr
, (memcached_hash_t
)(data
));
156 case MEMCACHED_BEHAVIOR_KETAMA_HASH
:
157 return memcached_behavior_set_distribution_hash(ptr
, (memcached_hash_t
)(data
));
159 case MEMCACHED_BEHAVIOR_CACHE_LOOKUPS
:
160 return memcached_set_error(
161 *ptr
, MEMCACHED_DEPRECATED
, MEMCACHED_AT
,
162 memcached_literal_param("MEMCACHED_BEHAVIOR_CACHE_LOOKUPS has been deprecated."));
164 case MEMCACHED_BEHAVIOR_VERIFY_KEY
:
165 if (ptr
->flags
.binary_protocol
) {
166 return memcached_set_error(
167 *ptr
, MEMCACHED_INVALID_ARGUMENTS
, MEMCACHED_AT
,
168 memcached_literal_param(
169 "MEMCACHED_BEHAVIOR_VERIFY_KEY if the binary protocol has been enabled."));
171 ptr
->flags
.verify_key
= bool(data
);
174 case MEMCACHED_BEHAVIOR_SORT_HOSTS
: {
175 ptr
->flags
.use_sort_hosts
= bool(data
);
176 return run_distribution(ptr
);
179 case MEMCACHED_BEHAVIOR_POLL_TIMEOUT
: ptr
->poll_timeout
= (int32_t) data
; break;
181 case MEMCACHED_BEHAVIOR_CONNECT_TIMEOUT
: ptr
->connect_timeout
= (int32_t) data
; break;
183 case MEMCACHED_BEHAVIOR_RETRY_TIMEOUT
: ptr
->retry_timeout
= int32_t(data
); break;
185 case MEMCACHED_BEHAVIOR_DEAD_TIMEOUT
: ptr
->dead_timeout
= int32_t(data
); break;
187 case MEMCACHED_BEHAVIOR_SOCKET_SEND_SIZE
:
188 ptr
->send_size
= (int32_t) data
;
192 case MEMCACHED_BEHAVIOR_SOCKET_RECV_SIZE
:
193 ptr
->recv_size
= (int32_t) data
;
197 case MEMCACHED_BEHAVIOR_TCP_KEEPIDLE
:
198 ptr
->tcp_keepidle
= (uint32_t) data
;
202 case MEMCACHED_BEHAVIOR_USER_DATA
:
203 return memcached_set_error(*ptr
, MEMCACHED_DEPRECATED
, MEMCACHED_AT
,
204 memcached_literal_param("MEMCACHED_BEHAVIOR_USER_DATA deprecated."));
206 case MEMCACHED_BEHAVIOR_HASH_WITH_PREFIX_KEY
: ptr
->flags
.hash_with_namespace
= bool(data
); break;
208 case MEMCACHED_BEHAVIOR_NOREPLY
:
209 if (memcached_is_udp(ptr
) and bool(data
) == false) {
210 return memcached_set_error(
211 *ptr
, MEMCACHED_INVALID_ARGUMENTS
, MEMCACHED_AT
,
212 memcached_literal_param("MEMCACHED_BEHAVIOR_NOREPLY cannot be disabled while "
213 "MEMCACHED_BEHAVIOR_USE_UDP is enabled."));
215 // We reverse the logic here to make it easier to understand throughout the
217 ptr
->flags
.reply
= bool(data
) ? false : true;
220 case MEMCACHED_BEHAVIOR_AUTO_EJECT_HOSTS
: ptr
->flags
.auto_eject_hosts
= bool(data
); break;
222 case MEMCACHED_BEHAVIOR_RANDOMIZE_REPLICA_READ
:
223 srandom((uint32_t) time(NULL
));
224 ptr
->flags
.randomize_replica_read
= bool(data
);
227 case MEMCACHED_BEHAVIOR_CORK
:
228 return memcached_set_error(
229 *ptr
, MEMCACHED_DEPRECATED
, MEMCACHED_AT
,
230 memcached_literal_param(
231 "MEMCACHED_BEHAVIOR_CORK is now incorporated into the driver by default."));
233 case MEMCACHED_BEHAVIOR_LOAD_FROM_FILE
:
234 return memcached_set_error(
235 *ptr
, MEMCACHED_INVALID_ARGUMENTS
, MEMCACHED_AT
,
236 memcached_literal_param(
237 "MEMCACHED_BEHAVIOR_LOAD_FROM_FILE can not be set with memcached_behavior_set()"));
239 case MEMCACHED_BEHAVIOR_MAX
:
241 /* Shouldn't get here */
242 assert_msg(0, "Invalid behavior passed to memcached_behavior_set()");
243 return memcached_set_error(
244 *ptr
, MEMCACHED_INVALID_ARGUMENTS
, MEMCACHED_AT
,
245 memcached_literal_param("Invalid behavior passed to memcached_behavior_set()"));
248 return MEMCACHED_SUCCESS
;
251 bool _is_auto_eject_host(const memcached_st
*ptr
) {
252 return ptr
->flags
.auto_eject_hosts
;
255 uint64_t memcached_behavior_get(memcached_st
*shell
, const memcached_behavior_t flag
) {
256 Memcached
*ptr
= memcached2Memcached(shell
);
258 return MEMCACHED_INVALID_ARGUMENTS
;
262 case MEMCACHED_BEHAVIOR_NUMBER_OF_REPLICAS
: return ptr
->number_of_replicas
;
264 case MEMCACHED_BEHAVIOR_IO_MSG_WATERMARK
: return ptr
->io_msg_watermark
;
266 case MEMCACHED_BEHAVIOR_IO_BYTES_WATERMARK
: return ptr
->io_bytes_watermark
;
268 case MEMCACHED_BEHAVIOR_IO_KEY_PREFETCH
: return ptr
->io_key_prefetch
;
270 case MEMCACHED_BEHAVIOR_BINARY_PROTOCOL
: return ptr
->flags
.binary_protocol
;
272 case MEMCACHED_BEHAVIOR_SUPPORT_CAS
: return ptr
->flags
.support_cas
;
274 case MEMCACHED_BEHAVIOR_CACHE_LOOKUPS
: return true;
276 case MEMCACHED_BEHAVIOR_NO_BLOCK
: return ptr
->flags
.no_block
;
278 case MEMCACHED_BEHAVIOR_BUFFER_REQUESTS
: return ptr
->flags
.buffer_requests
;
280 case MEMCACHED_BEHAVIOR_USE_UDP
: return memcached_is_udp(ptr
);
282 case MEMCACHED_BEHAVIOR_TCP_NODELAY
: return ptr
->flags
.tcp_nodelay
;
284 case MEMCACHED_BEHAVIOR_VERIFY_KEY
: return ptr
->flags
.verify_key
;
286 case MEMCACHED_BEHAVIOR_KETAMA_WEIGHTED
:
287 if (memcached_is_consistent_distribution(ptr
)) {
288 return memcached_is_weighted_ketama(ptr
);
292 case MEMCACHED_BEHAVIOR_DISTRIBUTION
: return ptr
->distribution
;
294 case MEMCACHED_BEHAVIOR_KETAMA
: return memcached_is_consistent_distribution(ptr
);
296 case MEMCACHED_BEHAVIOR_HASH
: return hashkit_get_function(&ptr
->hashkit
);
298 case MEMCACHED_BEHAVIOR_KETAMA_HASH
: return hashkit_get_distribution_function(&ptr
->hashkit
);
300 case MEMCACHED_BEHAVIOR_REMOVE_FAILED_SERVERS
: return ptr
->flags
.auto_eject_hosts
;
302 case MEMCACHED_BEHAVIOR_SERVER_FAILURE_LIMIT
: return ptr
->server_failure_limit
;
304 case MEMCACHED_BEHAVIOR_SERVER_TIMEOUT_LIMIT
: return ptr
->server_timeout_limit
;
306 case MEMCACHED_BEHAVIOR_SORT_HOSTS
: return ptr
->flags
.use_sort_hosts
;
308 case MEMCACHED_BEHAVIOR_POLL_TIMEOUT
: return (uint64_t) ptr
->poll_timeout
;
310 case MEMCACHED_BEHAVIOR_CONNECT_TIMEOUT
: return (uint64_t) ptr
->connect_timeout
;
312 case MEMCACHED_BEHAVIOR_RETRY_TIMEOUT
: return (uint64_t) ptr
->retry_timeout
;
314 case MEMCACHED_BEHAVIOR_DEAD_TIMEOUT
: return uint64_t(ptr
->dead_timeout
);
316 case MEMCACHED_BEHAVIOR_SND_TIMEOUT
: return (uint64_t) ptr
->snd_timeout
;
318 case MEMCACHED_BEHAVIOR_RCV_TIMEOUT
: return (uint64_t) ptr
->rcv_timeout
;
320 case MEMCACHED_BEHAVIOR_TCP_KEEPIDLE
: return (uint64_t) ptr
->tcp_keepidle
;
322 case MEMCACHED_BEHAVIOR_SOCKET_SEND_SIZE
: {
324 socklen_t sock_length
= sizeof(int);
326 if (ptr
->send_size
!= -1) // If value is -1 then we are using the default
328 return (uint64_t) ptr
->send_size
;
331 memcached_instance_st
*instance
= memcached_instance_fetch(ptr
, 0);
333 if (instance
) // If we have an instance we test, otherwise we just set and pray
336 /* We just try the first host, and if it is down we return zero */
337 if (memcached_failed(memcached_connect(instance
))) {
341 if (memcached_failed(memcached_io_wait_for_write(instance
))) {
345 if (getsockopt(instance
->fd
, SOL_SOCKET
, SO_SNDBUF
, (char *) &sock_size
, &sock_length
) < 0) {
346 memcached_set_errno(*ptr
, get_socket_errno(), MEMCACHED_AT
);
347 return 0; /* Zero means error */
351 return (uint64_t) sock_size
;
354 case MEMCACHED_BEHAVIOR_SOCKET_RECV_SIZE
: {
356 socklen_t sock_length
= sizeof(int);
358 if (ptr
->recv_size
!= -1) // If value is -1 then we are using the default
359 return (uint64_t) ptr
->recv_size
;
361 memcached_instance_st
*instance
= memcached_instance_fetch(ptr
, 0);
367 /* We just try the first host, and if it is down we return zero */
368 if (memcached_failed(memcached_connect(instance
))) {
372 if (memcached_failed(memcached_io_wait_for_write(instance
))) {
376 if (getsockopt(instance
->fd
, SOL_SOCKET
, SO_RCVBUF
, (char *) &sock_size
, &sock_length
) < 0) {
377 memcached_set_errno(*ptr
, get_socket_errno(), MEMCACHED_AT
);
378 return 0; /* Zero means error */
382 return (uint64_t) sock_size
;
385 case MEMCACHED_BEHAVIOR_USER_DATA
:
386 memcached_set_error(*ptr
, MEMCACHED_DEPRECATED
, MEMCACHED_AT
,
387 memcached_literal_param("MEMCACHED_BEHAVIOR_USER_DATA deprecated."));
390 case MEMCACHED_BEHAVIOR_HASH_WITH_PREFIX_KEY
: return ptr
->flags
.hash_with_namespace
;
392 case MEMCACHED_BEHAVIOR_NOREPLY
: return ptr
->flags
.reply
? false : true;
394 case MEMCACHED_BEHAVIOR_AUTO_EJECT_HOSTS
: return ptr
->flags
.auto_eject_hosts
;
396 case MEMCACHED_BEHAVIOR_RANDOMIZE_REPLICA_READ
: return ptr
->flags
.randomize_replica_read
;
398 case MEMCACHED_BEHAVIOR_CORK
:
405 case MEMCACHED_BEHAVIOR_TCP_KEEPALIVE
: return ptr
->flags
.tcp_keepalive
;
407 case MEMCACHED_BEHAVIOR_LOAD_FROM_FILE
: return bool(memcached_parse_filename(ptr
));
409 case MEMCACHED_BEHAVIOR_MAX
:
410 default: assert_msg(0, "Invalid behavior passed to memcached_behavior_get()"); return 0;
416 memcached_return_t
memcached_behavior_set_distribution(memcached_st
*shell
,
417 memcached_server_distribution_t type
) {
418 Memcached
*ptr
= memcached2Memcached(shell
);
421 case MEMCACHED_DISTRIBUTION_MODULA
: break;
423 case MEMCACHED_DISTRIBUTION_CONSISTENT
:
424 case MEMCACHED_DISTRIBUTION_CONSISTENT_KETAMA
: memcached_set_weighted_ketama(ptr
, false); break;
426 case MEMCACHED_DISTRIBUTION_RANDOM
: break;
428 case MEMCACHED_DISTRIBUTION_CONSISTENT_KETAMA_SPY
: break;
430 case MEMCACHED_DISTRIBUTION_CONSISTENT_WEIGHTED
:
431 memcached_set_weighted_ketama(ptr
, true);
434 case MEMCACHED_DISTRIBUTION_VIRTUAL_BUCKET
: break;
437 case MEMCACHED_DISTRIBUTION_CONSISTENT_MAX
:
438 return memcached_set_error(
439 *ptr
, MEMCACHED_INVALID_ARGUMENTS
, MEMCACHED_AT
,
440 memcached_literal_param("Invalid memcached_server_distribution_t"));
442 ptr
->distribution
= type
;
444 return run_distribution(ptr
);
447 return MEMCACHED_INVALID_ARGUMENTS
;
450 memcached_server_distribution_t
memcached_behavior_get_distribution(memcached_st
*shell
) {
451 Memcached
*ptr
= memcached2Memcached(shell
);
453 return ptr
->distribution
;
456 return MEMCACHED_DISTRIBUTION_CONSISTENT_MAX
;
459 memcached_return_t
memcached_behavior_set_key_hash(memcached_st
*shell
, memcached_hash_t type
) {
460 Memcached
*ptr
= memcached2Memcached(shell
);
462 if (hashkit_success(hashkit_set_function(&ptr
->hashkit
, (hashkit_hash_algorithm_t
) type
))) {
463 return MEMCACHED_SUCCESS
;
466 return memcached_set_error(*ptr
, MEMCACHED_INVALID_ARGUMENTS
, MEMCACHED_AT
,
467 memcached_literal_param("Invalid memcached_hash_t()"));
470 return MEMCACHED_INVALID_ARGUMENTS
;
473 memcached_hash_t
memcached_behavior_get_key_hash(memcached_st
*shell
) {
474 Memcached
*ptr
= memcached2Memcached(shell
);
476 return (memcached_hash_t
) hashkit_get_function(&ptr
->hashkit
);
479 return MEMCACHED_HASH_MAX
;
482 memcached_return_t
memcached_behavior_set_distribution_hash(memcached_st
*shell
,
483 memcached_hash_t type
) {
484 Memcached
*ptr
= memcached2Memcached(shell
);
487 hashkit_set_distribution_function(&ptr
->hashkit
, (hashkit_hash_algorithm_t
) type
)))
489 return MEMCACHED_SUCCESS
;
492 return memcached_set_error(*ptr
, MEMCACHED_INVALID_ARGUMENTS
, MEMCACHED_AT
,
493 memcached_literal_param("Invalid memcached_hash_t()"));
496 return MEMCACHED_INVALID_ARGUMENTS
;
499 memcached_hash_t
memcached_behavior_get_distribution_hash(memcached_st
*shell
) {
500 Memcached
*ptr
= memcached2Memcached(shell
);
502 return (memcached_hash_t
) hashkit_get_function(&ptr
->hashkit
);
505 return MEMCACHED_HASH_MAX
;
508 const char *libmemcached_string_behavior(const memcached_behavior_t flag
) {
510 case MEMCACHED_BEHAVIOR_SERVER_TIMEOUT_LIMIT
: return "MEMCACHED_BEHAVIOR_SERVER_TIMEOUT_LIMIT";
511 case MEMCACHED_BEHAVIOR_NO_BLOCK
: return "MEMCACHED_BEHAVIOR_NO_BLOCK";
512 case MEMCACHED_BEHAVIOR_TCP_NODELAY
: return "MEMCACHED_BEHAVIOR_TCP_NODELAY";
513 case MEMCACHED_BEHAVIOR_HASH
: return "MEMCACHED_BEHAVIOR_HASH";
514 case MEMCACHED_BEHAVIOR_KETAMA
: return "MEMCACHED_BEHAVIOR_KETAMA";
515 case MEMCACHED_BEHAVIOR_SOCKET_SEND_SIZE
: return "MEMCACHED_BEHAVIOR_SOCKET_SEND_SIZE";
516 case MEMCACHED_BEHAVIOR_SOCKET_RECV_SIZE
: return "MEMCACHED_BEHAVIOR_SOCKET_RECV_SIZE";
517 case MEMCACHED_BEHAVIOR_CACHE_LOOKUPS
: return "MEMCACHED_BEHAVIOR_CACHE_LOOKUPS";
518 case MEMCACHED_BEHAVIOR_SUPPORT_CAS
: return "MEMCACHED_BEHAVIOR_SUPPORT_CAS";
519 case MEMCACHED_BEHAVIOR_POLL_TIMEOUT
: return "MEMCACHED_BEHAVIOR_POLL_TIMEOUT";
520 case MEMCACHED_BEHAVIOR_DISTRIBUTION
: return "MEMCACHED_BEHAVIOR_DISTRIBUTION";
521 case MEMCACHED_BEHAVIOR_BUFFER_REQUESTS
: return "MEMCACHED_BEHAVIOR_BUFFER_REQUESTS";
522 case MEMCACHED_BEHAVIOR_USER_DATA
: return "MEMCACHED_BEHAVIOR_USER_DATA";
523 case MEMCACHED_BEHAVIOR_SORT_HOSTS
: return "MEMCACHED_BEHAVIOR_SORT_HOSTS";
524 case MEMCACHED_BEHAVIOR_VERIFY_KEY
: return "MEMCACHED_BEHAVIOR_VERIFY_KEY";
525 case MEMCACHED_BEHAVIOR_CONNECT_TIMEOUT
: return "MEMCACHED_BEHAVIOR_CONNECT_TIMEOUT";
526 case MEMCACHED_BEHAVIOR_RETRY_TIMEOUT
: return "MEMCACHED_BEHAVIOR_RETRY_TIMEOUT";
527 case MEMCACHED_BEHAVIOR_DEAD_TIMEOUT
: return "MEMCACHED_BEHAVIOR_DEAD_TIMEOUT";
528 case MEMCACHED_BEHAVIOR_KETAMA_WEIGHTED
: return "MEMCACHED_BEHAVIOR_KETAMA_WEIGHTED";
529 case MEMCACHED_BEHAVIOR_KETAMA_HASH
: return "MEMCACHED_BEHAVIOR_KETAMA_HASH";
530 case MEMCACHED_BEHAVIOR_BINARY_PROTOCOL
: return "MEMCACHED_BEHAVIOR_BINARY_PROTOCOL";
531 case MEMCACHED_BEHAVIOR_SND_TIMEOUT
: return "MEMCACHED_BEHAVIOR_SND_TIMEOUT";
532 case MEMCACHED_BEHAVIOR_RCV_TIMEOUT
: return "MEMCACHED_BEHAVIOR_RCV_TIMEOUT";
533 case MEMCACHED_BEHAVIOR_SERVER_FAILURE_LIMIT
: return "MEMCACHED_BEHAVIOR_SERVER_FAILURE_LIMIT";
534 case MEMCACHED_BEHAVIOR_IO_MSG_WATERMARK
: return "MEMCACHED_BEHAVIOR_IO_MSG_WATERMARK";
535 case MEMCACHED_BEHAVIOR_IO_BYTES_WATERMARK
: return "MEMCACHED_BEHAVIOR_IO_BYTES_WATERMARK";
536 case MEMCACHED_BEHAVIOR_IO_KEY_PREFETCH
: return "MEMCACHED_BEHAVIOR_IO_KEY_PREFETCH";
537 case MEMCACHED_BEHAVIOR_HASH_WITH_PREFIX_KEY
: return "MEMCACHED_BEHAVIOR_HASH_WITH_PREFIX_KEY";
538 case MEMCACHED_BEHAVIOR_NOREPLY
: return "MEMCACHED_BEHAVIOR_NOREPLY";
539 case MEMCACHED_BEHAVIOR_USE_UDP
: return "MEMCACHED_BEHAVIOR_USE_UDP";
540 case MEMCACHED_BEHAVIOR_AUTO_EJECT_HOSTS
: return "MEMCACHED_BEHAVIOR_AUTO_EJECT_HOSTS";
541 case MEMCACHED_BEHAVIOR_REMOVE_FAILED_SERVERS
: return "MEMCACHED_BEHAVIOR_REMOVE_FAILED_SERVERS";
542 case MEMCACHED_BEHAVIOR_NUMBER_OF_REPLICAS
: return "MEMCACHED_BEHAVIOR_NUMBER_OF_REPLICAS";
543 case MEMCACHED_BEHAVIOR_RANDOMIZE_REPLICA_READ
:
544 return "MEMCACHED_BEHAVIOR_RANDOMIZE_REPLICA_READ";
545 case MEMCACHED_BEHAVIOR_CORK
: return "MEMCACHED_BEHAVIOR_CORK";
546 case MEMCACHED_BEHAVIOR_TCP_KEEPALIVE
: return "MEMCACHED_BEHAVIOR_TCP_KEEPALIVE";
547 case MEMCACHED_BEHAVIOR_TCP_KEEPIDLE
: return "MEMCACHED_BEHAVIOR_TCP_KEEPIDLE";
548 case MEMCACHED_BEHAVIOR_LOAD_FROM_FILE
: return "MEMCACHED_BEHAVIOR_LOAD_FROM_FILE";
550 case MEMCACHED_BEHAVIOR_MAX
: return "INVALID memcached_behavior_t";
554 const char *libmemcached_string_distribution(const memcached_server_distribution_t flag
) {
556 case MEMCACHED_DISTRIBUTION_MODULA
: return "MEMCACHED_DISTRIBUTION_MODULA";
557 case MEMCACHED_DISTRIBUTION_CONSISTENT
: return "MEMCACHED_DISTRIBUTION_CONSISTENT";
558 case MEMCACHED_DISTRIBUTION_CONSISTENT_KETAMA
: return "MEMCACHED_DISTRIBUTION_CONSISTENT_KETAMA";
559 case MEMCACHED_DISTRIBUTION_RANDOM
: return "MEMCACHED_DISTRIBUTION_RANDOM";
560 case MEMCACHED_DISTRIBUTION_CONSISTENT_KETAMA_SPY
:
561 return "MEMCACHED_DISTRIBUTION_CONSISTENT_KETAMA_SPY";
562 case MEMCACHED_DISTRIBUTION_CONSISTENT_WEIGHTED
:
563 return "MEMCACHED_DISTRIBUTION_CONSISTENT_WEIGHTED";
564 case MEMCACHED_DISTRIBUTION_VIRTUAL_BUCKET
: return "MEMCACHED_DISTRIBUTION_VIRTUAL_BUCKET";
566 case MEMCACHED_DISTRIBUTION_CONSISTENT_MAX
: return "INVALID memcached_server_distribution_t";
570 memcached_return_t
memcached_bucket_set(memcached_st
*shell
, const uint32_t *host_map
,
571 const uint32_t *forward_map
, const uint32_t buckets
,
572 const uint32_t replicas
) {
573 Memcached
*self
= memcached2Memcached(shell
);
574 memcached_return_t rc
;
577 return MEMCACHED_INVALID_ARGUMENTS
;
580 if (host_map
== NULL
) {
581 return MEMCACHED_INVALID_ARGUMENTS
;
584 memcached_server_distribution_t old
= memcached_behavior_get_distribution(self
);
586 if (memcached_failed(
587 rc
= memcached_behavior_set_distribution(self
, MEMCACHED_DISTRIBUTION_VIRTUAL_BUCKET
)))
592 if (memcached_failed(
593 rc
= memcached_virtual_bucket_create(self
, host_map
, forward_map
, buckets
, replicas
)))
595 memcached_behavior_set_distribution(self
, old
);