X-Git-Url: https://git.m6w6.name/?a=blobdiff_plain;f=lib%2Fmemcached_delete.c;h=ff085453ab63a0fc602df81205b3348c59ddd118;hb=9e9572af3e9be02dd3de5c71bf1665217a770e1b;hp=d24c0f82a0b7d16ffff58e922a09a05f97817010;hpb=875200dc0c5668bba4f8a68b487f9d2f2a1eedb0;p=m6w6%2Flibmemcached diff --git a/lib/memcached_delete.c b/lib/memcached_delete.c index d24c0f82..ff085453 100644 --- a/lib/memcached_delete.c +++ b/lib/memcached_delete.c @@ -1,19 +1,20 @@ -#include +#include "common.h" memcached_return memcached_delete(memcached_st *ptr, char *key, size_t key_length, time_t expiration) { - size_t send_length; + size_t send_length, sent_length; memcached_return rc; char buffer[MEMCACHED_DEFAULT_COMMAND_SIZE]; unsigned int server_key; - rc= memcached_connect(ptr); + LIBMEMCACHED_MEMCACHED_DELETE_START(); - if (rc != MEMCACHED_SUCCESS) + server_key= memcached_generate_hash(ptr, key, key_length); + + if ((rc= memcached_connect(ptr, server_key)) != MEMCACHED_SUCCESS) return rc; - server_key= memcached_generate_hash(key, key_length) % ptr->number_of_hosts; if (expiration) send_length= snprintf(buffer, MEMCACHED_DEFAULT_COMMAND_SIZE, @@ -23,8 +24,27 @@ memcached_return memcached_delete(memcached_st *ptr, char *key, size_t key_lengt send_length= snprintf(buffer, MEMCACHED_DEFAULT_COMMAND_SIZE, "delete %.*s\r\n", (int)key_length, key); - if ((write(ptr->hosts[server_key].fd, buffer, send_length) == -1)) - return MEMCACHED_WRITE_FAILURE; + if (send_length >= MEMCACHED_DEFAULT_COMMAND_SIZE) + { + rc= MEMCACHED_WRITE_FAILURE; + goto error; + } + + sent_length= memcached_io_write(ptr, server_key, buffer, send_length, 1); + + if (sent_length == -1 || sent_length != send_length) + { + rc= MEMCACHED_WRITE_FAILURE; + goto error; + } + + rc= memcached_response(ptr, buffer, MEMCACHED_DEFAULT_COMMAND_SIZE, server_key); + + if (rc == MEMCACHED_DELETED) + rc= MEMCACHED_SUCCESS; + + LIBMEMCACHED_MEMCACHED_DELETE_END(); - return memcached_response(ptr, buffer, MEMCACHED_DEFAULT_COMMAND_SIZE, server_key); +error: + return rc; }