projects / m6w6 / libmemcached / blob
? search:


9d85ad434858897556244f0447f0564cfddadaf5
[m6w6/libmemcached] / libmemcached / auto.cc
1 /* vim:expandtab:shiftwidth=2:tabstop=2:smarttab:
2 *
3 * Libmemcached library
4 *
5 * Copyright (C) 2011 Data Differential, http://datadifferential.com/
6 * Copyright (C) 2006-2009 Brian Aker 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 #include <libmemcached/common.h>
39
40 static void auto_response(memcached_server_write_instance_st instance, const bool reply, memcached_return_t& rc, uint64_t* value)
41 {
42 // If the message was successfully sent, then get the response, otherwise
43 // fail.
44 if (memcached_success(rc))
45 {
46 if (reply == false)
47 {
48 *value= UINT64_MAX;
49 return;
50 }
51
52 rc= memcached_response(instance, &instance->root->result);
53 }
54
55 if (memcached_success(rc))
56 {
57 *value= instance->root->result.numeric_value;
58 }
59 else
60 {
61 *value= UINT64_MAX;
62 }
63 }
64
65 static memcached_return_t text_incr_decr(memcached_server_write_instance_st instance,
66 const bool is_incr,
67 const char *key, size_t key_length,
68 const uint64_t offset,
69 const bool reply)
70 {
71 char buffer[MEMCACHED_DEFAULT_COMMAND_SIZE];
72
73 int send_length= snprintf(buffer, sizeof(buffer), " %" PRIu64, offset);
74 if (size_t(send_length) >= sizeof(buffer) or send_length < 0)
75 {
76 return memcached_set_error(*instance, MEMCACHED_MEMORY_ALLOCATION_FAILURE, MEMCACHED_AT,
77 memcached_literal_param("snprintf(MEMCACHED_DEFAULT_COMMAND_SIZE)"));
78 }
79
80 libmemcached_io_vector_st vector[]=
81 {
82 { NULL, 0 },
83 { memcached_literal_param("incr ") },
84 { memcached_array_string(instance->root->_namespace), memcached_array_size(instance->root->_namespace) },
85 { key, key_length },
86 { buffer, send_length },
87 { " noreply", reply ? 0 : memcached_literal_param_size(" noreply") },
88 { memcached_literal_param("\r\n") }
89 };
90
91 if (is_incr == false)
92 {
93 vector[1].buffer= "decr ";
94 }
95
96 return memcached_vdo(instance, vector, 7, true);
97 }
98
99 static memcached_return_t binary_incr_decr(memcached_server_write_instance_st instance,
100 protocol_binary_command cmd,
101 const char *key, const size_t key_length,
102 const uint64_t offset,
103 const uint64_t initial,
104 const uint32_t expiration,
105 const bool reply)
106 {
107 if (reply == false)
108 {
109 if(cmd == PROTOCOL_BINARY_CMD_DECREMENT)
110 {
111 cmd= PROTOCOL_BINARY_CMD_DECREMENTQ;
112 }
113
114 if(cmd == PROTOCOL_BINARY_CMD_INCREMENT)
115 {
116 cmd= PROTOCOL_BINARY_CMD_INCREMENTQ;
117 }
118 }
119 protocol_binary_request_incr request= {}; // = {.bytes= {0}};
120
121 request.message.header.request.magic= PROTOCOL_BINARY_REQ;
122 request.message.header.request.opcode= cmd;
123 request.message.header.request.keylen= htons((uint16_t)(key_length + memcached_array_size(instance->root->_namespace)));
124 request.message.header.request.extlen= 20;
125 request.message.header.request.datatype= PROTOCOL_BINARY_RAW_BYTES;
126 request.message.header.request.bodylen= htonl((uint32_t)(key_length + memcached_array_size(instance->root->_namespace) +request.message.header.request.extlen));
127 request.message.body.delta= memcached_htonll(offset);
128 request.message.body.initial= memcached_htonll(initial);
129 request.message.body.expiration= htonl((uint32_t) expiration);
130
131 libmemcached_io_vector_st vector[]=
132 {
133 { NULL, 0 },
134 { request.bytes, sizeof(request.bytes) },
135 { memcached_array_string(instance->root->_namespace), memcached_array_size(instance->root->_namespace) },
136 { key, key_length }
137 };
138
139 return memcached_vdo(instance, vector, 4, true);
140 }
141
142 memcached_return_t memcached_increment(memcached_st *memc,
143 const char *key, size_t key_length,
144 uint32_t offset,
145 uint64_t *value)
146 {
147 return memcached_increment_by_key(memc, key, key_length, key, key_length, offset, value);
148 }
149
150 static memcached_return_t increment_decrement_by_key(const protocol_binary_command command,
151 memcached_st *memc,
152 const char *group_key, size_t group_key_length,
153 const char *key, size_t key_length,
154 uint64_t offset,
155 uint64_t *value)
156 {
157 uint64_t local_value;
158 if (value == NULL)
159 {
160 value= &local_value;
161 }
162
163 memcached_return_t rc;
164 if (memcached_failed(rc= initialize_query(memc, true)))
165 {
166 return rc;
167 }
168
169 if (memcached_failed(rc= memcached_key_test(*memc, (const char **)&key, &key_length, 1)))
170 {
171 return memcached_last_error(memc);
172 }
173
174 uint32_t server_key= memcached_generate_hash_with_redistribution(memc, group_key, group_key_length);
175 memcached_server_write_instance_st instance= memcached_server_instance_fetch(memc, server_key);
176
177 bool reply= memcached_is_replying(instance->root);
178
179 if (memcached_is_binary(memc))
180 {
181 rc= binary_incr_decr(instance, command,
182 key, key_length,
183 uint64_t(offset), 0, MEMCACHED_EXPIRATION_NOT_ADD,
184 reply);
185 }
186 else
187 {
188 rc= text_incr_decr(instance,
189 command == PROTOCOL_BINARY_CMD_INCREMENT ? true : false,
190 key, key_length,
191 offset, reply);
192 }
193
194 auto_response(instance, reply, rc, value);
195
196 return rc;
197 }
198
199 static memcached_return_t increment_decrement_with_initial_by_key(const protocol_binary_command command,
200 memcached_st *memc,
201 const char *group_key,
202 size_t group_key_length,
203 const char *key,
204 size_t key_length,
205 uint64_t offset,
206 uint64_t initial,
207 time_t expiration,
208 uint64_t *value)
209 {
210 uint64_t local_value;
211 if (value == NULL)
212 {
213 value= &local_value;
214 }
215
216 memcached_return_t rc;
217 if (memcached_failed(rc= initialize_query(memc, true)))
218 {
219 return rc;
220 }
221
222 if (memcached_failed(rc= memcached_key_test(*memc, (const char **)&key, &key_length, 1)))
223 {
224 return memcached_last_error(memc);
225 }
226
227 uint32_t server_key= memcached_generate_hash_with_redistribution(memc, group_key, group_key_length);
228 memcached_server_write_instance_st instance= memcached_server_instance_fetch(memc, server_key);
229
230 bool reply= memcached_is_replying(instance->root);
231
232 if (memcached_is_binary(memc))
233 {
234 rc= binary_incr_decr(instance, command,
235 key, key_length,
236 offset, initial, uint32_t(expiration),
237 reply);
238
239 }
240 else
241 {
242 rc= memcached_set_error(*memc, MEMCACHED_INVALID_ARGUMENTS, MEMCACHED_AT,
243 memcached_literal_param("memcached_increment_with_initial_by_key() is not supported via the ASCII protocol"));
244 }
245
246 auto_response(instance, reply, rc, value);
247
248 return rc;
249 }
250
251 memcached_return_t memcached_decrement(memcached_st *memc,
252 const char *key, size_t key_length,
253 uint32_t offset,
254 uint64_t *value)
255 {
256 return memcached_decrement_by_key(memc, key, key_length, key, key_length, offset, value);
257 }
258
259
260 memcached_return_t memcached_increment_by_key(memcached_st *memc,
261 const char *group_key, size_t group_key_length,
262 const char *key, size_t key_length,
263 uint64_t offset,
264 uint64_t *value)
265 {
266 LIBMEMCACHED_MEMCACHED_INCREMENT_START();
267 memcached_return_t rc= increment_decrement_by_key(PROTOCOL_BINARY_CMD_INCREMENT,
268 memc,
269 group_key, group_key_length,
270 key, key_length,
271 offset, value);
272
273 LIBMEMCACHED_MEMCACHED_INCREMENT_END();
274
275 return rc;
276 }
277
278 memcached_return_t memcached_decrement_by_key(memcached_st *memc,
279 const char *group_key, size_t group_key_length,
280 const char *key, size_t key_length,
281 uint64_t offset,
282 uint64_t *value)
283 {
284 LIBMEMCACHED_MEMCACHED_DECREMENT_START();
285 memcached_return_t rc= increment_decrement_by_key(PROTOCOL_BINARY_CMD_DECREMENT,
286 memc,
287 group_key, group_key_length,
288 key, key_length,
289 offset, value);
290 LIBMEMCACHED_MEMCACHED_DECREMENT_END();
291
292 return rc;
293 }
294
295 memcached_return_t memcached_increment_with_initial(memcached_st *memc,
296 const char *key,
297 size_t key_length,
298 uint64_t offset,
299 uint64_t initial,
300 time_t expiration,
301 uint64_t *value)
302 {
303 return memcached_increment_with_initial_by_key(memc, key, key_length,
304 key, key_length,
305 offset, initial, expiration, value);
306 }
307
308 memcached_return_t memcached_increment_with_initial_by_key(memcached_st *memc,
309 const char *group_key,
310 size_t group_key_length,
311 const char *key,
312 size_t key_length,
313 uint64_t offset,
314 uint64_t initial,
315 time_t expiration,
316 uint64_t *value)
317 {
318 LIBMEMCACHED_MEMCACHED_INCREMENT_WITH_INITIAL_START();
319 memcached_return_t rc= increment_decrement_with_initial_by_key(PROTOCOL_BINARY_CMD_INCREMENT,
320 memc,
321 group_key, group_key_length,
322 key, key_length,
323 offset, initial, expiration, value);
324 LIBMEMCACHED_MEMCACHED_INCREMENT_WITH_INITIAL_END();
325
326 return rc;
327 }
328
329 memcached_return_t memcached_decrement_with_initial(memcached_st *memc,
330 const char *key,
331 size_t key_length,
332 uint64_t offset,
333 uint64_t initial,
334 time_t expiration,
335 uint64_t *value)
336 {
337 return memcached_decrement_with_initial_by_key(memc, key, key_length,
338 key, key_length,
339 offset, initial, expiration, value);
340 }
341
342 memcached_return_t memcached_decrement_with_initial_by_key(memcached_st *memc,
343 const char *group_key,
344 size_t group_key_length,
345 const char *key,
346 size_t key_length,
347 uint64_t offset,
348 uint64_t initial,
349 time_t expiration,
350 uint64_t *value)
351 {
352 LIBMEMCACHED_MEMCACHED_INCREMENT_WITH_INITIAL_START();
353 memcached_return_t rc= increment_decrement_with_initial_by_key(PROTOCOL_BINARY_CMD_DECREMENT,
354 memc,
355 group_key, group_key_length,
356 key, key_length,
357 offset, initial, expiration, value);
358
359 LIBMEMCACHED_MEMCACHED_INCREMENT_WITH_INITIAL_END();
360
361 return rc;
362 }
Resurrection of libmemcached https://launchpad.net/libmemcached