X-Git-Url: https://git.m6w6.name/?a=blobdiff_plain;f=libmemcached%2Fmemcached_fetch.c;h=d3f012197d6a177a30356b90161650c4739c7653;hb=bbd526419742d4a9bb99fcb1a56a80969ebe90fb;hp=7f8a6b52ba622e0f0a5e3f39edc453d7656635eb;hpb=fa356c992dadfbd21eea474456d59dac4eba9495;p=awesomized%2Flibmemcached diff --git a/libmemcached/memcached_fetch.c b/libmemcached/memcached_fetch.c index 7f8a6b52..d3f01219 100644 --- a/libmemcached/memcached_fetch.c +++ b/libmemcached/memcached_fetch.c @@ -8,57 +8,35 @@ char *memcached_fetch(memcached_st *ptr, char *key, size_t *key_length, { memcached_result_st *result_buffer= &ptr->result; - if (ptr->flags & MEM_USE_UDP) + unlikely (ptr->flags & MEM_USE_UDP) { *error= MEMCACHED_NOT_SUPPORTED; return NULL; } - while (ptr->cursor_server < ptr->number_of_hosts) - { - char buffer[MEMCACHED_DEFAULT_COMMAND_SIZE]; - - if (memcached_server_response_count(&ptr->hosts[ptr->cursor_server]) == 0) - { - ptr->cursor_server++; - continue; - } + result_buffer= memcached_fetch_result(ptr, result_buffer, error); - *error= memcached_response(&ptr->hosts[ptr->cursor_server], buffer, MEMCACHED_DEFAULT_COMMAND_SIZE, result_buffer); - - if (*error == MEMCACHED_END) /* END means that we move on to the next */ - { - memcached_server_response_reset(&ptr->hosts[ptr->cursor_server]); - ptr->cursor_server++; - continue; - } - else if (*error == MEMCACHED_SUCCESS) - { - *value_length= memcached_string_length(&result_buffer->value); - - if (key) - { - strncpy(key, result_buffer->key, result_buffer->key_length); - *key_length= result_buffer->key_length; - } + if (result_buffer == NULL || *error != MEMCACHED_SUCCESS) + { + WATCHPOINT_ASSERT(result_buffer == NULL); + *value_length= 0; + return NULL; + } - if (result_buffer->flags) - *flags= result_buffer->flags; - else - *flags= 0; + *value_length= memcached_string_length(&result_buffer->value); - return memcached_string_c_copy(&result_buffer->value); - } - else - { - *value_length= 0; - return NULL; - } + if (key) + { + strncpy(key, result_buffer->key, result_buffer->key_length); + *key_length= result_buffer->key_length; } - ptr->cursor_server= 0; - *value_length= 0; - return NULL; + if (result_buffer->flags) + *flags= result_buffer->flags; + else + *flags= 0; + + return memcached_string_c_copy(&result_buffer->value); } memcached_result_st *memcached_fetch_result(memcached_st *ptr, @@ -67,7 +45,7 @@ memcached_result_st *memcached_fetch_result(memcached_st *ptr, { memcached_server_st *server; - if (ptr->flags & MEM_USE_UDP) + unlikely (ptr->flags & MEM_USE_UDP) { *error= MEMCACHED_NOT_SUPPORTED; return NULL; @@ -77,7 +55,8 @@ memcached_result_st *memcached_fetch_result(memcached_st *ptr, if ((result= memcached_result_create(ptr, NULL)) == NULL) return NULL; - while ((server = memcached_io_get_readable_server(ptr)) != NULL) { + while ((server = memcached_io_get_readable_server(ptr)) != NULL) + { char buffer[MEMCACHED_DEFAULT_COMMAND_SIZE]; *error= memcached_response(server, buffer, sizeof(buffer), result); @@ -98,3 +77,26 @@ memcached_result_st *memcached_fetch_result(memcached_st *ptr, return NULL; } +memcached_return memcached_fetch_execute(memcached_st *ptr, + memcached_execute_function *callback, + void *context, + unsigned int number_of_callbacks) +{ + memcached_result_st *result= &ptr->result; + memcached_return rc= MEMCACHED_FAILURE; + unsigned int x; + + while ((result= memcached_fetch_result(ptr, result, &rc)) != NULL) + { + if (rc == MEMCACHED_SUCCESS) + { + for (x= 0; x < number_of_callbacks; x++) + { + rc= (*callback[x])(ptr, result, context); + if (rc != MEMCACHED_SUCCESS) + break; + } + } + } + return rc; +}