git.m6w6.name Git - m6w6/libmemcached/blob - util/instance.hpp

   1 /*  vim:expandtab:shiftwidth=2:tabstop=2:smarttab:
   2  *
   3  *  DataDifferential Utility Library
   4  *
   5  *  Copyright (C) 2011 Data Differential, http://datadifferential.com/
   6  *  All rights reserved.
   7  *
   8  *  Redistribution and use in source and binary forms, with or without
   9  *  modification, are permitted provided that the following conditions are
  10  *  met:
  11  *
  12  *      * Redistributions of source code must retain the above copyright
  13  *  notice, this list of conditions and the following disclaimer.
  14  *
  15  *      * Redistributions in binary form must reproduce the above
  16  *  copyright notice, this list of conditions and the following disclaimer
  17  *  in the documentation and/or other materials provided with the
  18  *  distribution.
  19  *
  20  *      * The names of its contributors may not be used to endorse or
  21  *  promote products derived from this software without specific prior
  22  *  written permission.
  23  *
  24  *  THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
  25  *  "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
  26  *  LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
  27  *  A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
  28  *  OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
  29  *  SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
  30  *  LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
  31  *  DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
  32  *  THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  33  *  (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
  34  *  OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  35  *
  36  */
  37
  38 #pragma once
  39
  40 #include <arpa/inet.h>
  41 #include <cstdio>
  42 #include <cerrno>
  43 #include <cassert>
  44 #include <cstddef>
  45 #include <sys/socket.h>
  46 #include <string>
  47
  48 #include "util/operation.hpp"
  49
  50 struct addrinfo;
  51
  52 namespace datadifferential {
  53 namespace util {
  54
  55 class Instance
  56 {
  57 private:
  58   enum connection_state_t {
  59     NOT_WRITING,
  60     NEXT_CONNECT_ADDRINFO,
  61     CONNECT,
  62     CONNECTING,
  63     CONNECTED,
  64     WRITING,
  65     READING,
  66     FINISHED
  67   };
  68   std::string _last_error;
  69
  70 public: // Callbacks
  71   class Finish {
  72
  73   public:
  74     virtual ~Finish() { }
  75
  76     virtual bool call(const bool, const std::string &)= 0;
  77   };
  78
  79
  80 public:
  81   Instance(const std::string& hostname_arg, const std::string& service_arg);
  82
  83   Instance(const std::string& hostname_arg, const in_port_t port_arg);
  84
  85   ~Instance();
  86
  87   bool run();
  88
  89   void set_finish(Finish *arg)
  90   {
  91     _finish_fn= arg;
  92   }
  93
  94   void push(util::Operation *next)
  95   {
  96     _operations.push_back(next);
  97   }
  98
  99 private:
 100   void close_socket();
 101
 102   void free_addrinfo();
 103
 104   bool more_to_read() const;
 105
 106   std::string _host;
 107   std::string _service;
 108   int _sockfd;
 109   connection_state_t state;
 110   struct addrinfo *_addrinfo;
 111   struct addrinfo *_addrinfo_next;
 112   Finish *_finish_fn;
 113   Operation::vector _operations;
 114 };
 115
 116 } /* namespace util */
 117 } /* namespace datadifferential */