Merge in additional work to handle "weight" for servers.
[m6w6/libmemcached] / libmemcached / options / context.h
1 /* vim:expandtab:shiftwidth=2:tabstop=2:smarttab:
2 *
3 * Libmemcached 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 <libmemcached/memcached.h>
41
42 class Context
43 {
44 public:
45 Context(const char *option_string, size_t option_string_length, memcached_st *memc_arg,
46 memcached_return_t &rc_arg) :
47 scanner(NULL),
48 begin(NULL),
49 pos(0),
50 memc(NULL),
51 rc(rc_arg),
52 _is_server(false),
53 _end(false)
54 {
55 _hostname[0]= 0;
56 buf= option_string;
57 length= option_string_length;
58 memc= memc_arg;
59 init_scanner();
60 rc= MEMCACHED_SUCCESS;
61 }
62
63 bool end()
64 {
65 return _end;
66 }
67
68 void start();
69
70 void set_end()
71 {
72 rc= MEMCACHED_SUCCESS;
73 _end= true;
74 }
75
76 void set_server()
77 {
78 _is_server= true;
79 }
80
81 void unset_server()
82 {
83 _is_server= false;
84 }
85
86 bool is_server()
87 {
88 return _is_server;
89 }
90
91 const char *set_hostname(const char *str, size_t size)
92 {
93 size_t copy_length= std::min((size_t)NI_MAXHOST, size);
94 memcpy(_hostname, str, copy_length);
95 _hostname[copy_length]= 0;
96
97 return _hostname;
98 }
99
100 const char *hostname()
101 {
102 return _hostname;
103 }
104
105 ~Context()
106 {
107 destroy_scanner();
108 }
109
110 void *scanner;
111 const char *buf;
112 const char *begin;
113 size_t pos;
114 size_t length;
115 memcached_st *memc;
116 memcached_return_t &rc;
117
118 protected:
119 void init_scanner();
120 void destroy_scanner();
121
122 private:
123 bool _is_server;
124 bool _end;
125 char _hostname[NI_MAXHOST];
126 };