Fix for lp:962815, from https://launchpad.net/~harebox
[awesomized/libmemcached] / 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 <iosfwd>
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 &);
90
91 bool reconnect() const
92 {
93 return false;
94 }
95
96 private:
97 bool _expect_response;
98 Packet packet;
99 Packet _response;
100 };
101
102 } /* namespace util */
103 } /* namespace datadifferential */