X-Git-Url: https://git.m6w6.name/?a=blobdiff_plain;f=libmemcached%2Fmemcached.c;h=4e0308a043df72b29bdc43800a475d04942c73d7;hb=93723c9e5e41d8d9c41f610623060b28b4ec8ae3;hp=2709c084e0210b33333af70a04ad66ec1632ac7f;hpb=b41c57c2f84855cfd32a7b14008ca0aceb92f32b;p=m6w6%2Flibmemcached diff --git a/libmemcached/memcached.c b/libmemcached/memcached.c index 2709c084..4e0308a0 100644 --- a/libmemcached/memcached.c +++ b/libmemcached/memcached.c @@ -1,13 +1,102 @@ -/* - Memcached library -*/ +/* 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" +static const memcached_st global_copy= { + .state= { + .is_purging= false, + .is_processing_input= false, + }, + .flags= { + .auto_eject_hosts= false, + .binary_protocol= false, + .buffer_requests= false, + .cork= false, + .hash_with_prefix_key= false, + .ketama_weighted= false, + .no_block= false, + .no_reply= false, + .randomize_replica_read= false, + .reuse_memory= false, + .support_cas= false, + .tcp_nodelay= false, + .use_cache_lookups= false, + .use_sort_hosts= false, + .use_udp= false, + .verify_key= false + } +}; + +static inline bool _memcached_init(memcached_st *self) +{ + self->state= global_copy.state; + self->flags= global_copy.flags; + + self->distribution= MEMCACHED_DISTRIBUTION_MODULA; + + hashkit_st *hash_ptr; + hash_ptr= hashkit_create(&self->hashkit); + if (! hash_ptr) + return false; + + self->continuum_points_counter= 0; + + self->number_of_hosts= 0; + self->servers= NULL; + self->last_disconnected_server= NULL; + + self->snd_timeout= 0; + self->rcv_timeout= 0; + self->server_failure_limit= 0; + + /* TODO, Document why we picked these defaults */ + self->io_msg_watermark= 500; + self->io_bytes_watermark= 65 * 1024; + + self->io_key_prefetch= 0; + self->cached_errno= 0; + self->poll_timeout= MEMCACHED_DEFAULT_TIMEOUT; + self->connect_timeout= MEMCACHED_DEFAULT_TIMEOUT; + self->retry_timeout= 0; + self->continuum_count= 0; + + self->send_size= -1; + self->recv_size= -1; + + self->user_data= NULL; + self->next_distribution_rebuild= 0; + self->prefix_key_length= 0; + self->number_of_replicas= 0; + hash_ptr= hashkit_create(&self->distribution_hashkit); + if (! hash_ptr) + return false; + self->continuum= NULL; + + self->allocators= memcached_allocators_return_default(); + + self->on_clone= NULL; + self->on_cleanup= NULL; + self->get_key_failure= NULL; + self->delete_trigger= NULL; + self->callbacks= NULL; + + return true; +} + memcached_st *memcached_create(memcached_st *ptr) { if (ptr == NULL) { - ptr= (memcached_st *)calloc(1, sizeof(memcached_st)); + ptr= (memcached_st *)malloc(sizeof(memcached_st)); if (! ptr) { @@ -18,61 +107,34 @@ memcached_st *memcached_create(memcached_st *ptr) } else { - memset(ptr, 0, sizeof(memcached_st)); + ptr->options.is_allocated= false; } - ptr->options.is_initialized= true; +#if 0 + memcached_set_purging(ptr, false); + memcached_set_processing_input(ptr, false); +#endif - memcached_set_memory_allocators(ptr, NULL, NULL, NULL, NULL); + if (! _memcached_init(ptr)) + { + memcached_free(ptr); + return NULL; + } if (! memcached_result_create(ptr, &ptr->result)) { memcached_free(ptr); return NULL; } - ptr->poll_timeout= MEMCACHED_DEFAULT_TIMEOUT; - ptr->connect_timeout= MEMCACHED_DEFAULT_TIMEOUT; - ptr->retry_timeout= 0; - ptr->distribution= MEMCACHED_DISTRIBUTION_MODULA; - - /* TODO, Document why we picked these defaults */ - ptr->io_msg_watermark= 500; - ptr->io_bytes_watermark= 65 * 1024; WATCHPOINT_ASSERT_INITIALIZED(&ptr->result); return ptr; } -void server_list_free(memcached_st *ptr, memcached_server_st *servers) -{ - uint32_t x; - - if (servers == NULL) - return; - - for (x= 0; x < memcached_servers_count(servers); x++) - { - if (servers[x].address_info) - { - freeaddrinfo(servers[x].address_info); - servers[x].address_info= NULL; - } - } - - if (ptr) - { - ptr->call_free(ptr, servers); - } - else - { - free(servers); - } -} - void memcached_servers_reset(memcached_st *ptr) { - server_list_free(ptr, memcached_server_list(ptr)); + memcached_server_list_free(memcached_server_list(ptr)); memcached_server_list_set(ptr, NULL); ptr->number_of_hosts= 0; @@ -84,22 +146,18 @@ void memcached_free(memcached_st *ptr) { /* If we have anything open, lets close it now */ memcached_quit(ptr); - server_list_free(ptr, memcached_server_list(ptr)); + memcached_server_list_free(memcached_server_list(ptr)); memcached_result_free(&ptr->result); if (ptr->on_cleanup) ptr->on_cleanup(ptr); if (ptr->continuum) - ptr->call_free(ptr, ptr->continuum); + libmemcached_free(ptr, ptr->continuum); if (memcached_is_allocated(ptr)) { - ptr->call_free(ptr, ptr); - } - else - { - ptr->options.is_initialized= false; + libmemcached_free(ptr, ptr); } } @@ -108,7 +166,7 @@ void memcached_free(memcached_st *ptr) If source is NULL the call is the same as if a memcached_create() was called. */ -memcached_st *memcached_clone(memcached_st *clone, memcached_st *source) +memcached_st *memcached_clone(memcached_st *clone, const memcached_st *source) { memcached_return_t rc= MEMCACHED_SUCCESS; memcached_st *new_clone; @@ -133,8 +191,23 @@ memcached_st *memcached_clone(memcached_st *clone, memcached_st *source) new_clone->connect_timeout= source->connect_timeout; new_clone->retry_timeout= source->retry_timeout; new_clone->distribution= source->distribution; - new_clone->hash= source->hash; - new_clone->distribution_hash= source->distribution_hash; + + hashkit_st *hash_ptr; + + hash_ptr= hashkit_clone(&new_clone->hashkit, &source->hashkit); + if (! hash_ptr) + { + memcached_free(new_clone); + return NULL; + } + + hash_ptr= hashkit_clone(&new_clone->distribution_hashkit, &source->distribution_hashkit); + if (! hash_ptr) + { + memcached_free(new_clone); + return NULL; + } + new_clone->user_data= source->user_data; new_clone->snd_timeout= source->snd_timeout; @@ -142,10 +215,9 @@ memcached_st *memcached_clone(memcached_st *clone, memcached_st *source) new_clone->on_clone= source->on_clone; new_clone->on_cleanup= source->on_cleanup; - new_clone->call_free= source->call_free; - new_clone->call_malloc= source->call_malloc; - new_clone->call_realloc= source->call_realloc; - new_clone->call_calloc= source->call_calloc; + + new_clone->allocators= source->allocators; + new_clone->get_key_failure= source->get_key_failure; new_clone->delete_trigger= source->delete_trigger; new_clone->server_failure_limit= source->server_failure_limit; @@ -154,8 +226,8 @@ memcached_st *memcached_clone(memcached_st *clone, memcached_st *source) new_clone->io_key_prefetch= source->io_key_prefetch; new_clone->number_of_replicas= source->number_of_replicas; - if (memcached_server_list(source)) - rc= memcached_server_push(new_clone, memcached_server_list(source)); + if (memcached_server_count(source)) + rc= memcached_push(new_clone, source); if (rc != MEMCACHED_SUCCESS) { @@ -165,7 +237,7 @@ memcached_st *memcached_clone(memcached_st *clone, memcached_st *source) } - if (source->prefix_key[0] != 0) + if (source->prefix_key_length) { strcpy(new_clone->prefix_key, source->prefix_key); new_clone->prefix_key_length= source->prefix_key_length; @@ -181,12 +253,12 @@ memcached_st *memcached_clone(memcached_st *clone, memcached_st *source) } if (source->on_clone) - source->on_clone(source, new_clone); + source->on_clone(new_clone, source); return new_clone; } -void *memcached_get_user_data(memcached_st *ptr) +void *memcached_get_user_data(const memcached_st *ptr) { return ptr->user_data; } @@ -198,3 +270,18 @@ void *memcached_set_user_data(memcached_st *ptr, void *data) return ret; } + +memcached_return_t memcached_push(memcached_st *destination, const memcached_st *source) +{ + return memcached_server_push(destination, source->servers); +} + +inline memcached_server_write_instance_st memcached_server_instance_fetch(memcached_st *ptr, uint32_t server_key) +{ + return &ptr->servers[server_key]; +} + +inline memcached_server_instance_st memcached_server_instance_by_position(const memcached_st *ptr, uint32_t server_key) +{ + return &ptr->servers[server_key]; +}