X-Git-Url: https://git.m6w6.name/?a=blobdiff_plain;f=libmemcached%2Fserver_list.cc;h=537e9edd8dfb04e7817a22e369793b6072d8fc3d;hb=dbb1511b2cb2e89ab09278e8c624c5050e3ff93d;hp=a599e0b6166d5d085b8cc0332610fce5bab1ab8d;hpb=3c4d734f56530d43520f385bff97162c04ac81ac;p=awesomized%2Flibmemcached diff --git a/libmemcached/server_list.cc b/libmemcached/server_list.cc index a599e0b6..537e9edd 100644 --- a/libmemcached/server_list.cc +++ b/libmemcached/server_list.cc @@ -46,7 +46,6 @@ memcached_server_list_append_with_weight(memcached_server_list_st ptr, memcached_return_t *error) { uint32_t count; - memcached_server_list_st new_host_list; memcached_return_t unused; if (error == NULL) @@ -66,6 +65,7 @@ memcached_server_list_append_with_weight(memcached_server_list_st ptr, port= MEMCACHED_DEFAULT_PORT; } + /* Increment count for hosts */ count= 1; if (ptr != NULL) @@ -73,17 +73,23 @@ memcached_server_list_append_with_weight(memcached_server_list_st ptr, count+= memcached_server_list_count(ptr); } - new_host_list= (memcached_server_write_instance_st)realloc(ptr, sizeof(memcached_server_st) * count); - if (not new_host_list) + memcached_server_list_st new_host_list= (memcached_server_st*)realloc(ptr, sizeof(memcached_server_st) * count); + if (new_host_list == NULL) { +#if 0 *error= memcached_set_error(*ptr, MEMCACHED_MEMORY_ALLOCATION_FAILURE, MEMCACHED_AT); +#endif return NULL; } + memcached_string_t _hostname= { memcached_string_make_from_cstr(hostname) }; /* @todo Check return type */ - if (not __server_create_with(NULL, &new_host_list[count-1], hostname, port, weight, port ? MEMCACHED_CONNECTION_TCP : MEMCACHED_CONNECTION_UNIX_SOCKET)) + if (__server_create_with(NULL, &new_host_list[count-1], _hostname, port, weight, port ? MEMCACHED_CONNECTION_TCP : MEMCACHED_CONNECTION_UNIX_SOCKET) == NULL) { +#if 0 *error= memcached_set_errno(*ptr, MEMCACHED_MEMORY_ALLOCATION_FAILURE, MEMCACHED_AT); +#endif + free(new_host_list); return NULL; } @@ -114,26 +120,43 @@ uint32_t memcached_server_list_count(const memcached_server_list_st self) : self->number_of_hosts; } -memcached_server_st *memcached_server_list(const memcached_st *self) +uint32_t memcached_instance_list_count(const memcached_st* self) { - return self->servers; + return (self == NULL) + ? 0 + : self->number_of_hosts; } -void memcached_server_list_set(memcached_st *self, memcached_server_st *list) +void memcached_instance_set(memcached_st* memc, org::libmemcached::Instance* list, const uint32_t host_list_size) { - self->servers= list; + memc->servers= list; + memc->number_of_hosts= host_list_size; } void memcached_server_list_free(memcached_server_list_st self) { - if (not self) - return; - - for (uint32_t x= 0; x < memcached_server_list_count(self); x++) + if (self) { - assert_msg(not memcached_is_allocated(&self[x]), "You have called memcached_server_list_free(), but you did not pass it a valid memcached_server_list_st"); - __server_free(&self[x]); + for (uint32_t x= 0; x < memcached_server_list_count(self); x++) + { + assert_msg(not memcached_is_allocated(&self[x]), "You have called memcached_server_list_free(), but you did not pass it a valid memcached_server_list_st"); + __server_free(&self[x]); + } + + libmemcached_free(self->root, self); } +} - libmemcached_free(self->root, self); +void memcached_instance_list_free(org::libmemcached::Instance* self, uint32_t instance_count) +{ + if (self) + { + for (uint32_t x= 0; x < instance_count; x++) + { + assert_msg(memcached_is_allocated(&self[x]) == false, "You have called memcached_server_list_free(), but you did not pass it a valid memcached_server_list_st"); + __instance_free(&self[x]); + } + + libmemcached_free(self->root, self); + } }