git.m6w6.name Git - awesomized/libmemcached/blob - lib/memcached_quit.c

   1 #include "common.h"
   2
   3 /*
   4   This closes all connections (forces flush of input as well).
   5
   6   Maybe add a host specific, or key specific version?
   7
   8   The reason we send "quit" is that in case we have buffered IO, this
   9   will force data to be completed.
  10 */
  11
  12 void memcached_quit_server(memcached_server_st *ptr, uint8_t io_death)
  13 {
  14   if (ptr->fd != -1)
  15   {
  16     if (io_death == 0)
  17     {
  18       memcached_return rc;
  19       ssize_t read_length;
  20       char buffer[MEMCACHED_MAX_BUFFER];
  21
  22       rc= memcached_do(ptr, "quit\r\n", 6, 1);
  23       WATCHPOINT_ASSERT(rc == MEMCACHED_SUCCESS || rc == MEMCACHED_FETCH_NOTFINISHED);
  24
  25       /* read until socket is closed, or there is an error
  26        * closing the socket before all data is read
  27        * results in server throwing away all data which is
  28        * not read
  29        */
  30       while ((read_length=
  31               memcached_io_read(ptr, buffer, sizeof(buffer)/sizeof(*buffer)))
  32              > 0)
  33         {
  34           ;
  35         }
  36       memcached_io_close(ptr);
  37     }
  38
  39     ptr->fd= -1;
  40     ptr->write_buffer_offset= 0;
  41     ptr->read_buffer_length= 0;
  42     ptr->read_ptr= ptr->read_buffer;
  43     memcached_server_response_reset(ptr);
  44   }
  45 }
  46
  47 void memcached_quit(memcached_st *ptr)
  48 {
  49   unsigned int x;
  50
  51   if (ptr->hosts == NULL ||
  52       ptr->number_of_hosts == 0)
  53     return;
  54
  55   if (ptr->hosts && ptr->number_of_hosts)
  56   {
  57     for (x= 0; x < ptr->number_of_hosts; x++)
  58       memcached_quit_server(&ptr->hosts[x], 0);
  59   }
  60 }