X-Git-Url: https://git.m6w6.name/?a=blobdiff_plain;f=src%2Flibmemcached%2Finstance.cc;h=972f6b6f6ae5e3f6e3cfacd4f2b4aa3f82a023a9;hb=2f289c64f625962d945ec3bee80f36bc5c61ee35;hp=df08c6066b3eabcb32959e39fd66d654d8d76200;hpb=5e760300d15ef4c5b7eed3fb9f37920ebca2f6ec;p=awesomized%2Flibmemcached diff --git a/src/libmemcached/instance.cc b/src/libmemcached/instance.cc index df08c606..972f6b6f 100644 --- a/src/libmemcached/instance.cc +++ b/src/libmemcached/instance.cc @@ -1,170 +1,130 @@ -/* vim:expandtab:shiftwidth=2:tabstop=2:smarttab: - * - * Libmemcached library - * - * Copyright (C) 2011 Data Differential, http://datadifferential.com/ - * Copyright (C) 2006-2009 Brian Aker All rights reserved. - * - * Redistribution and use in source and binary forms, with or without - * modification, are permitted provided that the following conditions are - * met: - * - * * Redistributions of source code must retain the above copyright - * notice, this list of conditions and the following disclaimer. - * - * * Redistributions in binary form must reproduce the above - * copyright notice, this list of conditions and the following disclaimer - * in the documentation and/or other materials provided with the - * distribution. - * - * * The names of its contributors may not be used to endorse or - * promote products derived from this software without specific prior - * written permission. - * - * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS - * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT - * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR - * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT - * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, - * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT - * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, - * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY - * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT - * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE - * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. - * - */ - -#include - -static inline void _server_init(memcached_instance_st* self, Memcached *root, - const memcached_string_t& hostname, - in_port_t port, - uint32_t weight, memcached_connection_t type) -{ - self->options.is_shutting_down= false; - self->options.is_dead= false; - self->options.ready= false; - self->_events= 0; - self->_revents= 0; - self->cursor_active_= 0; - self->port_= port; - self->fd= INVALID_SOCKET; - self->io_bytes_sent= 0; - self->request_id= 0; - self->server_failure_counter= 0; - self->server_failure_counter_query_id= 0; - self->server_timeout_counter= 0; - self->server_timeout_counter_query_id= 0; - self->weight= weight ? weight : 1; // 1 is the default weight value - self->io_wait_count.read= 0; - self->io_wait_count.write= 0; - self->io_wait_count.timeouts= 0; - self->io_wait_count._bytes_read= 0; - self->major_version= UINT8_MAX; - self->micro_version= UINT8_MAX; - self->minor_version= UINT8_MAX; - self->type= type; - self->error_messages= NULL; - self->read_ptr= self->read_buffer; - self->read_buffer_length= 0; - self->write_buffer_offset= 0; - self->address_info= NULL; - self->address_info_next= NULL; - - self->state= MEMCACHED_SERVER_STATE_NEW; - self->next_retry= 0; - - self->root= root; - if (root) - { - self->version= ++root->server_info.version; - } - else - { - self->version= UINT_MAX; +/* + +--------------------------------------------------------------------+ + | libmemcached - C/C++ Client Library for memcached | + +--------------------------------------------------------------------+ + | Redistribution and use in source and binary forms, with or without | + | modification, are permitted under the terms of the BSD license. | + | You should have received a copy of the license in a bundled file | + | named LICENSE; in case you did not receive a copy you can review | + | the terms online at: https://opensource.org/licenses/BSD-3-Clause | + +--------------------------------------------------------------------+ + | Copyright (c) 2006-2014 Brian Aker https://datadifferential.com/ | + | Copyright (c) 2020 Michael Wallner | + +--------------------------------------------------------------------+ +*/ + +#include "libmemcached/common.h" +#include "p9y/socket.hpp" + +static inline void _server_init(memcached_instance_st *self, Memcached *root, + const memcached_string_t &hostname, in_port_t port, uint32_t weight, + memcached_connection_t type) { + self->options.is_shutting_down = false; + self->options.is_dead = false; + self->options.ready = false; + self->_events = 0; + self->_revents = 0; + self->cursor_active_ = 0; + self->port_ = port; + self->fd = INVALID_SOCKET; + self->io_bytes_sent = 0; + self->request_id = 0; + self->server_failure_counter = 0; + self->server_failure_counter_query_id = 0; + self->server_timeout_counter = 0; + self->server_timeout_counter_query_id = 0; + self->weight = weight ? weight : 1; // 1 is the default weight value + self->io_wait_count.read = 0; + self->io_wait_count.write = 0; + self->io_wait_count.timeouts = 0; + self->io_wait_count._bytes_read = 0; + self->major_version = UINT8_MAX; + self->micro_version = UINT8_MAX; + self->minor_version = UINT8_MAX; + self->type = type; + self->error_messages = NULL; + self->read_ptr = self->read_buffer; + self->read_buffer_length = 0; + self->write_buffer_offset = 0; + self->address_info = NULL; + self->address_info_next = NULL; + + self->state = MEMCACHED_SERVER_STATE_NEW; + self->next_retry = 0; + + self->root = root; + if (root) { + self->version = ++root->server_info.version; + } else { + self->version = UINT_MAX; } - self->limit_maxbytes= 0; + self->limit_maxbytes = 0; self->hostname(hostname); } -static memcached_instance_st* _server_create(memcached_instance_st* self, const memcached_st *memc) -{ - if (self == NULL) - { - self= libmemcached_xmalloc(memc, memcached_instance_st); +static memcached_instance_st *_server_create(memcached_instance_st *self, + const memcached_st *memc) { + if (self == NULL) { + self = libmemcached_xmalloc(memc, memcached_instance_st); - if (self == NULL) - { + if (self == NULL) { return NULL; /* MEMCACHED_MEMORY_ALLOCATION_FAILURE */ } - self->options.is_allocated= true; - } - else - { - self->options.is_allocated= false; + self->options.is_allocated = true; + } else { + self->options.is_allocated = false; } - self->options.is_initialized= true; + self->options.is_initialized = true; return self; } -void memcached_instance_st::events(short arg) -{ - if ((_events | arg) == _events) - { +void memcached_instance_st::events(short arg) { + if ((_events | arg) == _events) { return; } - _events|= arg; + _events |= arg; } -void memcached_instance_st::revents(short arg) -{ - if (arg) - { - options.ready= true; +void memcached_instance_st::revents(short arg) { + if (arg) { + options.ready = true; } - _revents= arg; - _events&= short(~arg); + _revents = arg; + _events &= short(~arg); } -memcached_instance_st* __instance_create_with(memcached_st *memc, - memcached_instance_st* self, - const memcached_string_t& hostname, - const in_port_t port, - uint32_t weight, - const memcached_connection_t type) -{ - if (memcached_is_valid_servername(hostname) == false) - { - memcached_set_error(*memc, MEMCACHED_INVALID_ARGUMENTS, MEMCACHED_AT, memcached_literal_param("Invalid hostname provided")); +memcached_instance_st *__instance_create_with(memcached_st *memc, memcached_instance_st *self, + const memcached_string_t &hostname, + const in_port_t port, uint32_t weight, + const memcached_connection_t type) { + if (memcached_is_valid_servername(hostname) == false) { + memcached_set_error(*memc, MEMCACHED_INVALID_ARGUMENTS, MEMCACHED_AT, + memcached_literal_param("Invalid hostname provided")); return NULL; } - self= _server_create(self, memc); + self = _server_create(self, memc); - if (self == NULL) - { + if (self == NULL) { return NULL; } _server_init(self, const_cast(memc), hostname, port, weight, type); - if (memc and memcached_is_udp(memc)) - { - self->write_buffer_offset= UDP_DATAGRAM_HEADER_LENGTH; + if (memc and memcached_is_udp(memc)) { + self->write_buffer_offset = UDP_DATAGRAM_HEADER_LENGTH; memcached_io_init_udp_header(self, 0); } return self; } -void __instance_free(memcached_instance_st* self) -{ +void __instance_free(memcached_instance_st *self) { memcached_quit_server(self, false); self->clear_addrinfo(); @@ -172,47 +132,36 @@ void __instance_free(memcached_instance_st* self) memcached_error_free(*self); - if (memcached_is_allocated(self)) - { + if (memcached_is_allocated(self)) { libmemcached_free(self->root, self); - } - else - { - self->options.is_initialized= false; + } else { + self->options.is_initialized = false; } } -void memcached_instance_free(memcached_instance_st* self) -{ - if (self) - { +void memcached_instance_free(memcached_instance_st *self) { + if (self) { __instance_free(self); } } -memcached_return_t memcached_server_cursor(const memcached_st* shell, - const memcached_server_fn *callback, - void *context, - uint32_t number_of_callbacks) -{ - const Memcached* memc= memcached2Memcached(shell); +memcached_return_t memcached_server_cursor(const memcached_st *shell, + const memcached_server_fn *callback, void *context, + uint32_t number_of_callbacks) { + const Memcached *memc = memcached2Memcached(shell); memcached_return_t rc; - if (memcached_failed(rc= initialize_const_query(memc))) - { + if (memcached_failed(rc = initialize_const_query(memc))) { return rc; } - size_t errors= 0; - for (uint32_t x= 0; x < memcached_instance_list_count(memc); x++) - { - memcached_instance_st* instance= memcached_instance_by_position(memc, x); + size_t errors = 0; + for (uint32_t x = 0; x < memcached_instance_list_count(memc); x++) { + memcached_instance_st *instance = memcached_instance_by_position(memc, x); - for (uint32_t y= 0; y < number_of_callbacks; y++) - { - memcached_return_t ret= (*callback[y])(memc, instance, context); + for (uint32_t y = 0; y < number_of_callbacks; y++) { + memcached_return_t ret = (*callback[y])(memc, instance, context); - if (memcached_failed(ret)) - { + if (memcached_failed(ret)) { errors++; continue; } @@ -223,139 +172,112 @@ memcached_return_t memcached_server_cursor(const memcached_st* shell, } memcached_return_t memcached_server_execute(memcached_st *memc, - memcached_server_execute_fn callback, - void *context) -{ - if (callback == NULL) - { + memcached_server_execute_fn callback, void *context) { + if (callback == NULL) { return MEMCACHED_INVALID_ARGUMENTS; } - bool some_errors= false;; - for (uint32_t x= 0; x < memcached_instance_list_count(memc); x++) - { - memcached_instance_st* instance= memcached_instance_fetch(memc, x); + bool some_errors = false; + ; + for (uint32_t x = 0; x < memcached_instance_list_count(memc); x++) { + memcached_instance_st *instance = memcached_instance_fetch(memc, x); - memcached_return_t rc= (*callback)(memc, instance, context); - if (rc == MEMCACHED_INVALID_ARGUMENTS) - { + memcached_return_t rc = (*callback)(memc, instance, context); + if (rc == MEMCACHED_INVALID_ARGUMENTS) { return rc; - } - else if (memcached_fatal(rc)) - { - some_errors= true; + } else if (memcached_fatal(rc)) { + some_errors = true; } } - (void)some_errors; + (void) some_errors; return MEMCACHED_SUCCESS; } -const memcached_instance_st * memcached_server_by_key(memcached_st *shell, - const char *key, - size_t key_length, - memcached_return_t *error) -{ - Memcached* memc= memcached2Memcached(shell); +const memcached_instance_st *memcached_server_by_key(memcached_st *shell, const char *key, + size_t key_length, memcached_return_t *error) { + Memcached *memc = memcached2Memcached(shell); memcached_return_t unused; - if (error == NULL) - { - error= &unused; + if (error == NULL) { + error = &unused; } - memcached_return_t rc; - if (memcached_failed(rc= initialize_const_query(memc))) - { - *error= rc; + if (memcached_failed(rc = initialize_const_query(memc))) { + *error = rc; return NULL; } - if (memcached_failed((memcached_key_test(*memc, (const char **)&key, &key_length, 1)))) - { - *error= memcached_last_error(memc); + if (memcached_failed((memcached_key_test(*memc, (const char **) &key, &key_length, 1)))) { + *error = memcached_last_error(memc); return NULL; } - uint32_t server_key= memcached_generate_hash(memc, key, key_length); + uint32_t server_key = memcached_generate_hash(memc, key, key_length); return memcached_instance_by_position(memc, server_key); } /* If we do not have a valid object to clone from, we toss an error. */ -static memcached_instance_st* memcached_instance_clone(memcached_instance_st* source) -{ +static memcached_instance_st *memcached_instance_clone(memcached_instance_st *source) { /* We just do a normal create if source is missing */ - if (source == NULL) - { + if (source == NULL) { return NULL; } - memcached_string_t hostname_= { memcached_string_make_from_cstr(source->hostname()) }; - return __instance_create_with(source->root, - NULL, - hostname_, - source->port(), source->weight, + memcached_string_t hostname_ = {memcached_string_make_from_cstr(source->hostname())}; + return __instance_create_with(source->root, NULL, hostname_, source->port(), source->weight, source->type); } -void set_last_disconnected_host(memcached_instance_st* self) -{ +void set_last_disconnected_host(memcached_instance_st *self) { assert(self->root); - if (self->root) - { - if (memcached_server_get_last_disconnect(self->root) and - memcached_server_get_last_disconnect(self->root)->version == self->version) + if (self->root) { + if (memcached_server_get_last_disconnect(self->root) + and memcached_server_get_last_disconnect(self->root)->version == self->version) { return; } // const_cast - memcached_st *root= (memcached_st *)self->root; + memcached_st *root = (memcached_st *) self->root; - memcached_instance_free((memcached_instance_st*)(root->last_disconnected_server)); + memcached_instance_free((memcached_instance_st *) (root->last_disconnected_server)); // We set is_parsing so that no lookup happens - root->state.is_parsing= true; - root->last_disconnected_server= memcached_instance_clone(self); - root->state.is_parsing= false; + root->state.is_parsing = true; + root->last_disconnected_server = memcached_instance_clone(self); + root->state.is_parsing = false; - ((memcached_instance_st*)memcached_server_get_last_disconnect(root))->version= self->version; + ((memcached_instance_st *) memcached_server_get_last_disconnect(root))->version = self->version; } } -const memcached_instance_st * memcached_server_get_last_disconnect(const memcached_st *shell) -{ - const Memcached* self= memcached2Memcached(shell); - if (self) - { - return (const memcached_instance_st *)self->last_disconnected_server; +const memcached_instance_st *memcached_server_get_last_disconnect(const memcached_st *shell) { + const Memcached *self = memcached2Memcached(shell); + if (self) { + return (const memcached_instance_st *) self->last_disconnected_server; } return 0; } -void memcached_instance_next_retry(const memcached_instance_st * self, const time_t absolute_time) -{ +void memcached_instance_next_retry(const memcached_instance_st *self, const time_t absolute_time) { WATCHPOINT_ASSERT(self); - if (self) - { - ((memcached_instance_st*)self)->next_retry= absolute_time; + if (self) { + ((memcached_instance_st *) self)->next_retry = absolute_time; } } -bool memcached_instance_st::valid() const -{ - if (fd == INVALID_SOCKET) - { +bool memcached_instance_st::valid() const { + if (fd == INVALID_SOCKET) { return false; } return true; } -bool memcached_instance_st::is_shutting_down() const -{ +bool memcached_instance_st::is_shutting_down() const { return options.is_shutting_down; }