git.m6w6.name Git - awesomized/libmemcached/blob - libmemcached/memcached_flush.c

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