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
:
31 case MEMCACHED_DISTRIBUTION_MODULA
:
32 case MEMCACHED_DISTRIBUTION_RANDOM
:
33 case MEMCACHED_DISTRIBUTION_VIRTUAL_BUCKET
:
34 case MEMCACHED_DISTRIBUTION_CONSISTENT_MAX
:
42 This function is used to modify the behavior of running client.
44 We quit all connections so we can reset the sockets.
47 memcached_return_t
memcached_behavior_set(memcached_st
*shell
, const memcached_behavior_t flag
,
49 Memcached
*ptr
= memcached2Memcached(shell
);
51 return MEMCACHED_INVALID_ARGUMENTS
;
55 case MEMCACHED_BEHAVIOR_NUMBER_OF_REPLICAS
:
56 ptr
->number_of_replicas
= (uint32_t) data
;
59 case MEMCACHED_BEHAVIOR_IO_MSG_WATERMARK
:
60 ptr
->io_msg_watermark
= (uint32_t) data
;
63 case MEMCACHED_BEHAVIOR_IO_BYTES_WATERMARK
:
64 ptr
->io_bytes_watermark
= (uint32_t) data
;
67 case MEMCACHED_BEHAVIOR_IO_KEY_PREFETCH
:
68 ptr
->io_key_prefetch
= (uint32_t) data
;
71 case MEMCACHED_BEHAVIOR_SND_TIMEOUT
:
72 ptr
->snd_timeout
= (int32_t) data
;
75 case MEMCACHED_BEHAVIOR_RCV_TIMEOUT
:
76 ptr
->rcv_timeout
= (int32_t) data
;
79 case MEMCACHED_BEHAVIOR_REMOVE_FAILED_SERVERS
:
80 ptr
->flags
.auto_eject_hosts
= bool(data
);
83 case MEMCACHED_BEHAVIOR_SERVER_FAILURE_LIMIT
:
85 return memcached_set_error(
86 *ptr
, MEMCACHED_INVALID_ARGUMENTS
, MEMCACHED_AT
,
87 memcached_literal_param(
88 "MEMCACHED_BEHAVIOR_SERVER_FAILURE_LIMIT requires a value greater then zero."));
90 ptr
->server_failure_limit
= uint32_t(data
);
93 case MEMCACHED_BEHAVIOR_SERVER_TIMEOUT_LIMIT
:
94 ptr
->server_timeout_limit
= uint32_t(data
);
97 case MEMCACHED_BEHAVIOR_BINARY_PROTOCOL
:
99 ptr
); // We need t shutdown all of the connections to make sure we do the correct protocol
101 ptr
->flags
.verify_key
= false;
103 ptr
->flags
.binary_protocol
= bool(data
);
106 case MEMCACHED_BEHAVIOR_SUPPORT_CAS
:
107 ptr
->flags
.support_cas
= bool(data
);
110 case MEMCACHED_BEHAVIOR_NO_BLOCK
:
111 ptr
->flags
.no_block
= bool(data
);
115 case MEMCACHED_BEHAVIOR_BUFFER_REQUESTS
:
116 if (memcached_is_udp(ptr
)) {
117 return memcached_set_error(
118 *ptr
, MEMCACHED_INVALID_ARGUMENTS
, MEMCACHED_AT
,
119 memcached_literal_param("MEMCACHED_BEHAVIOR_BUFFER_REQUESTS cannot be set while "
120 "MEMCACHED_BEHAVIOR_USE_UDP is enabled."));
122 ptr
->flags
.buffer_requests
= bool(data
);
126 case MEMCACHED_BEHAVIOR_USE_UDP
:
128 ptr
); // We need t shutdown all of the connections to make sure we do the correct protocol
129 ptr
->flags
.use_udp
= bool(data
);
131 ptr
->flags
.reply
= false;
132 ptr
->flags
.buffer_requests
= false;
134 ptr
->flags
.reply
= true;
138 case MEMCACHED_BEHAVIOR_TCP_NODELAY
:
139 ptr
->flags
.tcp_nodelay
= bool(data
);
143 case MEMCACHED_BEHAVIOR_TCP_KEEPALIVE
:
144 ptr
->flags
.tcp_keepalive
= bool(data
);
148 case MEMCACHED_BEHAVIOR_DISTRIBUTION
:
149 return memcached_behavior_set_distribution(ptr
, (memcached_server_distribution_t
) data
);
151 case MEMCACHED_BEHAVIOR_KETAMA
: {
154 return memcached_behavior_set_distribution(ptr
, MEMCACHED_DISTRIBUTION_CONSISTENT_KETAMA
);
157 return memcached_behavior_set_distribution(ptr
, MEMCACHED_DISTRIBUTION_MODULA
);
160 case MEMCACHED_BEHAVIOR_KETAMA_WEIGHTED
: {
161 if (bool(data
) == false) {
162 return memcached_behavior_set(ptr
, MEMCACHED_BEHAVIOR_KETAMA
, true);
165 (void) memcached_behavior_set_key_hash(ptr
, MEMCACHED_HASH_MD5
);
166 (void) memcached_behavior_set_distribution_hash(ptr
, MEMCACHED_HASH_MD5
);
168 @note We try to keep the same distribution going. This should be deprecated and rewritten.
170 return memcached_behavior_set_distribution(ptr
, MEMCACHED_DISTRIBUTION_CONSISTENT_WEIGHTED
);
173 case MEMCACHED_BEHAVIOR_HASH
:
174 return memcached_behavior_set_key_hash(ptr
, (memcached_hash_t
)(data
));
176 case MEMCACHED_BEHAVIOR_KETAMA_HASH
:
177 return memcached_behavior_set_distribution_hash(ptr
, (memcached_hash_t
)(data
));
179 case MEMCACHED_BEHAVIOR_CACHE_LOOKUPS
:
180 return memcached_set_error(
181 *ptr
, MEMCACHED_DEPRECATED
, MEMCACHED_AT
,
182 memcached_literal_param("MEMCACHED_BEHAVIOR_CACHE_LOOKUPS has been deprecated."));
184 case MEMCACHED_BEHAVIOR_VERIFY_KEY
:
185 if (ptr
->flags
.binary_protocol
) {
186 return memcached_set_error(
187 *ptr
, MEMCACHED_INVALID_ARGUMENTS
, MEMCACHED_AT
,
188 memcached_literal_param(
189 "MEMCACHED_BEHAVIOR_VERIFY_KEY if the binary protocol has been enabled."));
191 ptr
->flags
.verify_key
= bool(data
);
194 case MEMCACHED_BEHAVIOR_SORT_HOSTS
: {
195 ptr
->flags
.use_sort_hosts
= bool(data
);
196 return run_distribution(ptr
);
199 case MEMCACHED_BEHAVIOR_POLL_TIMEOUT
:
200 ptr
->poll_timeout
= (int32_t) data
;
203 case MEMCACHED_BEHAVIOR_CONNECT_TIMEOUT
:
204 ptr
->connect_timeout
= (int32_t) data
;
207 case MEMCACHED_BEHAVIOR_RETRY_TIMEOUT
:
208 ptr
->retry_timeout
= int32_t(data
);
211 case MEMCACHED_BEHAVIOR_DEAD_TIMEOUT
:
212 ptr
->dead_timeout
= int32_t(data
);
215 case MEMCACHED_BEHAVIOR_SOCKET_SEND_SIZE
:
216 ptr
->send_size
= (int32_t) data
;
220 case MEMCACHED_BEHAVIOR_SOCKET_RECV_SIZE
:
221 ptr
->recv_size
= (int32_t) data
;
225 case MEMCACHED_BEHAVIOR_TCP_KEEPIDLE
:
226 ptr
->tcp_keepidle
= (uint32_t) data
;
230 case MEMCACHED_BEHAVIOR_USER_DATA
:
231 return memcached_set_error(*ptr
, MEMCACHED_DEPRECATED
, MEMCACHED_AT
,
232 memcached_literal_param("MEMCACHED_BEHAVIOR_USER_DATA deprecated."));
234 case MEMCACHED_BEHAVIOR_HASH_WITH_PREFIX_KEY
:
235 ptr
->flags
.hash_with_namespace
= bool(data
);
238 case MEMCACHED_BEHAVIOR_NOREPLY
:
239 if (memcached_is_udp(ptr
) and bool(data
) == false) {
240 return memcached_set_error(
241 *ptr
, MEMCACHED_INVALID_ARGUMENTS
, MEMCACHED_AT
,
242 memcached_literal_param("MEMCACHED_BEHAVIOR_NOREPLY cannot be disabled while "
243 "MEMCACHED_BEHAVIOR_USE_UDP is enabled."));
245 // We reverse the logic here to make it easier to understand throughout the
247 ptr
->flags
.reply
= bool(data
) ? false : true;
250 case MEMCACHED_BEHAVIOR_AUTO_EJECT_HOSTS
:
251 ptr
->flags
.auto_eject_hosts
= bool(data
);
254 case MEMCACHED_BEHAVIOR_RANDOMIZE_REPLICA_READ
:
255 srandom((uint32_t) time(NULL
));
256 ptr
->flags
.randomize_replica_read
= bool(data
);
259 case MEMCACHED_BEHAVIOR_CORK
:
260 return memcached_set_error(
261 *ptr
, MEMCACHED_DEPRECATED
, MEMCACHED_AT
,
262 memcached_literal_param(
263 "MEMCACHED_BEHAVIOR_CORK is now incorporated into the driver by default."));
265 case MEMCACHED_BEHAVIOR_LOAD_FROM_FILE
:
266 return memcached_set_error(
267 *ptr
, MEMCACHED_INVALID_ARGUMENTS
, MEMCACHED_AT
,
268 memcached_literal_param(
269 "MEMCACHED_BEHAVIOR_LOAD_FROM_FILE can not be set with memcached_behavior_set()"));
271 case MEMCACHED_BEHAVIOR_MAX
:
273 /* Shouldn't get here */
274 assert_msg(0, "Invalid behavior passed to memcached_behavior_set()");
275 return memcached_set_error(
276 *ptr
, MEMCACHED_INVALID_ARGUMENTS
, MEMCACHED_AT
,
277 memcached_literal_param("Invalid behavior passed to memcached_behavior_set()"));
280 return MEMCACHED_SUCCESS
;
283 bool _is_auto_eject_host(const memcached_st
*ptr
) {
284 return ptr
->flags
.auto_eject_hosts
;
287 uint64_t memcached_behavior_get(memcached_st
*shell
, const memcached_behavior_t flag
) {
288 Memcached
*ptr
= memcached2Memcached(shell
);
290 return MEMCACHED_INVALID_ARGUMENTS
;
294 case MEMCACHED_BEHAVIOR_NUMBER_OF_REPLICAS
:
295 return ptr
->number_of_replicas
;
297 case MEMCACHED_BEHAVIOR_IO_MSG_WATERMARK
:
298 return ptr
->io_msg_watermark
;
300 case MEMCACHED_BEHAVIOR_IO_BYTES_WATERMARK
:
301 return ptr
->io_bytes_watermark
;
303 case MEMCACHED_BEHAVIOR_IO_KEY_PREFETCH
:
304 return ptr
->io_key_prefetch
;
306 case MEMCACHED_BEHAVIOR_BINARY_PROTOCOL
:
307 return ptr
->flags
.binary_protocol
;
309 case MEMCACHED_BEHAVIOR_SUPPORT_CAS
:
310 return ptr
->flags
.support_cas
;
312 case MEMCACHED_BEHAVIOR_CACHE_LOOKUPS
:
315 case MEMCACHED_BEHAVIOR_NO_BLOCK
:
316 return ptr
->flags
.no_block
;
318 case MEMCACHED_BEHAVIOR_BUFFER_REQUESTS
:
319 return ptr
->flags
.buffer_requests
;
321 case MEMCACHED_BEHAVIOR_USE_UDP
:
322 return memcached_is_udp(ptr
);
324 case MEMCACHED_BEHAVIOR_TCP_NODELAY
:
325 return ptr
->flags
.tcp_nodelay
;
327 case MEMCACHED_BEHAVIOR_VERIFY_KEY
:
328 return ptr
->flags
.verify_key
;
330 case MEMCACHED_BEHAVIOR_KETAMA_WEIGHTED
:
331 if (memcached_is_consistent_distribution(ptr
)) {
332 return memcached_is_weighted_ketama(ptr
);
336 case MEMCACHED_BEHAVIOR_DISTRIBUTION
:
337 return ptr
->distribution
;
339 case MEMCACHED_BEHAVIOR_KETAMA
:
340 return memcached_is_consistent_distribution(ptr
);
342 case MEMCACHED_BEHAVIOR_HASH
:
343 return hashkit_get_function(&ptr
->hashkit
);
345 case MEMCACHED_BEHAVIOR_KETAMA_HASH
:
346 return hashkit_get_distribution_function(&ptr
->hashkit
);
348 case MEMCACHED_BEHAVIOR_REMOVE_FAILED_SERVERS
:
349 return ptr
->flags
.auto_eject_hosts
;
351 case MEMCACHED_BEHAVIOR_SERVER_FAILURE_LIMIT
:
352 return ptr
->server_failure_limit
;
354 case MEMCACHED_BEHAVIOR_SERVER_TIMEOUT_LIMIT
:
355 return ptr
->server_timeout_limit
;
357 case MEMCACHED_BEHAVIOR_SORT_HOSTS
:
358 return ptr
->flags
.use_sort_hosts
;
360 case MEMCACHED_BEHAVIOR_POLL_TIMEOUT
:
361 return (uint64_t) ptr
->poll_timeout
;
363 case MEMCACHED_BEHAVIOR_CONNECT_TIMEOUT
:
364 return (uint64_t) ptr
->connect_timeout
;
366 case MEMCACHED_BEHAVIOR_RETRY_TIMEOUT
:
367 return (uint64_t) ptr
->retry_timeout
;
369 case MEMCACHED_BEHAVIOR_DEAD_TIMEOUT
:
370 return uint64_t(ptr
->dead_timeout
);
372 case MEMCACHED_BEHAVIOR_SND_TIMEOUT
:
373 return (uint64_t) ptr
->snd_timeout
;
375 case MEMCACHED_BEHAVIOR_RCV_TIMEOUT
:
376 return (uint64_t) ptr
->rcv_timeout
;
378 case MEMCACHED_BEHAVIOR_TCP_KEEPIDLE
:
379 return (uint64_t) ptr
->tcp_keepidle
;
381 case MEMCACHED_BEHAVIOR_SOCKET_SEND_SIZE
: {
383 socklen_t sock_length
= sizeof(int);
385 if (ptr
->send_size
!= -1) // If value is -1 then we are using the default
387 return (uint64_t) ptr
->send_size
;
390 memcached_instance_st
*instance
= memcached_instance_fetch(ptr
, 0);
392 if (instance
) // If we have an instance we test, otherwise we just set and pray
395 /* We just try the first host, and if it is down we return zero */
396 if (memcached_failed(memcached_connect(instance
))) {
400 if (memcached_failed(memcached_io_wait_for_write(instance
))) {
404 if (getsockopt(instance
->fd
, SOL_SOCKET
, SO_SNDBUF
, (char *) &sock_size
, &sock_length
) < 0) {
405 memcached_set_errno(*ptr
, get_socket_errno(), MEMCACHED_AT
);
406 return 0; /* Zero means error */
410 return (uint64_t) sock_size
;
413 case MEMCACHED_BEHAVIOR_SOCKET_RECV_SIZE
: {
415 socklen_t sock_length
= sizeof(int);
417 if (ptr
->recv_size
!= -1) // If value is -1 then we are using the default
418 return (uint64_t) ptr
->recv_size
;
420 memcached_instance_st
*instance
= memcached_instance_fetch(ptr
, 0);
426 /* We just try the first host, and if it is down we return zero */
427 if (memcached_failed(memcached_connect(instance
))) {
431 if (memcached_failed(memcached_io_wait_for_write(instance
))) {
435 if (getsockopt(instance
->fd
, SOL_SOCKET
, SO_RCVBUF
, (char *) &sock_size
, &sock_length
) < 0) {
436 memcached_set_errno(*ptr
, get_socket_errno(), MEMCACHED_AT
);
437 return 0; /* Zero means error */
441 return (uint64_t) sock_size
;
444 case MEMCACHED_BEHAVIOR_USER_DATA
:
445 memcached_set_error(*ptr
, MEMCACHED_DEPRECATED
, MEMCACHED_AT
,
446 memcached_literal_param("MEMCACHED_BEHAVIOR_USER_DATA deprecated."));
449 case MEMCACHED_BEHAVIOR_HASH_WITH_PREFIX_KEY
:
450 return ptr
->flags
.hash_with_namespace
;
452 case MEMCACHED_BEHAVIOR_NOREPLY
:
453 return ptr
->flags
.reply
? false : true;
455 case MEMCACHED_BEHAVIOR_AUTO_EJECT_HOSTS
:
456 return ptr
->flags
.auto_eject_hosts
;
458 case MEMCACHED_BEHAVIOR_RANDOMIZE_REPLICA_READ
:
459 return ptr
->flags
.randomize_replica_read
;
461 case MEMCACHED_BEHAVIOR_CORK
:
468 case MEMCACHED_BEHAVIOR_TCP_KEEPALIVE
:
469 return ptr
->flags
.tcp_keepalive
;
471 case MEMCACHED_BEHAVIOR_LOAD_FROM_FILE
:
472 return bool(memcached_parse_filename(ptr
));
474 case MEMCACHED_BEHAVIOR_MAX
:
476 assert_msg(0, "Invalid behavior passed to memcached_behavior_get()");
483 memcached_return_t
memcached_behavior_set_distribution(memcached_st
*shell
,
484 memcached_server_distribution_t type
) {
485 Memcached
*ptr
= memcached2Memcached(shell
);
488 case MEMCACHED_DISTRIBUTION_MODULA
:
491 case MEMCACHED_DISTRIBUTION_CONSISTENT
:
492 case MEMCACHED_DISTRIBUTION_CONSISTENT_KETAMA
:
493 memcached_set_weighted_ketama(ptr
, false);
496 case MEMCACHED_DISTRIBUTION_RANDOM
:
499 case MEMCACHED_DISTRIBUTION_CONSISTENT_KETAMA_SPY
:
502 case MEMCACHED_DISTRIBUTION_CONSISTENT_WEIGHTED
:
503 memcached_set_weighted_ketama(ptr
, true);
506 case MEMCACHED_DISTRIBUTION_VIRTUAL_BUCKET
:
510 case MEMCACHED_DISTRIBUTION_CONSISTENT_MAX
:
511 return memcached_set_error(
512 *ptr
, MEMCACHED_INVALID_ARGUMENTS
, MEMCACHED_AT
,
513 memcached_literal_param("Invalid memcached_server_distribution_t"));
515 ptr
->distribution
= type
;
517 return run_distribution(ptr
);
520 return MEMCACHED_INVALID_ARGUMENTS
;
523 memcached_server_distribution_t
memcached_behavior_get_distribution(memcached_st
*shell
) {
524 Memcached
*ptr
= memcached2Memcached(shell
);
526 return ptr
->distribution
;
529 return MEMCACHED_DISTRIBUTION_CONSISTENT_MAX
;
532 memcached_return_t
memcached_behavior_set_key_hash(memcached_st
*shell
, memcached_hash_t type
) {
533 Memcached
*ptr
= memcached2Memcached(shell
);
535 if (hashkit_success(hashkit_set_function(&ptr
->hashkit
, (hashkit_hash_algorithm_t
) type
))) {
536 return MEMCACHED_SUCCESS
;
539 return memcached_set_error(*ptr
, MEMCACHED_INVALID_ARGUMENTS
, MEMCACHED_AT
,
540 memcached_literal_param("Invalid memcached_hash_t()"));
543 return MEMCACHED_INVALID_ARGUMENTS
;
546 memcached_hash_t
memcached_behavior_get_key_hash(memcached_st
*shell
) {
547 Memcached
*ptr
= memcached2Memcached(shell
);
549 return (memcached_hash_t
) hashkit_get_function(&ptr
->hashkit
);
552 return MEMCACHED_HASH_MAX
;
555 memcached_return_t
memcached_behavior_set_distribution_hash(memcached_st
*shell
,
556 memcached_hash_t type
) {
557 Memcached
*ptr
= memcached2Memcached(shell
);
560 hashkit_set_distribution_function(&ptr
->hashkit
, (hashkit_hash_algorithm_t
) type
)))
562 return MEMCACHED_SUCCESS
;
565 return memcached_set_error(*ptr
, MEMCACHED_INVALID_ARGUMENTS
, MEMCACHED_AT
,
566 memcached_literal_param("Invalid memcached_hash_t()"));
569 return MEMCACHED_INVALID_ARGUMENTS
;
572 memcached_hash_t
memcached_behavior_get_distribution_hash(memcached_st
*shell
) {
573 Memcached
*ptr
= memcached2Memcached(shell
);
575 return (memcached_hash_t
) hashkit_get_function(&ptr
->hashkit
);
578 return MEMCACHED_HASH_MAX
;
581 const char *libmemcached_string_behavior(const memcached_behavior_t flag
) {
583 case MEMCACHED_BEHAVIOR_SERVER_TIMEOUT_LIMIT
:
584 return "MEMCACHED_BEHAVIOR_SERVER_TIMEOUT_LIMIT";
585 case MEMCACHED_BEHAVIOR_NO_BLOCK
:
586 return "MEMCACHED_BEHAVIOR_NO_BLOCK";
587 case MEMCACHED_BEHAVIOR_TCP_NODELAY
:
588 return "MEMCACHED_BEHAVIOR_TCP_NODELAY";
589 case MEMCACHED_BEHAVIOR_HASH
:
590 return "MEMCACHED_BEHAVIOR_HASH";
591 case MEMCACHED_BEHAVIOR_KETAMA
:
592 return "MEMCACHED_BEHAVIOR_KETAMA";
593 case MEMCACHED_BEHAVIOR_SOCKET_SEND_SIZE
:
594 return "MEMCACHED_BEHAVIOR_SOCKET_SEND_SIZE";
595 case MEMCACHED_BEHAVIOR_SOCKET_RECV_SIZE
:
596 return "MEMCACHED_BEHAVIOR_SOCKET_RECV_SIZE";
597 case MEMCACHED_BEHAVIOR_CACHE_LOOKUPS
:
598 return "MEMCACHED_BEHAVIOR_CACHE_LOOKUPS";
599 case MEMCACHED_BEHAVIOR_SUPPORT_CAS
:
600 return "MEMCACHED_BEHAVIOR_SUPPORT_CAS";
601 case MEMCACHED_BEHAVIOR_POLL_TIMEOUT
:
602 return "MEMCACHED_BEHAVIOR_POLL_TIMEOUT";
603 case MEMCACHED_BEHAVIOR_DISTRIBUTION
:
604 return "MEMCACHED_BEHAVIOR_DISTRIBUTION";
605 case MEMCACHED_BEHAVIOR_BUFFER_REQUESTS
:
606 return "MEMCACHED_BEHAVIOR_BUFFER_REQUESTS";
607 case MEMCACHED_BEHAVIOR_USER_DATA
:
608 return "MEMCACHED_BEHAVIOR_USER_DATA";
609 case MEMCACHED_BEHAVIOR_SORT_HOSTS
:
610 return "MEMCACHED_BEHAVIOR_SORT_HOSTS";
611 case MEMCACHED_BEHAVIOR_VERIFY_KEY
:
612 return "MEMCACHED_BEHAVIOR_VERIFY_KEY";
613 case MEMCACHED_BEHAVIOR_CONNECT_TIMEOUT
:
614 return "MEMCACHED_BEHAVIOR_CONNECT_TIMEOUT";
615 case MEMCACHED_BEHAVIOR_RETRY_TIMEOUT
:
616 return "MEMCACHED_BEHAVIOR_RETRY_TIMEOUT";
617 case MEMCACHED_BEHAVIOR_DEAD_TIMEOUT
:
618 return "MEMCACHED_BEHAVIOR_DEAD_TIMEOUT";
619 case MEMCACHED_BEHAVIOR_KETAMA_WEIGHTED
:
620 return "MEMCACHED_BEHAVIOR_KETAMA_WEIGHTED";
621 case MEMCACHED_BEHAVIOR_KETAMA_HASH
:
622 return "MEMCACHED_BEHAVIOR_KETAMA_HASH";
623 case MEMCACHED_BEHAVIOR_BINARY_PROTOCOL
:
624 return "MEMCACHED_BEHAVIOR_BINARY_PROTOCOL";
625 case MEMCACHED_BEHAVIOR_SND_TIMEOUT
:
626 return "MEMCACHED_BEHAVIOR_SND_TIMEOUT";
627 case MEMCACHED_BEHAVIOR_RCV_TIMEOUT
:
628 return "MEMCACHED_BEHAVIOR_RCV_TIMEOUT";
629 case MEMCACHED_BEHAVIOR_SERVER_FAILURE_LIMIT
:
630 return "MEMCACHED_BEHAVIOR_SERVER_FAILURE_LIMIT";
631 case MEMCACHED_BEHAVIOR_IO_MSG_WATERMARK
:
632 return "MEMCACHED_BEHAVIOR_IO_MSG_WATERMARK";
633 case MEMCACHED_BEHAVIOR_IO_BYTES_WATERMARK
:
634 return "MEMCACHED_BEHAVIOR_IO_BYTES_WATERMARK";
635 case MEMCACHED_BEHAVIOR_IO_KEY_PREFETCH
:
636 return "MEMCACHED_BEHAVIOR_IO_KEY_PREFETCH";
637 case MEMCACHED_BEHAVIOR_HASH_WITH_PREFIX_KEY
:
638 return "MEMCACHED_BEHAVIOR_HASH_WITH_PREFIX_KEY";
639 case MEMCACHED_BEHAVIOR_NOREPLY
:
640 return "MEMCACHED_BEHAVIOR_NOREPLY";
641 case MEMCACHED_BEHAVIOR_USE_UDP
:
642 return "MEMCACHED_BEHAVIOR_USE_UDP";
643 case MEMCACHED_BEHAVIOR_AUTO_EJECT_HOSTS
:
644 return "MEMCACHED_BEHAVIOR_AUTO_EJECT_HOSTS";
645 case MEMCACHED_BEHAVIOR_REMOVE_FAILED_SERVERS
:
646 return "MEMCACHED_BEHAVIOR_REMOVE_FAILED_SERVERS";
647 case MEMCACHED_BEHAVIOR_NUMBER_OF_REPLICAS
:
648 return "MEMCACHED_BEHAVIOR_NUMBER_OF_REPLICAS";
649 case MEMCACHED_BEHAVIOR_RANDOMIZE_REPLICA_READ
:
650 return "MEMCACHED_BEHAVIOR_RANDOMIZE_REPLICA_READ";
651 case MEMCACHED_BEHAVIOR_CORK
:
652 return "MEMCACHED_BEHAVIOR_CORK";
653 case MEMCACHED_BEHAVIOR_TCP_KEEPALIVE
:
654 return "MEMCACHED_BEHAVIOR_TCP_KEEPALIVE";
655 case MEMCACHED_BEHAVIOR_TCP_KEEPIDLE
:
656 return "MEMCACHED_BEHAVIOR_TCP_KEEPIDLE";
657 case MEMCACHED_BEHAVIOR_LOAD_FROM_FILE
:
658 return "MEMCACHED_BEHAVIOR_LOAD_FROM_FILE";
660 case MEMCACHED_BEHAVIOR_MAX
:
661 return "INVALID memcached_behavior_t";
665 const char *libmemcached_string_distribution(const memcached_server_distribution_t flag
) {
667 case MEMCACHED_DISTRIBUTION_MODULA
:
668 return "MEMCACHED_DISTRIBUTION_MODULA";
669 case MEMCACHED_DISTRIBUTION_CONSISTENT
:
670 return "MEMCACHED_DISTRIBUTION_CONSISTENT";
671 case MEMCACHED_DISTRIBUTION_CONSISTENT_KETAMA
:
672 return "MEMCACHED_DISTRIBUTION_CONSISTENT_KETAMA";
673 case MEMCACHED_DISTRIBUTION_RANDOM
:
674 return "MEMCACHED_DISTRIBUTION_RANDOM";
675 case MEMCACHED_DISTRIBUTION_CONSISTENT_KETAMA_SPY
:
676 return "MEMCACHED_DISTRIBUTION_CONSISTENT_KETAMA_SPY";
677 case MEMCACHED_DISTRIBUTION_CONSISTENT_WEIGHTED
:
678 return "MEMCACHED_DISTRIBUTION_CONSISTENT_WEIGHTED";
679 case MEMCACHED_DISTRIBUTION_VIRTUAL_BUCKET
:
680 return "MEMCACHED_DISTRIBUTION_VIRTUAL_BUCKET";
682 case MEMCACHED_DISTRIBUTION_CONSISTENT_MAX
:
683 return "INVALID memcached_server_distribution_t";
687 memcached_return_t
memcached_bucket_set(memcached_st
*shell
, const uint32_t *host_map
,
688 const uint32_t *forward_map
, const uint32_t buckets
,
689 const uint32_t replicas
) {
690 Memcached
*self
= memcached2Memcached(shell
);
691 memcached_return_t rc
;
694 return MEMCACHED_INVALID_ARGUMENTS
;
697 if (host_map
== NULL
) {
698 return MEMCACHED_INVALID_ARGUMENTS
;
701 memcached_server_distribution_t old
= memcached_behavior_get_distribution(self
);
703 if (memcached_failed(
704 rc
= memcached_behavior_set_distribution(self
, MEMCACHED_DISTRIBUTION_VIRTUAL_BUCKET
)))
709 if (memcached_failed(
710 rc
= memcached_virtual_bucket_create(self
, host_map
, forward_map
, buckets
, replicas
)))
712 memcached_behavior_set_distribution(self
, old
);