git.m6w6.name Git - m6w6/libmemcached/blob - lib/memcached_auto.c

   1 #include "common.h"
   2
   3 static memcached_return memcached_auto(memcached_st *ptr,
   4                                        char *verb,
   5                                        char *key, size_t key_length,
   6                                        unsigned int offset,
   7                                        unsigned int *value)
   8 {
   9   size_t send_length, sent_length;
  10   memcached_return rc;
  11   char buffer[MEMCACHED_DEFAULT_COMMAND_SIZE];
  12   unsigned int server_key;
  13
  14   server_key= memcached_generate_hash(ptr, key, key_length);
  15
  16   if ((rc= memcached_connect(ptr, server_key)) != MEMCACHED_SUCCESS)
  17     return rc;
  18
  19   send_length= snprintf(buffer, MEMCACHED_DEFAULT_COMMAND_SIZE,
  20                         "%s %.*s %u\r\n", verb,
  21                         (int)key_length, key,
  22                         offset);
  23   if (send_length >= MEMCACHED_DEFAULT_COMMAND_SIZE)
  24     return MEMCACHED_WRITE_FAILURE;
  25   sent_length= memcached_io_write(ptr, server_key, buffer, send_length, 1);
  26
  27   if (sent_length == -1 || sent_length != send_length)
  28     return MEMCACHED_WRITE_FAILURE;
  29
  30   memset(buffer, 0, MEMCACHED_DEFAULT_COMMAND_SIZE);
  31
  32   rc= memcached_response(ptr, buffer, MEMCACHED_DEFAULT_COMMAND_SIZE, server_key);
  33
  34   /*
  35     So why recheck responce? Because the protocol is brain dead :)
  36     The number returned might end up equaling one of the string
  37     values. Less chance of a mistake with memcmp() so we will
  38     use it. We still called memcached_response() though since it
  39     worked its magic for non-blocking IO.
  40   */
  41   if (!memcmp(buffer, "ERROR\r\n", MEMCACHED_DEFAULT_COMMAND_SIZE))
  42   {
  43     *value= 0;
  44     rc= MEMCACHED_PROTOCOL_ERROR;
  45   }
  46   else if (!memcmp(buffer, "NOT_FOUND\r\n", MEMCACHED_DEFAULT_COMMAND_SIZE))
  47   {
  48     *value= 0;
  49     rc= MEMCACHED_NOTFOUND;
  50   }
  51   else
  52   {
  53     *value= strtol(buffer, (char **)NULL, 10);
  54     rc= MEMCACHED_SUCCESS;
  55   }
  56
  57   return rc;
  58 }
  59
  60 memcached_return memcached_increment(memcached_st *ptr,
  61                                      char *key, size_t key_length,
  62                                      unsigned int offset,
  63                                      unsigned int *value)
  64 {
  65   memcached_return rc;
  66
  67   LIBMEMCACHED_MEMCACHED_INCREMENT_START();
  68   rc= memcached_auto(ptr, "incr", key, key_length, offset, value);
  69   LIBMEMCACHED_MEMCACHED_INCREMENT_END();
  70
  71   return rc;
  72 }
  73
  74 memcached_return memcached_decrement(memcached_st *ptr,
  75                                      char *key, size_t key_length,
  76                                      unsigned int offset,
  77                                      unsigned int *value)
  78 {
  79   memcached_return rc;
  80
  81   LIBMEMCACHED_MEMCACHED_DECREMENT_START();
  82   rc= memcached_auto(ptr, "decr", key, key_length, offset, value);
  83   LIBMEMCACHED_MEMCACHED_DECREMENT_END();
  84
  85   return rc;
  86 }