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_st *ptr, unsigned int server_key, uint8_t io_death)
  13 {
  14   if (server_key > ptr->number_of_hosts)
  15   {
  16     WATCHPOINT_ASSERT(0);
  17     return;
  18   }
  19
  20   if (ptr->hosts[server_key].fd != -1)
  21   {
  22     if (io_death == 0)
  23     {
  24       memcached_return rc;
  25       rc= memcached_do(ptr, server_key, "quit\r\n", 6, 1);
  26       WATCHPOINT_ASSERT(rc == MEMCACHED_SUCCESS || rc == MEMCACHED_FETCH_NOTFINISHED);
  27
  28       memcached_io_close(ptr, server_key);
  29     }
  30     ptr->hosts[server_key].fd= -1;
  31     ptr->hosts[server_key].stack_responses= 0;
  32     ptr->hosts[server_key].cursor_active= 0;
  33     ptr->hosts[server_key].write_buffer_offset= 0;
  34     ptr->hosts[server_key].read_buffer_length= 0;
  35     ptr->hosts[server_key].read_ptr= ptr->hosts[server_key].read_buffer;
  36     ptr->hosts[server_key].write_ptr= ptr->hosts[server_key].write_buffer;
  37   }
  38
  39   ptr->connected--;
  40 }
  41
  42 void memcached_quit(memcached_st *ptr)
  43 {
  44   unsigned int x;
  45   if (ptr->hosts == NULL ||
  46       ptr->number_of_hosts == 0)
  47     return;
  48
  49   if (ptr->hosts && ptr->number_of_hosts)
  50   {
  51     for (x= 0; x < ptr->number_of_hosts; x++)
  52       memcached_quit_server(ptr, x, 0);
  53   }
  54
  55   ptr->connected= 0;
  56 }