X-Git-Url: https://git.m6w6.name/?a=blobdiff_plain;f=libmemcached%2Fauto.c;h=608324df23ff8d99acde7960997ab74cd8c824a1;hb=3716f9e7bd4b70a234a43aecb6491a65aac0e07a;hp=8d3e1bc7952e11ff11ebb6cb789788a1e4ccedd2;hpb=85062bec3fe9000e4da8e0a08cea88ebbccc20b2;p=m6w6%2Flibmemcached diff --git a/libmemcached/auto.c b/libmemcached/auto.c index 8d3e1bc7..608324df 100644 --- a/libmemcached/auto.c +++ b/libmemcached/auto.c @@ -11,7 +11,7 @@ #include "common.h" -static memcached_return_t memcached_auto(memcached_st *ptr, +static memcached_return_t text_incr_decr(memcached_st *ptr, const char *verb, const char *master_key, size_t master_key_length, const char *key, size_t key_length, @@ -25,7 +25,7 @@ static memcached_return_t memcached_auto(memcached_st *ptr, memcached_server_instance_st *instance; bool no_reply= ptr->flags.no_reply; - unlikely (ptr->hosts == NULL || memcached_server_count(ptr) == 0) + unlikely (memcached_server_count(ptr) == 0) return MEMCACHED_NO_SERVERS; if (ptr->flags.verify_key && (memcached_key_test((const char **)&key, &key_length, 1) == MEMCACHED_BAD_KEY_PROVIDED)) @@ -35,14 +35,15 @@ static memcached_return_t memcached_auto(memcached_st *ptr, instance= memcached_server_instance_fetch(ptr, server_key); send_length= (size_t)snprintf(buffer, MEMCACHED_DEFAULT_COMMAND_SIZE, - "%s %s%.*s %" PRIu64 "%s\r\n", verb, + "%s %.*s%.*s %" PRIu64 "%s\r\n", verb, + (int)ptr->prefix_key_length, ptr->prefix_key, (int)key_length, key, offset, no_reply ? " noreply" : ""); unlikely (send_length >= MEMCACHED_DEFAULT_COMMAND_SIZE) return MEMCACHED_WRITE_FAILURE; - rc= memcached_do(instance, buffer, send_length, 1); + rc= memcached_do(instance, buffer, send_length, true); if (no_reply || rc != MEMCACHED_SUCCESS) return rc; @@ -85,7 +86,7 @@ static memcached_return_t binary_incr_decr(memcached_st *ptr, uint8_t cmd, memcached_server_instance_st *instance; bool no_reply= ptr->flags.no_reply; - unlikely (ptr->hosts == NULL || memcached_server_count(ptr) == 0) + unlikely (memcached_server_count(ptr) == 0) return MEMCACHED_NO_SERVERS; server_key= memcached_generate_hash(ptr, master_key, master_key_length); @@ -111,8 +112,8 @@ static memcached_return_t binary_incr_decr(memcached_st *ptr, uint8_t cmd, request.message.body.expiration= htonl((uint32_t) expiration); if ((memcached_do(instance, request.bytes, - sizeof(request.bytes), 0)!=MEMCACHED_SUCCESS) || - (memcached_io_write(instance, key, key_length, 1) == -1)) + sizeof(request.bytes), false) != MEMCACHED_SUCCESS) || + (memcached_io_write(instance, key, key_length, true) == -1)) { memcached_io_reset(instance); return MEMCACHED_WRITE_FAILURE; @@ -159,7 +160,7 @@ memcached_return_t memcached_increment_by_key(memcached_st *ptr, } else { - rc= memcached_auto(ptr, "incr", master_key, master_key_length, key, key_length, offset, value); + rc= text_incr_decr(ptr, "incr", master_key, master_key_length, key, key_length, offset, value); } LIBMEMCACHED_MEMCACHED_INCREMENT_END(); @@ -179,12 +180,16 @@ memcached_return_t memcached_decrement_by_key(memcached_st *ptr, LIBMEMCACHED_MEMCACHED_DECREMENT_START(); if (ptr->flags.binary_protocol) + { rc= binary_incr_decr(ptr, PROTOCOL_BINARY_CMD_DECREMENT, master_key, master_key_length, key, key_length, (uint64_t)offset, 0, MEMCACHED_EXPIRATION_NOT_ADD, value); + } else - rc= memcached_auto(ptr, "decr", master_key, master_key_length, key, key_length, offset, value); + { + rc= text_incr_decr(ptr, "decr", master_key, master_key_length, key, key_length, offset, value); + } LIBMEMCACHED_MEMCACHED_DECREMENT_END();