X-Git-Url: https://git.m6w6.name/?a=blobdiff_plain;ds=inline;f=lib%2Fmemcached_response.c;h=10547504743c476956ade3a7a5ebfe664f44aa4c;hb=d6505b16fb85f77527934510d0395adfcbb66377;hp=91ae3fe60870e48092bd2c1bff8f28c56c81717e;hpb=b8aaaefdcb0b10d22668da2b530b67f55ffa653a;p=m6w6%2Flibmemcached diff --git a/lib/memcached_response.c b/lib/memcached_response.c index 91ae3fe6..10547504 100644 --- a/lib/memcached_response.c +++ b/lib/memcached_response.c @@ -5,31 +5,45 @@ from an issued command. */ -#include +#include "common.h" +#include "memcached_io.h" memcached_return memcached_response(memcached_st *ptr, char *buffer, size_t buffer_length, unsigned int server_key) { + unsigned int x; size_t send_length; char *buffer_ptr; + unsigned int max_messages; + memset(buffer, 0, buffer_length); send_length= 0; - buffer_ptr= buffer; - while (1) + max_messages= memcached_server_response_count(ptr, server_key); + for (x= 0; x <= max_messages; x++) { - unsigned int read_length; - read_length= read(ptr->hosts[server_key].fd, buffer_ptr, 1); + buffer_ptr= buffer; + + while (1) + { + unsigned int read_length; - if (read_length != 1) - return MEMCACHED_UNKNOWN_READ_FAILURE; + read_length= memcached_io_read(ptr, server_key, + buffer_ptr, 1); - if (*buffer_ptr == '\n') - break; - else - buffer_ptr++; + if (read_length != 1) + return MEMCACHED_UNKNOWN_READ_FAILURE; + + if (*buffer_ptr == '\n') + break; + else + buffer_ptr++; + } + + if (memcached_server_response_count(ptr, server_key)) + memcached_server_response_decrement(ptr, server_key); } switch(buffer[0]) @@ -40,15 +54,17 @@ memcached_return memcached_response(memcached_st *ptr, return MEMCACHED_SUCCESS; case 'S': /* STORED STATS SERVER_ERROR */ { - if (buffer[1] == 'T') /* STORED STATS */ - return MEMCACHED_SUCCESS; + if (buffer[2] == 'A') /* STORED STATS */ + return MEMCACHED_STAT; else if (buffer[1] == 'E') return MEMCACHED_SERVER_ERROR; + else if (buffer[1] == 'T') + return MEMCACHED_STORED; else return MEMCACHED_UNKNOWN_READ_FAILURE; } case 'D': /* DELETED */ - return MEMCACHED_SUCCESS; + return MEMCACHED_DELETED; case 'N': /* NOT_FOUND */ { if (buffer[4] == 'F') @@ -61,7 +77,7 @@ memcached_return memcached_response(memcached_st *ptr, case 'E': /* PROTOCOL ERROR or END */ { if (buffer[1] == 'N') - return MEMCACHED_NOTFOUND; + return MEMCACHED_END; else if (buffer[1] == 'R') return MEMCACHED_PROTOCOL_ERROR; else @@ -72,7 +88,6 @@ memcached_return memcached_response(memcached_st *ptr, default: return MEMCACHED_UNKNOWN_READ_FAILURE; - return MEMCACHED_READ_FAILURE; } return MEMCACHED_SUCCESS;