1 /* vim:expandtab:shiftwidth=2:tabstop=2:smarttab:
5 * Copyright (C) 2011 Data Differential, http://datadifferential.com/
6 * Copyright (C) 2006-2009 Brian Aker All rights reserved.
8 * Redistribution and use in source and binary forms, with or without
9 * modification, are permitted provided that the following conditions are
12 * * Redistributions of source code must retain the above copyright
13 * notice, this list of conditions and the following disclaimer.
15 * * Redistributions in binary form must reproduce the above
16 * copyright notice, this list of conditions and the following disclaimer
17 * in the documentation and/or other materials provided with the
20 * * The names of its contributors may not be used to endorse or
21 * promote products derived from this software without specific prior
24 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
25 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
26 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
27 * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
28 * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
29 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
30 * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
31 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
32 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
33 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
34 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
38 #include <libmemcached/common.h>
39 #include <libmemcached/options.hpp>
40 #include <libmemcached/virtual_bucket.h>
43 #include <sys/types.h>
45 bool memcached_is_consistent_distribution(const Memcached
* memc
)
47 switch (memc
->distribution
)
49 case MEMCACHED_DISTRIBUTION_CONSISTENT
:
50 case MEMCACHED_DISTRIBUTION_CONSISTENT_KETAMA
:
51 case MEMCACHED_DISTRIBUTION_CONSISTENT_KETAMA_SPY
:
52 case MEMCACHED_DISTRIBUTION_CONSISTENT_WEIGHTED
:
55 case MEMCACHED_DISTRIBUTION_MODULA
:
56 case MEMCACHED_DISTRIBUTION_RANDOM
:
57 case MEMCACHED_DISTRIBUTION_VIRTUAL_BUCKET
:
58 case MEMCACHED_DISTRIBUTION_CONSISTENT_MAX
:
66 This function is used to modify the behavior of running client.
68 We quit all connections so we can reset the sockets.
71 memcached_return_t
memcached_behavior_set(memcached_st
*shell
,
72 const memcached_behavior_t flag
,
75 Memcached
* ptr
= memcached2Memcached(shell
);
78 return MEMCACHED_INVALID_ARGUMENTS
;
83 case MEMCACHED_BEHAVIOR_NUMBER_OF_REPLICAS
:
84 ptr
->number_of_replicas
= (uint32_t)data
;
87 case MEMCACHED_BEHAVIOR_IO_MSG_WATERMARK
:
88 ptr
->io_msg_watermark
= (uint32_t) data
;
91 case MEMCACHED_BEHAVIOR_IO_BYTES_WATERMARK
:
92 ptr
->io_bytes_watermark
= (uint32_t)data
;
95 case MEMCACHED_BEHAVIOR_IO_KEY_PREFETCH
:
96 ptr
->io_key_prefetch
= (uint32_t)data
;
99 case MEMCACHED_BEHAVIOR_SND_TIMEOUT
:
100 ptr
->snd_timeout
= (int32_t)data
;
103 case MEMCACHED_BEHAVIOR_RCV_TIMEOUT
:
104 ptr
->rcv_timeout
= (int32_t)data
;
107 case MEMCACHED_BEHAVIOR_REMOVE_FAILED_SERVERS
:
108 ptr
->flags
.auto_eject_hosts
= bool(data
);
110 case MEMCACHED_BEHAVIOR_SERVER_FAILURE_LIMIT
:
113 return memcached_set_error(*ptr
, MEMCACHED_INVALID_ARGUMENTS
, MEMCACHED_AT
,
114 memcached_literal_param("MEMCACHED_BEHAVIOR_SERVER_FAILURE_LIMIT requires a value greater then zero."));
116 ptr
->server_failure_limit
= uint32_t(data
);
119 case MEMCACHED_BEHAVIOR_SERVER_TIMEOUT_LIMIT
:
120 ptr
->server_timeout_limit
= uint32_t(data
);
123 case MEMCACHED_BEHAVIOR_BINARY_PROTOCOL
:
124 send_quit(ptr
); // We need t shutdown all of the connections to make sure we do the correct protocol
127 ptr
->flags
.verify_key
= false;
129 ptr
->flags
.binary_protocol
= bool(data
);
132 case MEMCACHED_BEHAVIOR_SUPPORT_CAS
:
133 ptr
->flags
.support_cas
= bool(data
);
136 case MEMCACHED_BEHAVIOR_NO_BLOCK
:
137 ptr
->flags
.no_block
= bool(data
);
141 case MEMCACHED_BEHAVIOR_BUFFER_REQUESTS
:
142 if (memcached_is_udp(ptr
))
144 return memcached_set_error(*ptr
, MEMCACHED_INVALID_ARGUMENTS
, MEMCACHED_AT
,
145 memcached_literal_param("MEMCACHED_BEHAVIOR_BUFFER_REQUESTS cannot be set while MEMCACHED_BEHAVIOR_USE_UDP is enabled."));
147 ptr
->flags
.buffer_requests
= bool(data
);
151 case MEMCACHED_BEHAVIOR_USE_UDP
:
152 send_quit(ptr
); // We need t shutdown all of the connections to make sure we do the correct protocol
153 ptr
->flags
.use_udp
= bool(data
);
156 ptr
->flags
.reply
= false;
157 ptr
->flags
.buffer_requests
= false;
161 ptr
->flags
.reply
= true;
165 case MEMCACHED_BEHAVIOR_TCP_NODELAY
:
166 ptr
->flags
.tcp_nodelay
= bool(data
);
170 case MEMCACHED_BEHAVIOR_TCP_KEEPALIVE
:
171 ptr
->flags
.tcp_keepalive
= bool(data
);
175 case MEMCACHED_BEHAVIOR_DISTRIBUTION
:
176 return memcached_behavior_set_distribution(ptr
, (memcached_server_distribution_t
)data
);
178 case MEMCACHED_BEHAVIOR_KETAMA
:
182 return memcached_behavior_set_distribution(ptr
, MEMCACHED_DISTRIBUTION_CONSISTENT_KETAMA
);
185 return memcached_behavior_set_distribution(ptr
, MEMCACHED_DISTRIBUTION_MODULA
);
188 case MEMCACHED_BEHAVIOR_KETAMA_WEIGHTED
:
190 if (bool(data
) == false)
192 return memcached_behavior_set(ptr
, MEMCACHED_BEHAVIOR_KETAMA
, true);
195 (void)memcached_behavior_set_key_hash(ptr
, MEMCACHED_HASH_MD5
);
196 (void)memcached_behavior_set_distribution_hash(ptr
, MEMCACHED_HASH_MD5
);
198 @note We try to keep the same distribution going. This should be deprecated and rewritten.
200 return memcached_behavior_set_distribution(ptr
, MEMCACHED_DISTRIBUTION_CONSISTENT_WEIGHTED
);
203 case MEMCACHED_BEHAVIOR_HASH
:
204 return memcached_behavior_set_key_hash(ptr
, (memcached_hash_t
)(data
));
206 case MEMCACHED_BEHAVIOR_KETAMA_HASH
:
207 return memcached_behavior_set_distribution_hash(ptr
, (memcached_hash_t
)(data
));
209 case MEMCACHED_BEHAVIOR_CACHE_LOOKUPS
:
210 return memcached_set_error(*ptr
, MEMCACHED_DEPRECATED
, MEMCACHED_AT
,
211 memcached_literal_param("MEMCACHED_BEHAVIOR_CACHE_LOOKUPS has been deprecated."));
213 case MEMCACHED_BEHAVIOR_VERIFY_KEY
:
214 if (ptr
->flags
.binary_protocol
)
216 return memcached_set_error(*ptr
, MEMCACHED_INVALID_ARGUMENTS
, MEMCACHED_AT
,
217 memcached_literal_param("MEMCACHED_BEHAVIOR_VERIFY_KEY if the binary protocol has been enabled."));
219 ptr
->flags
.verify_key
= bool(data
);
222 case MEMCACHED_BEHAVIOR_SORT_HOSTS
:
224 ptr
->flags
.use_sort_hosts
= bool(data
);
225 return run_distribution(ptr
);
228 case MEMCACHED_BEHAVIOR_POLL_TIMEOUT
:
229 ptr
->poll_timeout
= (int32_t)data
;
232 case MEMCACHED_BEHAVIOR_CONNECT_TIMEOUT
:
233 ptr
->connect_timeout
= (int32_t)data
;
236 case MEMCACHED_BEHAVIOR_RETRY_TIMEOUT
:
237 ptr
->retry_timeout
= int32_t(data
);
240 case MEMCACHED_BEHAVIOR_DEAD_TIMEOUT
:
241 ptr
->dead_timeout
= int32_t(data
);
244 case MEMCACHED_BEHAVIOR_SOCKET_SEND_SIZE
:
245 ptr
->send_size
= (int32_t)data
;
249 case MEMCACHED_BEHAVIOR_SOCKET_RECV_SIZE
:
250 ptr
->recv_size
= (int32_t)data
;
254 case MEMCACHED_BEHAVIOR_TCP_KEEPIDLE
:
255 ptr
->tcp_keepidle
= (uint32_t)data
;
259 case MEMCACHED_BEHAVIOR_USER_DATA
:
260 return memcached_set_error(*ptr
, MEMCACHED_DEPRECATED
, MEMCACHED_AT
,
261 memcached_literal_param("MEMCACHED_BEHAVIOR_USER_DATA deprecated."));
263 case MEMCACHED_BEHAVIOR_HASH_WITH_PREFIX_KEY
:
264 ptr
->flags
.hash_with_namespace
= bool(data
);
267 case MEMCACHED_BEHAVIOR_NOREPLY
:
268 if (memcached_is_udp(ptr
) and bool(data
) == false)
270 return memcached_set_error(*ptr
, MEMCACHED_INVALID_ARGUMENTS
, MEMCACHED_AT
,
271 memcached_literal_param("MEMCACHED_BEHAVIOR_NOREPLY cannot be disabled while MEMCACHED_BEHAVIOR_USE_UDP is enabled."));
273 // We reverse the logic here to make it easier to understand throughout the
275 ptr
->flags
.reply
= bool(data
) ? false : true;
278 case MEMCACHED_BEHAVIOR_AUTO_EJECT_HOSTS
:
279 ptr
->flags
.auto_eject_hosts
= bool(data
);
282 case MEMCACHED_BEHAVIOR_RANDOMIZE_REPLICA_READ
:
283 srandom((uint32_t) time(NULL
));
284 ptr
->flags
.randomize_replica_read
= bool(data
);
287 case MEMCACHED_BEHAVIOR_CORK
:
288 return memcached_set_error(*ptr
, MEMCACHED_DEPRECATED
, MEMCACHED_AT
,
289 memcached_literal_param("MEMCACHED_BEHAVIOR_CORK is now incorporated into the driver by default."));
291 case MEMCACHED_BEHAVIOR_LOAD_FROM_FILE
:
292 return memcached_set_error(*ptr
, MEMCACHED_INVALID_ARGUMENTS
, MEMCACHED_AT
,
293 memcached_literal_param("MEMCACHED_BEHAVIOR_LOAD_FROM_FILE can not be set with memcached_behavior_set()"));
295 case MEMCACHED_BEHAVIOR_MAX
:
297 /* Shouldn't get here */
298 assert_msg(0, "Invalid behavior passed to memcached_behavior_set()");
299 return memcached_set_error(*ptr
, MEMCACHED_INVALID_ARGUMENTS
, MEMCACHED_AT
,
300 memcached_literal_param("Invalid behavior passed to memcached_behavior_set()"));
303 return MEMCACHED_SUCCESS
;
306 bool _is_auto_eject_host(const memcached_st
*ptr
)
308 return ptr
->flags
.auto_eject_hosts
;
311 uint64_t memcached_behavior_get(memcached_st
*shell
,
312 const memcached_behavior_t flag
)
314 Memcached
* ptr
= memcached2Memcached(shell
);
317 return MEMCACHED_INVALID_ARGUMENTS
;
322 case MEMCACHED_BEHAVIOR_NUMBER_OF_REPLICAS
:
323 return ptr
->number_of_replicas
;
325 case MEMCACHED_BEHAVIOR_IO_MSG_WATERMARK
:
326 return ptr
->io_msg_watermark
;
328 case MEMCACHED_BEHAVIOR_IO_BYTES_WATERMARK
:
329 return ptr
->io_bytes_watermark
;
331 case MEMCACHED_BEHAVIOR_IO_KEY_PREFETCH
:
332 return ptr
->io_key_prefetch
;
334 case MEMCACHED_BEHAVIOR_BINARY_PROTOCOL
:
335 return ptr
->flags
.binary_protocol
;
337 case MEMCACHED_BEHAVIOR_SUPPORT_CAS
:
338 return ptr
->flags
.support_cas
;
340 case MEMCACHED_BEHAVIOR_CACHE_LOOKUPS
:
343 case MEMCACHED_BEHAVIOR_NO_BLOCK
:
344 return ptr
->flags
.no_block
;
346 case MEMCACHED_BEHAVIOR_BUFFER_REQUESTS
:
347 return ptr
->flags
.buffer_requests
;
349 case MEMCACHED_BEHAVIOR_USE_UDP
:
350 return memcached_is_udp(ptr
);
352 case MEMCACHED_BEHAVIOR_TCP_NODELAY
:
353 return ptr
->flags
.tcp_nodelay
;
355 case MEMCACHED_BEHAVIOR_VERIFY_KEY
:
356 return ptr
->flags
.verify_key
;
358 case MEMCACHED_BEHAVIOR_KETAMA_WEIGHTED
:
359 if (memcached_is_consistent_distribution(ptr
))
361 return memcached_is_weighted_ketama(ptr
);
365 case MEMCACHED_BEHAVIOR_DISTRIBUTION
:
366 return ptr
->distribution
;
368 case MEMCACHED_BEHAVIOR_KETAMA
:
369 return memcached_is_consistent_distribution(ptr
);
371 case MEMCACHED_BEHAVIOR_HASH
:
372 return hashkit_get_function(&ptr
->hashkit
);
374 case MEMCACHED_BEHAVIOR_KETAMA_HASH
:
375 return hashkit_get_function(&ptr
->hashkit
);
377 case MEMCACHED_BEHAVIOR_REMOVE_FAILED_SERVERS
:
378 case MEMCACHED_BEHAVIOR_SERVER_FAILURE_LIMIT
:
379 return ptr
->server_failure_limit
;
381 case MEMCACHED_BEHAVIOR_SERVER_TIMEOUT_LIMIT
:
382 return ptr
->server_timeout_limit
;
384 case MEMCACHED_BEHAVIOR_SORT_HOSTS
:
385 return ptr
->flags
.use_sort_hosts
;
387 case MEMCACHED_BEHAVIOR_POLL_TIMEOUT
:
388 return (uint64_t)ptr
->poll_timeout
;
390 case MEMCACHED_BEHAVIOR_CONNECT_TIMEOUT
:
391 return (uint64_t)ptr
->connect_timeout
;
393 case MEMCACHED_BEHAVIOR_RETRY_TIMEOUT
:
394 return (uint64_t)ptr
->retry_timeout
;
396 case MEMCACHED_BEHAVIOR_DEAD_TIMEOUT
:
397 return uint64_t(ptr
->dead_timeout
);
399 case MEMCACHED_BEHAVIOR_SND_TIMEOUT
:
400 return (uint64_t)ptr
->snd_timeout
;
402 case MEMCACHED_BEHAVIOR_RCV_TIMEOUT
:
403 return (uint64_t)ptr
->rcv_timeout
;
405 case MEMCACHED_BEHAVIOR_TCP_KEEPIDLE
:
406 return (uint64_t)ptr
->tcp_keepidle
;
408 case MEMCACHED_BEHAVIOR_SOCKET_SEND_SIZE
:
411 socklen_t sock_length
= sizeof(int);
413 if (ptr
->send_size
!= -1) // If value is -1 then we are using the default
415 return (uint64_t) ptr
->send_size
;
418 memcached_instance_st
* instance
= memcached_instance_fetch(ptr
, 0);
420 if (instance
) // If we have an instance we test, otherwise we just set and pray
423 /* We just try the first host, and if it is down we return zero */
424 if (memcached_failed(memcached_connect(instance
)))
429 if (memcached_failed(memcached_io_wait_for_write(instance
)))
434 if (getsockopt(instance
->fd
, SOL_SOCKET
, SO_SNDBUF
, (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_SOCKET_RECV_SIZE
:
447 socklen_t sock_length
= sizeof(int);
449 if (ptr
->recv_size
!= -1) // If value is -1 then we are using the default
450 return (uint64_t) ptr
->recv_size
;
452 memcached_instance_st
* instance
= memcached_instance_fetch(ptr
, 0);
459 /* We just try the first host, and if it is down we return zero */
460 if (memcached_failed(memcached_connect(instance
)))
465 if (memcached_failed(memcached_io_wait_for_write(instance
)))
470 if (getsockopt(instance
->fd
, SOL_SOCKET
, SO_RCVBUF
, (char*)&sock_size
, &sock_length
) < 0)
472 memcached_set_errno(*ptr
, get_socket_errno(), MEMCACHED_AT
);
473 return 0; /* Zero means error */
477 return (uint64_t) sock_size
;
480 case MEMCACHED_BEHAVIOR_USER_DATA
:
481 memcached_set_error(*ptr
, MEMCACHED_DEPRECATED
, MEMCACHED_AT
,
482 memcached_literal_param("MEMCACHED_BEHAVIOR_USER_DATA deprecated."));
485 case MEMCACHED_BEHAVIOR_HASH_WITH_PREFIX_KEY
:
486 return ptr
->flags
.hash_with_namespace
;
488 case MEMCACHED_BEHAVIOR_NOREPLY
:
489 return ptr
->flags
.reply
? false : true;
491 case MEMCACHED_BEHAVIOR_AUTO_EJECT_HOSTS
:
492 return ptr
->flags
.auto_eject_hosts
;
494 case MEMCACHED_BEHAVIOR_RANDOMIZE_REPLICA_READ
:
495 return ptr
->flags
.randomize_replica_read
;
497 case MEMCACHED_BEHAVIOR_CORK
:
504 case MEMCACHED_BEHAVIOR_TCP_KEEPALIVE
:
505 return ptr
->flags
.tcp_keepalive
;
507 case MEMCACHED_BEHAVIOR_LOAD_FROM_FILE
:
508 return bool(memcached_parse_filename(ptr
));
510 case MEMCACHED_BEHAVIOR_MAX
:
512 assert_msg(0, "Invalid behavior passed to memcached_behavior_get()");
520 memcached_return_t
memcached_behavior_set_distribution(memcached_st
*shell
, memcached_server_distribution_t type
)
522 Memcached
* ptr
= memcached2Memcached(shell
);
527 case MEMCACHED_DISTRIBUTION_MODULA
:
530 case MEMCACHED_DISTRIBUTION_CONSISTENT
:
531 case MEMCACHED_DISTRIBUTION_CONSISTENT_KETAMA
:
532 memcached_set_weighted_ketama(ptr
, false);
535 case MEMCACHED_DISTRIBUTION_RANDOM
:
538 case MEMCACHED_DISTRIBUTION_CONSISTENT_KETAMA_SPY
:
541 case MEMCACHED_DISTRIBUTION_CONSISTENT_WEIGHTED
:
542 memcached_set_weighted_ketama(ptr
, true);
545 case MEMCACHED_DISTRIBUTION_VIRTUAL_BUCKET
:
549 case MEMCACHED_DISTRIBUTION_CONSISTENT_MAX
:
550 return memcached_set_error(*ptr
, MEMCACHED_INVALID_ARGUMENTS
, MEMCACHED_AT
,
551 memcached_literal_param("Invalid memcached_server_distribution_t"));
553 ptr
->distribution
= type
;
555 return run_distribution(ptr
);
558 return MEMCACHED_INVALID_ARGUMENTS
;
562 memcached_server_distribution_t
memcached_behavior_get_distribution(memcached_st
*shell
)
564 Memcached
* ptr
= memcached2Memcached(shell
);
567 return ptr
->distribution
;
570 return MEMCACHED_DISTRIBUTION_CONSISTENT_MAX
;
573 memcached_return_t
memcached_behavior_set_key_hash(memcached_st
*shell
, memcached_hash_t type
)
575 Memcached
* ptr
= memcached2Memcached(shell
);
578 if (hashkit_success(hashkit_set_function(&ptr
->hashkit
, (hashkit_hash_algorithm_t
)type
)))
580 return MEMCACHED_SUCCESS
;
583 return memcached_set_error(*ptr
, MEMCACHED_INVALID_ARGUMENTS
, MEMCACHED_AT
,
584 memcached_literal_param("Invalid memcached_hash_t()"));
587 return MEMCACHED_INVALID_ARGUMENTS
;
590 memcached_hash_t
memcached_behavior_get_key_hash(memcached_st
*shell
)
592 Memcached
* ptr
= memcached2Memcached(shell
);
595 return (memcached_hash_t
)hashkit_get_function(&ptr
->hashkit
);
598 return MEMCACHED_HASH_MAX
;
601 memcached_return_t
memcached_behavior_set_distribution_hash(memcached_st
*shell
, memcached_hash_t type
)
603 Memcached
* ptr
= memcached2Memcached(shell
);
606 if (hashkit_success(hashkit_set_distribution_function(&ptr
->hashkit
, (hashkit_hash_algorithm_t
)type
)))
608 return MEMCACHED_SUCCESS
;
611 return memcached_set_error(*ptr
, MEMCACHED_INVALID_ARGUMENTS
, MEMCACHED_AT
,
612 memcached_literal_param("Invalid memcached_hash_t()"));
615 return MEMCACHED_INVALID_ARGUMENTS
;
618 memcached_hash_t
memcached_behavior_get_distribution_hash(memcached_st
*shell
)
620 Memcached
* ptr
= memcached2Memcached(shell
);
623 return (memcached_hash_t
)hashkit_get_function(&ptr
->hashkit
);
626 return MEMCACHED_HASH_MAX
;
629 const char *libmemcached_string_behavior(const memcached_behavior_t flag
)
633 case MEMCACHED_BEHAVIOR_SERVER_TIMEOUT_LIMIT
: return "MEMCACHED_BEHAVIOR_SERVER_TIMEOUT_LIMIT";
634 case MEMCACHED_BEHAVIOR_NO_BLOCK
: return "MEMCACHED_BEHAVIOR_NO_BLOCK";
635 case MEMCACHED_BEHAVIOR_TCP_NODELAY
: return "MEMCACHED_BEHAVIOR_TCP_NODELAY";
636 case MEMCACHED_BEHAVIOR_HASH
: return "MEMCACHED_BEHAVIOR_HASH";
637 case MEMCACHED_BEHAVIOR_KETAMA
: return "MEMCACHED_BEHAVIOR_KETAMA";
638 case MEMCACHED_BEHAVIOR_SOCKET_SEND_SIZE
: return "MEMCACHED_BEHAVIOR_SOCKET_SEND_SIZE";
639 case MEMCACHED_BEHAVIOR_SOCKET_RECV_SIZE
: return "MEMCACHED_BEHAVIOR_SOCKET_RECV_SIZE";
640 case MEMCACHED_BEHAVIOR_CACHE_LOOKUPS
: return "MEMCACHED_BEHAVIOR_CACHE_LOOKUPS";
641 case MEMCACHED_BEHAVIOR_SUPPORT_CAS
: return "MEMCACHED_BEHAVIOR_SUPPORT_CAS";
642 case MEMCACHED_BEHAVIOR_POLL_TIMEOUT
: return "MEMCACHED_BEHAVIOR_POLL_TIMEOUT";
643 case MEMCACHED_BEHAVIOR_DISTRIBUTION
: return "MEMCACHED_BEHAVIOR_DISTRIBUTION";
644 case MEMCACHED_BEHAVIOR_BUFFER_REQUESTS
: return "MEMCACHED_BEHAVIOR_BUFFER_REQUESTS";
645 case MEMCACHED_BEHAVIOR_USER_DATA
: return "MEMCACHED_BEHAVIOR_USER_DATA";
646 case MEMCACHED_BEHAVIOR_SORT_HOSTS
: return "MEMCACHED_BEHAVIOR_SORT_HOSTS";
647 case MEMCACHED_BEHAVIOR_VERIFY_KEY
: return "MEMCACHED_BEHAVIOR_VERIFY_KEY";
648 case MEMCACHED_BEHAVIOR_CONNECT_TIMEOUT
: return "MEMCACHED_BEHAVIOR_CONNECT_TIMEOUT";
649 case MEMCACHED_BEHAVIOR_RETRY_TIMEOUT
: return "MEMCACHED_BEHAVIOR_RETRY_TIMEOUT";
650 case MEMCACHED_BEHAVIOR_DEAD_TIMEOUT
: return "MEMCACHED_BEHAVIOR_DEAD_TIMEOUT";
651 case MEMCACHED_BEHAVIOR_KETAMA_WEIGHTED
: return "MEMCACHED_BEHAVIOR_KETAMA_WEIGHTED";
652 case MEMCACHED_BEHAVIOR_KETAMA_HASH
: return "MEMCACHED_BEHAVIOR_KETAMA_HASH";
653 case MEMCACHED_BEHAVIOR_BINARY_PROTOCOL
: return "MEMCACHED_BEHAVIOR_BINARY_PROTOCOL";
654 case MEMCACHED_BEHAVIOR_SND_TIMEOUT
: return "MEMCACHED_BEHAVIOR_SND_TIMEOUT";
655 case MEMCACHED_BEHAVIOR_RCV_TIMEOUT
: return "MEMCACHED_BEHAVIOR_RCV_TIMEOUT";
656 case MEMCACHED_BEHAVIOR_SERVER_FAILURE_LIMIT
: return "MEMCACHED_BEHAVIOR_SERVER_FAILURE_LIMIT";
657 case MEMCACHED_BEHAVIOR_IO_MSG_WATERMARK
: return "MEMCACHED_BEHAVIOR_IO_MSG_WATERMARK";
658 case MEMCACHED_BEHAVIOR_IO_BYTES_WATERMARK
: return "MEMCACHED_BEHAVIOR_IO_BYTES_WATERMARK";
659 case MEMCACHED_BEHAVIOR_IO_KEY_PREFETCH
: return "MEMCACHED_BEHAVIOR_IO_KEY_PREFETCH";
660 case MEMCACHED_BEHAVIOR_HASH_WITH_PREFIX_KEY
: return "MEMCACHED_BEHAVIOR_HASH_WITH_PREFIX_KEY";
661 case MEMCACHED_BEHAVIOR_NOREPLY
: return "MEMCACHED_BEHAVIOR_NOREPLY";
662 case MEMCACHED_BEHAVIOR_USE_UDP
: return "MEMCACHED_BEHAVIOR_USE_UDP";
663 case MEMCACHED_BEHAVIOR_AUTO_EJECT_HOSTS
: return "MEMCACHED_BEHAVIOR_AUTO_EJECT_HOSTS";
664 case MEMCACHED_BEHAVIOR_REMOVE_FAILED_SERVERS
: return "MEMCACHED_BEHAVIOR_REMOVE_FAILED_SERVERS";
665 case MEMCACHED_BEHAVIOR_NUMBER_OF_REPLICAS
: return "MEMCACHED_BEHAVIOR_NUMBER_OF_REPLICAS";
666 case MEMCACHED_BEHAVIOR_RANDOMIZE_REPLICA_READ
: return "MEMCACHED_BEHAVIOR_RANDOMIZE_REPLICA_READ";
667 case MEMCACHED_BEHAVIOR_CORK
: return "MEMCACHED_BEHAVIOR_CORK";
668 case MEMCACHED_BEHAVIOR_TCP_KEEPALIVE
: return "MEMCACHED_BEHAVIOR_TCP_KEEPALIVE";
669 case MEMCACHED_BEHAVIOR_TCP_KEEPIDLE
: return "MEMCACHED_BEHAVIOR_TCP_KEEPIDLE";
670 case MEMCACHED_BEHAVIOR_LOAD_FROM_FILE
: return "MEMCACHED_BEHAVIOR_LOAD_FROM_FILE";
672 case MEMCACHED_BEHAVIOR_MAX
: return "INVALID memcached_behavior_t";
676 const char *libmemcached_string_distribution(const memcached_server_distribution_t flag
)
680 case MEMCACHED_DISTRIBUTION_MODULA
: return "MEMCACHED_DISTRIBUTION_MODULA";
681 case MEMCACHED_DISTRIBUTION_CONSISTENT
: return "MEMCACHED_DISTRIBUTION_CONSISTENT";
682 case MEMCACHED_DISTRIBUTION_CONSISTENT_KETAMA
: return "MEMCACHED_DISTRIBUTION_CONSISTENT_KETAMA";
683 case MEMCACHED_DISTRIBUTION_RANDOM
: return "MEMCACHED_DISTRIBUTION_RANDOM";
684 case MEMCACHED_DISTRIBUTION_CONSISTENT_KETAMA_SPY
: return "MEMCACHED_DISTRIBUTION_CONSISTENT_KETAMA_SPY";
685 case MEMCACHED_DISTRIBUTION_CONSISTENT_WEIGHTED
: return "MEMCACHED_DISTRIBUTION_CONSISTENT_WEIGHTED";
686 case MEMCACHED_DISTRIBUTION_VIRTUAL_BUCKET
: return "MEMCACHED_DISTRIBUTION_VIRTUAL_BUCKET";
688 case MEMCACHED_DISTRIBUTION_CONSISTENT_MAX
: return "INVALID memcached_server_distribution_t";
692 memcached_return_t
memcached_bucket_set(memcached_st
*shell
,
693 const uint32_t *host_map
,
694 const uint32_t *forward_map
,
695 const uint32_t buckets
,
696 const uint32_t replicas
)
698 Memcached
* self
= memcached2Memcached(shell
);
699 memcached_return_t rc
;
703 return MEMCACHED_INVALID_ARGUMENTS
;
706 if (host_map
== NULL
)
708 return MEMCACHED_INVALID_ARGUMENTS
;
711 memcached_server_distribution_t old
= memcached_behavior_get_distribution(self
);
713 if (memcached_failed(rc
=memcached_behavior_set_distribution(self
, MEMCACHED_DISTRIBUTION_VIRTUAL_BUCKET
)))
718 if (memcached_failed(rc
= memcached_virtual_bucket_create(self
, host_map
, forward_map
, buckets
, replicas
)))
720 memcached_behavior_set_distribution(self
, old
);