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 <cstdio>
  41 #include <cerrno>
  42 #include <cassert>
  43 #include <cstddef>
  44 #include <sys/socket.h>
  45 #include <string>
  46
  47 #include "util/operation.hpp"
  48
  49 struct addrinfo;
  50
  51 namespace datadifferential {
  52 namespace util {
  53
  54 class Instance
  55 {
  56 private:
  57   enum connection_state_t {
  58     NOT_WRITING,
  59     NEXT_CONNECT_ADDRINFO,
  60     CONNECT,
  61     CONNECTING,
  62     CONNECTED,
  63     WRITING,
  64     READING,
  65     FINISHED
  66   };
  67   std::string _last_error;
  68
  69 public: // Callbacks
  70   class Finish {
  71
  72   public:
  73     virtual ~Finish() { }
  74
  75     virtual bool call(const bool, const std::string &)= 0;
  76   };
  77
  78
  79 public:
  80   Instance(const std::string& hostname_arg, const std::string& service_arg);
  81
  82   Instance(const std::string& hostname_arg, const in_port_t port_arg);
  83
  84   ~Instance();
  85
  86   bool run();
  87
  88   void set_finish(Finish *arg)
  89   {
  90     _finish_fn= arg;
  91   }
  92
  93   void push(util::Operation *next)
  94   {
  95     _operations.push_back(next);
  96   }
  97
  98 private:
  99   void close_socket();
 100
 101   void free_addrinfo();
 102
 103   bool more_to_read() const;
 104
 105   std::string _host;
 106   std::string _service;
 107   int _sockfd;
 108   connection_state_t state;
 109   struct addrinfo *_addrinfo;
 110   struct addrinfo *_addrinfo_next;
 111   Finish *_finish_fn;
 112   Operation::vector _operations;
 113 };
 114
 115 } /* namespace util */
 116 } /* namespace datadifferential */