1 /* vim:expandtab:shiftwidth=2:tabstop=2:smarttab:
5 * Copyright (C) 2011 Data Differential, http://datadifferential.com/
6 * Copyright (C) 2006-2009 Brian Aker All rights reserved.
8 * Redistribution and use in source and binary forms, with or without
9 * modification, are permitted provided that the following conditions are
12 * * Redistributions of source code must retain the above copyright
13 * notice, this list of conditions and the following disclaimer.
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
20 * * The names of its contributors may not be used to endorse or
21 * promote products derived from this software without specific prior
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.
38 #include <libmemcached/common.h>
40 static void auto_response(memcached_instance_st
* instance
, const bool reply
, memcached_return_t
& rc
, uint64_t* value
)
42 // If the message was successfully sent, then get the response, otherwise
44 if (memcached_success(rc
))
52 rc
= memcached_response(instance
, &instance
->root
->result
);
55 if (memcached_fatal(rc
))
57 fprintf(stderr
, "%s:%d %s\n", __FILE__
, __LINE__
, memcached_strerror(NULL
, rc
));
58 assert(memcached_last_error(instance
->root
) != MEMCACHED_SUCCESS
);
61 else if (memcached_failed(rc
))
67 assert(memcached_last_error(instance
->root
) != MEMCACHED_NOTFOUND
);
68 *value
= instance
->root
->result
.numeric_value
;
72 static memcached_return_t
text_incr_decr(memcached_instance_st
* instance
,
74 const char *key
, size_t key_length
,
75 const uint64_t offset
,
78 char buffer
[MEMCACHED_DEFAULT_COMMAND_SIZE
];
80 int send_length
= snprintf(buffer
, sizeof(buffer
), " %" PRIu64
, offset
);
81 if (size_t(send_length
) >= sizeof(buffer
) or send_length
< 0)
83 return memcached_set_error(*instance
, MEMCACHED_MEMORY_ALLOCATION_FAILURE
, MEMCACHED_AT
,
84 memcached_literal_param("snprintf(MEMCACHED_DEFAULT_COMMAND_SIZE)"));
87 libmemcached_io_vector_st vector
[]=
90 { memcached_literal_param("incr ") },
91 { memcached_array_string(instance
->root
->_namespace
), memcached_array_size(instance
->root
->_namespace
) },
93 { buffer
, size_t(send_length
) },
94 { " noreply", reply
? 0 : memcached_literal_param_size(" noreply") },
95 { memcached_literal_param("\r\n") }
100 vector
[1].buffer
= "decr ";
103 return memcached_vdo(instance
, vector
, 7, true);
106 static memcached_return_t
binary_incr_decr(memcached_instance_st
* instance
,
107 protocol_binary_command cmd
,
108 const char *key
, const size_t key_length
,
109 const uint64_t offset
,
110 const uint64_t initial
,
111 const uint32_t expiration
,
116 if(cmd
== PROTOCOL_BINARY_CMD_DECREMENT
)
118 cmd
= PROTOCOL_BINARY_CMD_DECREMENTQ
;
121 if(cmd
== PROTOCOL_BINARY_CMD_INCREMENT
)
123 cmd
= PROTOCOL_BINARY_CMD_INCREMENTQ
;
126 protocol_binary_request_incr request
= {}; // = {.bytes= {0}};
128 initialize_binary_request(instance
, request
.message
.header
);
130 request
.message
.header
.request
.opcode
= cmd
;
131 request
.message
.header
.request
.keylen
= htons((uint16_t)(key_length
+ memcached_array_size(instance
->root
->_namespace
)));
132 request
.message
.header
.request
.extlen
= 20;
133 request
.message
.header
.request
.datatype
= PROTOCOL_BINARY_RAW_BYTES
;
134 request
.message
.header
.request
.bodylen
= htonl((uint32_t)(key_length
+ memcached_array_size(instance
->root
->_namespace
) +request
.message
.header
.request
.extlen
));
135 request
.message
.body
.delta
= memcached_htonll(offset
);
136 request
.message
.body
.initial
= memcached_htonll(initial
);
137 request
.message
.body
.expiration
= htonl((uint32_t) expiration
);
139 libmemcached_io_vector_st vector
[]=
142 { request
.bytes
, sizeof(request
.bytes
) },
143 { memcached_array_string(instance
->root
->_namespace
), memcached_array_size(instance
->root
->_namespace
) },
147 return memcached_vdo(instance
, vector
, 4, true);
150 memcached_return_t
memcached_increment(memcached_st
*memc
,
151 const char *key
, size_t key_length
,
155 return memcached_increment_by_key(memc
, key
, key_length
, key
, key_length
, offset
, value
);
158 static memcached_return_t
increment_decrement_by_key(const protocol_binary_command command
,
160 const char *group_key
, size_t group_key_length
,
161 const char *key
, size_t key_length
,
165 uint64_t local_value
;
171 memcached_return_t rc
;
172 if (memcached_failed(rc
= initialize_query(memc
, true)))
177 if (memcached_is_encrypted(memc
))
179 return memcached_set_error(*memc
, MEMCACHED_NOT_SUPPORTED
, MEMCACHED_AT
,
180 memcached_literal_param("Operation not allowed while encyrption is enabled"));
183 if (memcached_failed(rc
= memcached_key_test(*memc
, (const char **)&key
, &key_length
, 1)))
185 return memcached_last_error(memc
);
188 uint32_t server_key
= memcached_generate_hash_with_redistribution(memc
, group_key
, group_key_length
);
189 memcached_instance_st
* instance
= memcached_instance_fetch(memc
, server_key
);
191 bool reply
= memcached_is_replying(instance
->root
);
193 if (memcached_is_binary(memc
))
195 rc
= binary_incr_decr(instance
, command
,
197 uint64_t(offset
), 0, MEMCACHED_EXPIRATION_NOT_ADD
,
202 rc
= text_incr_decr(instance
,
203 command
== PROTOCOL_BINARY_CMD_INCREMENT
? true : false,
208 auto_response(instance
, reply
, rc
, value
);
213 static memcached_return_t
increment_decrement_with_initial_by_key(const protocol_binary_command command
,
215 const char *group_key
,
216 size_t group_key_length
,
224 uint64_t local_value
;
230 memcached_return_t rc
;
231 if (memcached_failed(rc
= initialize_query(memc
, true)))
236 if (memcached_is_encrypted(memc
))
238 return memcached_set_error(*memc
, MEMCACHED_NOT_SUPPORTED
, MEMCACHED_AT
,
239 memcached_literal_param("Operation not allowed while encyrption is enabled"));
242 if (memcached_failed(rc
= memcached_key_test(*memc
, (const char **)&key
, &key_length
, 1)))
244 return memcached_last_error(memc
);
247 uint32_t server_key
= memcached_generate_hash_with_redistribution(memc
, group_key
, group_key_length
);
248 memcached_instance_st
* instance
= memcached_instance_fetch(memc
, server_key
);
250 bool reply
= memcached_is_replying(instance
->root
);
252 if (memcached_is_binary(memc
))
254 rc
= binary_incr_decr(instance
, command
,
256 offset
, initial
, uint32_t(expiration
),
262 rc
= memcached_set_error(*memc
, MEMCACHED_INVALID_ARGUMENTS
, MEMCACHED_AT
,
263 memcached_literal_param("memcached_increment_with_initial_by_key() is not supported via the ASCII protocol"));
266 auto_response(instance
, reply
, rc
, value
);
271 memcached_return_t
memcached_decrement(memcached_st
*memc
,
272 const char *key
, size_t key_length
,
276 return memcached_decrement_by_key(memc
, key
, key_length
, key
, key_length
, offset
, value
);
280 memcached_return_t
memcached_increment_by_key(memcached_st
*shell
,
281 const char *group_key
, size_t group_key_length
,
282 const char *key
, size_t key_length
,
286 Memcached
* memc
= memcached2Memcached(shell
);
287 LIBMEMCACHED_MEMCACHED_INCREMENT_START();
288 memcached_return_t rc
= increment_decrement_by_key(PROTOCOL_BINARY_CMD_INCREMENT
,
290 group_key
, group_key_length
,
294 LIBMEMCACHED_MEMCACHED_INCREMENT_END();
299 memcached_return_t
memcached_decrement_by_key(memcached_st
*shell
,
300 const char *group_key
, size_t group_key_length
,
301 const char *key
, size_t key_length
,
305 Memcached
* memc
= memcached2Memcached(shell
);
306 LIBMEMCACHED_MEMCACHED_DECREMENT_START();
307 memcached_return_t rc
= increment_decrement_by_key(PROTOCOL_BINARY_CMD_DECREMENT
,
309 group_key
, group_key_length
,
312 LIBMEMCACHED_MEMCACHED_DECREMENT_END();
317 memcached_return_t
memcached_increment_with_initial(memcached_st
*memc
,
325 return memcached_increment_with_initial_by_key(memc
, key
, key_length
,
327 offset
, initial
, expiration
, value
);
330 memcached_return_t
memcached_increment_with_initial_by_key(memcached_st
*shell
,
331 const char *group_key
,
332 size_t group_key_length
,
340 LIBMEMCACHED_MEMCACHED_INCREMENT_WITH_INITIAL_START();
341 Memcached
* memc
= memcached2Memcached(shell
);
342 memcached_return_t rc
= increment_decrement_with_initial_by_key(PROTOCOL_BINARY_CMD_INCREMENT
,
344 group_key
, group_key_length
,
346 offset
, initial
, expiration
, value
);
347 LIBMEMCACHED_MEMCACHED_INCREMENT_WITH_INITIAL_END();
352 memcached_return_t
memcached_decrement_with_initial(memcached_st
*memc
,
360 return memcached_decrement_with_initial_by_key(memc
, key
, key_length
,
362 offset
, initial
, expiration
, value
);
365 memcached_return_t
memcached_decrement_with_initial_by_key(memcached_st
*shell
,
366 const char *group_key
,
367 size_t group_key_length
,
375 LIBMEMCACHED_MEMCACHED_INCREMENT_WITH_INITIAL_START();
376 Memcached
* memc
= memcached2Memcached(shell
);
377 memcached_return_t rc
= increment_decrement_with_initial_by_key(PROTOCOL_BINARY_CMD_DECREMENT
,
379 group_key
, group_key_length
,
381 offset
, initial
, expiration
, value
);
383 LIBMEMCACHED_MEMCACHED_INCREMENT_WITH_INITIAL_END();