git.m6w6.name Git - m6w6/libmemcached/blob - libmemcached/flush.c

   1 #include "common.h"
   2
   3 static memcached_return_t memcached_flush_binary(memcached_st *ptr,
   4                                                  time_t expiration);
   5 static memcached_return_t memcached_flush_textual(memcached_st *ptr,
   6                                                   time_t expiration);
   7
   8 memcached_return_t memcached_flush(memcached_st *ptr, time_t expiration)
   9 {
  10   memcached_return_t rc;
  11
  12   LIBMEMCACHED_MEMCACHED_FLUSH_START();
  13   if (ptr->flags.binary_protocol)
  14     rc= memcached_flush_binary(ptr, expiration);
  15   else
  16     rc= memcached_flush_textual(ptr, expiration);
  17   LIBMEMCACHED_MEMCACHED_FLUSH_END();
  18   return rc;
  19 }
  20
  21 static memcached_return_t memcached_flush_textual(memcached_st *ptr,
  22                                                   time_t expiration)
  23 {
  24   unsigned int x;
  25   size_t send_length;
  26   memcached_return_t rc;
  27   char buffer[MEMCACHED_DEFAULT_COMMAND_SIZE];
  28
  29   unlikely (memcached_server_count(ptr) == 0)
  30     return MEMCACHED_NO_SERVERS;
  31
  32   for (x= 0; x < memcached_server_count(ptr); x++)
  33   {
  34     bool no_reply= ptr->flags.no_reply;
  35     memcached_server_write_instance_st instance=
  36       memcached_server_instance_fetch(ptr, x);
  37
  38     if (expiration)
  39       send_length= (size_t) snprintf(buffer, MEMCACHED_DEFAULT_COMMAND_SIZE,
  40                                      "flush_all %llu%s\r\n",
  41                                      (unsigned long long)expiration, no_reply ? " noreply" : "");
  42     else
  43       send_length= (size_t) snprintf(buffer, MEMCACHED_DEFAULT_COMMAND_SIZE,
  44                                      "flush_all%s\r\n", no_reply ? " noreply" : "");
  45
  46     rc= memcached_do(instance, buffer, send_length, true);
  47
  48     if (rc == MEMCACHED_SUCCESS && !no_reply)
  49       (void)memcached_response(instance, buffer, MEMCACHED_DEFAULT_COMMAND_SIZE, NULL);
  50   }
  51
  52   return MEMCACHED_SUCCESS;
  53 }
  54
  55 static memcached_return_t memcached_flush_binary(memcached_st *ptr,
  56                                                  time_t expiration)
  57 {
  58   uint32_t x;
  59   protocol_binary_request_flush request= {.bytes= {0}};
  60
  61   unlikely (memcached_server_count(ptr) == 0)
  62     return MEMCACHED_NO_SERVERS;
  63
  64   request.message.header.request.magic= (uint8_t)PROTOCOL_BINARY_REQ;
  65   request.message.header.request.opcode= PROTOCOL_BINARY_CMD_FLUSH;
  66   request.message.header.request.extlen= 4;
  67   request.message.header.request.datatype= PROTOCOL_BINARY_RAW_BYTES;
  68   request.message.header.request.bodylen= htonl(request.message.header.request.extlen);
  69   request.message.body.expiration= htonl((uint32_t) expiration);
  70
  71   for (x= 0; x < memcached_server_count(ptr); x++)
  72   {
  73     memcached_server_write_instance_st instance=
  74       memcached_server_instance_fetch(ptr, x);
  75
  76     if (ptr->flags.no_reply)
  77     {
  78       request.message.header.request.opcode= PROTOCOL_BINARY_CMD_FLUSHQ;
  79     }
  80     else
  81     {
  82       request.message.header.request.opcode= PROTOCOL_BINARY_CMD_FLUSH;
  83     }
  84
  85     if (memcached_do(instance, request.bytes,
  86                      sizeof(request.bytes), true) != MEMCACHED_SUCCESS)
  87     {
  88       memcached_io_reset(instance);
  89       return MEMCACHED_WRITE_FAILURE;
  90     }
  91   }
  92
  93   for (x= 0; x < memcached_server_count(ptr); x++)
  94   {
  95     memcached_server_write_instance_st instance=
  96       memcached_server_instance_fetch(ptr, x);
  97
  98     if (memcached_server_response_count(instance) > 0)
  99       (void)memcached_response(instance, NULL, 0, NULL);
 100   }
 101
 102   return MEMCACHED_SUCCESS;
 103 }