git.m6w6.name Git - m6w6/libmemcached/blob - util/operation.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
  41 #include <cstring>
  42 #include <iostream>
  43 #include <vector>
  44
  45 namespace datadifferential {
  46 namespace util {
  47
  48 class Operation {
  49   typedef std::vector<char> Packet;
  50
  51 public:
  52   typedef std::vector<Operation *> vector;
  53
  54   Operation(const char *command, size_t command_length, bool expect_response= true) :
  55     _expect_response(expect_response),
  56     packet(),
  57     _response()
  58   {
  59     packet.resize(command_length);
  60     memcpy(&packet[0], command, command_length);
  61   }
  62
  63   ~Operation()
  64   { }
  65
  66   size_t size() const
  67   {
  68     return packet.size();
  69   }
  70
  71   const char* ptr() const
  72   {
  73     return &(packet)[0];
  74   }
  75
  76   bool has_response() const
  77   {
  78     return _expect_response;
  79   }
  80
  81   void push(const char *buffer, size_t buffer_size)
  82   {
  83     size_t response_size= _response.size();
  84     _response.resize(response_size +buffer_size);
  85     memcpy(&_response[0] +response_size, buffer, buffer_size);
  86   }
  87
  88   // Return false on error
  89   bool response(std::string &arg)
  90   {
  91     if (_response.empty())
  92       return false;
  93
  94     if (not memcmp("OK\r\n", &_response[0], 3))
  95     { }
  96     else if (not memcmp("OK ", &_response[0], 3))
  97     {
  98       arg.append(&_response[3], _response.size() -3);
  99     }
 100     else if (not memcmp("ERR ", &_response[0], 4))
 101     {
 102       arg.append(&_response[4], _response.size() -4);
 103       return false;
 104     }
 105     else
 106     {
 107       arg.append(&_response[0], _response.size());
 108     }
 109
 110     return true;
 111   }
 112
 113   bool reconnect() const
 114   {
 115     return false;
 116   }
 117
 118 private:
 119   bool _expect_response;
 120   Packet packet;
 121   Packet _response;
 122 };
 123
 124 } /* namespace util */
 125 } /* namespace datadifferential */