X-Git-Url: https://git.m6w6.name/?a=blobdiff_plain;f=libmemcached%2Fhosts.c;h=0836c97e02128ddaf36c07287d66d051dc046fac;hb=93723c9e5e41d8d9c41f610623060b28b4ec8ae3;hp=9d8870dca7c731b8b00bd246e176e8963a910e8e;hpb=5d4981688b335b52f18e25c4c42c1314ed7e17df;p=m6w6%2Flibmemcached diff --git a/libmemcached/hosts.c b/libmemcached/hosts.c index 9d8870dc..0836c97e 100644 --- a/libmemcached/hosts.c +++ b/libmemcached/hosts.c @@ -1,3 +1,14 @@ +/* LibMemcached + * Copyright (C) 2006-2010 Brian Aker + * All rights reserved. + * + * Use and distribution licensed under the BSD license. See + * the COPYING file in the parent directory for full text. + * + * Summary: + * + */ + #include "common.h" #include @@ -11,8 +22,8 @@ static memcached_return_t update_continuum(memcached_st *ptr); static int compare_servers(const void *p1, const void *p2) { int return_value; - memcached_server_instance_st *a= (memcached_server_instance_st *)p1; - memcached_server_instance_st *b= (memcached_server_instance_st *)p2; + memcached_server_instance_st a= (memcached_server_instance_st)p1; + memcached_server_instance_st b= (memcached_server_instance_st)p2; return_value= strcmp(a->hostname, b->hostname); @@ -28,9 +39,9 @@ static void sort_hosts(memcached_st *ptr) { if (memcached_server_count(ptr)) { - memcached_server_instance_st *instance; + memcached_server_write_instance_st instance; - qsort(memcached_server_list(ptr), memcached_server_count(ptr), sizeof(memcached_server_instance_st), compare_servers); + qsort(memcached_server_list(ptr), memcached_server_count(ptr), sizeof(memcached_server_st), compare_servers); instance= memcached_server_instance_fetch(ptr, 0); instance->number_of_hosts= memcached_server_count(ptr); } @@ -63,7 +74,7 @@ memcached_return_t run_distribution(memcached_st *ptr) return MEMCACHED_SUCCESS; } -static uint32_t ketama_server_hash(const char *key, unsigned int key_length, uint32_t alignment) +static uint32_t ketama_server_hash(const char *key, size_t key_length, uint32_t alignment) { unsigned char results[16]; @@ -95,7 +106,7 @@ static memcached_return_t update_continuum(memcached_st *ptr) uint32_t host_index; uint32_t continuum_index= 0; uint32_t value; - memcached_server_instance_st *list; + memcached_server_st *list; uint32_t pointer_index; uint32_t pointer_counter= 0; uint32_t pointer_per_server= MEMCACHED_POINTS_PER_SERVER; @@ -116,7 +127,7 @@ static memcached_return_t update_continuum(memcached_st *ptr) list = memcached_server_list(ptr); /* count live servers (those without a retry delay set) */ - is_auto_ejecting= memcached_behavior_get(ptr, MEMCACHED_BEHAVIOR_AUTO_EJECT_HOSTS); + is_auto_ejecting= _is_auto_eject_host(ptr); if (is_auto_ejecting) { live_servers= 0; @@ -165,12 +176,12 @@ static memcached_return_t update_continuum(memcached_st *ptr) { list[host_index].weight = 1; } - if (!is_auto_ejecting || list[host_index].next_retry <= now.tv_sec) + if (! is_auto_ejecting || list[host_index].next_retry <= now.tv_sec) total_weight += list[host_index].weight; } } - for (host_index = 0; host_index < memcached_server_count(ptr); ++host_index) + for (host_index= 0; host_index < memcached_server_count(ptr); ++host_index) { if (is_auto_ejecting && list[host_index].next_retry > now.tv_sec) continue; @@ -214,9 +225,9 @@ static memcached_return_t update_continuum(memcached_st *ptr) if (is_ketama_weighted) { - for (uint32_t x = 0; x < pointer_per_hash; x++) + for (uint32_t x= 0; x < pointer_per_hash; x++) { - value= ketama_server_hash(sort_host, (uint32_t) sort_host_length, x); + value= ketama_server_hash(sort_host, sort_host_length, x); ptr->continuum[continuum_index].index= host_index; ptr->continuum[continuum_index++].value= value; } @@ -260,7 +271,7 @@ static memcached_return_t update_continuum(memcached_st *ptr) { for (uint32_t x = 0; x < pointer_per_hash; x++) { - value= ketama_server_hash(sort_host, (uint32_t) sort_host_length, (int) x); + value= ketama_server_hash(sort_host, sort_host_length, x); ptr->continuum[continuum_index].index= host_index; ptr->continuum[continuum_index++].value= value; } @@ -294,7 +305,7 @@ static memcached_return_t update_continuum(memcached_st *ptr) } -memcached_return_t memcached_server_push(memcached_st *ptr, memcached_server_st *list) +memcached_return_t memcached_server_push(memcached_st *ptr, const memcached_server_list_st list) { uint32_t count; memcached_server_st *new_host_list; @@ -302,9 +313,9 @@ memcached_return_t memcached_server_push(memcached_st *ptr, memcached_server_st if (! list) return MEMCACHED_SUCCESS; - count= memcached_servers_count(list); + count= memcached_server_list_count(list); new_host_list= libmemcached_realloc(ptr, memcached_server_list(ptr), - sizeof(memcached_server_instance_st) * (count + memcached_server_count(ptr))); + sizeof(memcached_server_st) * (count + memcached_server_count(ptr))); if (! new_host_list) return MEMCACHED_MEMORY_ALLOCATION_FAILURE; @@ -313,7 +324,7 @@ memcached_return_t memcached_server_push(memcached_st *ptr, memcached_server_st for (uint32_t x= 0; x < count; x++) { - memcached_server_instance_st *instance; + memcached_server_write_instance_st instance; if ((ptr->flags.use_udp && list[x].type != MEMCACHED_CONNECTION_UDP) || ((list[x].type == MEMCACHED_CONNECTION_UDP) @@ -332,7 +343,7 @@ memcached_return_t memcached_server_push(memcached_st *ptr, memcached_server_st // Provides backwards compatibility with server list. { - memcached_server_instance_st *instance; + memcached_server_write_instance_st instance; instance= memcached_server_instance_fetch(ptr, 0); instance->number_of_hosts= memcached_server_count(ptr); } @@ -403,15 +414,15 @@ static memcached_return_t server_add(memcached_st *ptr, const char *hostname, uint32_t weight, memcached_connection_t type) { - memcached_server_instance_st *new_host_list; - memcached_server_instance_st *instance; + memcached_server_st *new_host_list; + memcached_server_write_instance_st instance; if ( (ptr->flags.use_udp && type != MEMCACHED_CONNECTION_UDP) || ( (type == MEMCACHED_CONNECTION_UDP) && (! ptr->flags.use_udp) ) ) return MEMCACHED_INVALID_HOST_PROTOCOL; new_host_list= libmemcached_realloc(ptr, memcached_server_list(ptr), - sizeof(memcached_server_instance_st) * (ptr->number_of_hosts + 1)); + sizeof(memcached_server_st) * (ptr->number_of_hosts + 1)); if (new_host_list == NULL) return MEMCACHED_MEMORY_ALLOCATION_FAILURE; @@ -428,83 +439,3 @@ static memcached_return_t server_add(memcached_st *ptr, const char *hostname, return run_distribution(ptr); } - -/** - @todo allow lists to query themselves even if they lack a root -*/ -memcached_return_t memcached_server_remove(memcached_server_st *st_ptr) -{ - uint32_t x, host_index; - memcached_st *root= (memcached_st *)st_ptr->root; - memcached_server_st *list; - - if (root == NULL) - return MEMCACHED_FAILURE; - - list= memcached_server_list(root); - - for (x= 0, host_index= 0; x < memcached_server_count(root); x++) - { - if (strncmp(list[x].hostname, st_ptr->hostname, MEMCACHED_MAX_HOST_LENGTH) != 0 || list[x].port != st_ptr->port) - { - if (host_index != x) - memcpy(list+host_index, list+x, sizeof(memcached_server_st)); - host_index++; - } - } - root->number_of_hosts= host_index; - - if (st_ptr->address_info) - { - freeaddrinfo(st_ptr->address_info); - st_ptr->address_info= NULL; - } - run_distribution(root); - - return MEMCACHED_SUCCESS; -} - -memcached_server_st *memcached_server_list_append(memcached_server_st *ptr, - const char *hostname, in_port_t port, - memcached_return_t *error) -{ - return memcached_server_list_append_with_weight(ptr, hostname, port, 0, error); -} - -memcached_server_st *memcached_server_list_append_with_weight(memcached_server_st *ptr, - const char *hostname, in_port_t port, - uint32_t weight, - memcached_return_t *error) -{ - unsigned int count; - memcached_server_instance_st *new_host_list; - - if (hostname == NULL || error == NULL) - return NULL; - - if (! port) - port= MEMCACHED_DEFAULT_PORT; - - /* Increment count for hosts */ - count= 1; - if (ptr != NULL) - { - count+= memcached_servers_count(ptr); - } - - new_host_list= (memcached_server_instance_st *)realloc(ptr, sizeof(memcached_server_instance_st) * count); - if (!new_host_list) - { - *error= MEMCACHED_MEMORY_ALLOCATION_FAILURE; - return NULL; - } - - /* TODO: Check return type */ - memcached_server_create_with(NULL, &new_host_list[count-1], hostname, port, weight, MEMCACHED_CONNECTION_TCP); - - /* Backwards compatibility hack */ - memcached_servers_set_count(new_host_list, count); - - *error= MEMCACHED_SUCCESS; - return new_host_list; -}