2 * Copyright (C) 2006-2009 Brian Aker
5 * Use and distribution licensed under the BSD license. See
6 * the COPYING file in the parent directory for full text.
8 * Summary: Methods for adding or decrementing values from an object in memcached
14 static memcached_return_t
text_incr_decr(memcached_st
*ptr
,
16 const char *master_key
, size_t master_key_length
,
17 const char *key
, size_t key_length
,
22 memcached_return_t rc
;
23 char buffer
[MEMCACHED_DEFAULT_COMMAND_SIZE
];
25 memcached_server_write_instance_st instance
;
26 bool no_reply
= ptr
->flags
.no_reply
;
28 unlikely (memcached_server_count(ptr
) == 0)
29 return MEMCACHED_NO_SERVERS
;
31 if (ptr
->flags
.verify_key
&& (memcached_key_test((const char **)&key
, &key_length
, 1) == MEMCACHED_BAD_KEY_PROVIDED
))
32 return MEMCACHED_BAD_KEY_PROVIDED
;
34 server_key
= memcached_generate_hash_with_redistribution(ptr
, master_key
, master_key_length
);
35 instance
= memcached_server_instance_fetch(ptr
, server_key
);
37 send_length
= (size_t)snprintf(buffer
, MEMCACHED_DEFAULT_COMMAND_SIZE
,
38 "%s %.*s%.*s %" PRIu64
"%s\r\n", verb
,
39 (int)ptr
->prefix_key_length
,
42 offset
, no_reply
? " noreply" : "");
43 unlikely (send_length
>= MEMCACHED_DEFAULT_COMMAND_SIZE
)
44 return MEMCACHED_WRITE_FAILURE
;
46 rc
= memcached_do(instance
, buffer
, send_length
, true);
47 if (no_reply
|| rc
!= MEMCACHED_SUCCESS
)
50 rc
= memcached_response(instance
, buffer
, MEMCACHED_DEFAULT_COMMAND_SIZE
, NULL
);
53 So why recheck responce? Because the protocol is brain dead :)
54 The number returned might end up equaling one of the string
55 values. Less chance of a mistake with strncmp() so we will
56 use it. We still called memcached_response() though since it
57 worked its magic for non-blocking IO.
59 if (! strncmp(buffer
, "ERROR\r\n", 7))
62 rc
= MEMCACHED_PROTOCOL_ERROR
;
64 else if (! strncmp(buffer
, "CLIENT_ERROR\r\n", 14))
67 rc
= MEMCACHED_PROTOCOL_ERROR
;
69 else if (!strncmp(buffer
, "NOT_FOUND\r\n", 11))
72 rc
= MEMCACHED_NOTFOUND
;
76 *value
= strtoull(buffer
, (char **)NULL
, 10);
77 rc
= MEMCACHED_SUCCESS
;
83 static memcached_return_t
binary_incr_decr(memcached_st
*ptr
, uint8_t cmd
,
84 const char *master_key
, size_t master_key_length
,
85 const char *key
, size_t key_length
,
86 uint64_t offset
, uint64_t initial
,
91 memcached_server_write_instance_st instance
;
92 bool no_reply
= ptr
->flags
.no_reply
;
94 unlikely (memcached_server_count(ptr
) == 0)
95 return MEMCACHED_NO_SERVERS
;
97 server_key
= memcached_generate_hash_with_redistribution(ptr
, master_key
, master_key_length
);
98 instance
= memcached_server_instance_fetch(ptr
, server_key
);
102 if(cmd
== PROTOCOL_BINARY_CMD_DECREMENT
)
103 cmd
= PROTOCOL_BINARY_CMD_DECREMENTQ
;
104 if(cmd
== PROTOCOL_BINARY_CMD_INCREMENT
)
105 cmd
= PROTOCOL_BINARY_CMD_INCREMENTQ
;
107 protocol_binary_request_incr request
= {.bytes
= {0}};
109 request
.message
.header
.request
.magic
= PROTOCOL_BINARY_REQ
;
110 request
.message
.header
.request
.opcode
= cmd
;
111 request
.message
.header
.request
.keylen
= htons((uint16_t) key_length
);
112 request
.message
.header
.request
.extlen
= 20;
113 request
.message
.header
.request
.datatype
= PROTOCOL_BINARY_RAW_BYTES
;
114 request
.message
.header
.request
.bodylen
= htonl((uint32_t) (key_length
+ request
.message
.header
.request
.extlen
));
115 request
.message
.body
.delta
= htonll(offset
);
116 request
.message
.body
.initial
= htonll(initial
);
117 request
.message
.body
.expiration
= htonl((uint32_t) expiration
);
119 memcached_return_t rc
;
120 if (((rc
= memcached_do(instance
, request
.bytes
,
121 sizeof(request
.bytes
), false)) != MEMCACHED_SUCCESS
) ||
122 (memcached_io_write(instance
, key
, key_length
, true) == -1))
124 memcached_io_reset(instance
);
125 return (rc
== MEMCACHED_SUCCESS
) ? MEMCACHED_WRITE_FAILURE
: rc
;
129 return MEMCACHED_SUCCESS
;
130 return memcached_response(instance
, (char*)value
, sizeof(*value
), NULL
);
133 memcached_return_t
memcached_increment(memcached_st
*ptr
,
134 const char *key
, size_t key_length
,
138 return memcached_increment_by_key(ptr
, key
, key_length
, key
, key_length
, offset
, value
);
141 memcached_return_t
memcached_decrement(memcached_st
*ptr
,
142 const char *key
, size_t key_length
,
146 return memcached_decrement_by_key(ptr
, key
, key_length
, key
, key_length
, offset
, value
);
149 memcached_return_t
memcached_increment_by_key(memcached_st
*ptr
,
150 const char *master_key
, size_t master_key_length
,
151 const char *key
, size_t key_length
,
155 memcached_return_t rc
= memcached_validate_key_length(key_length
, ptr
->flags
.binary_protocol
);
156 unlikely (rc
!= MEMCACHED_SUCCESS
)
159 LIBMEMCACHED_MEMCACHED_INCREMENT_START();
160 if (ptr
->flags
.binary_protocol
)
162 rc
= binary_incr_decr(ptr
, PROTOCOL_BINARY_CMD_INCREMENT
,
163 master_key
, master_key_length
, key
, key_length
,
164 (uint64_t)offset
, 0, MEMCACHED_EXPIRATION_NOT_ADD
,
169 rc
= text_incr_decr(ptr
, "incr", master_key
, master_key_length
, key
, key_length
, offset
, value
);
172 LIBMEMCACHED_MEMCACHED_INCREMENT_END();
177 memcached_return_t
memcached_decrement_by_key(memcached_st
*ptr
,
178 const char *master_key
, size_t master_key_length
,
179 const char *key
, size_t key_length
,
183 memcached_return_t rc
= memcached_validate_key_length(key_length
, ptr
->flags
.binary_protocol
);
184 unlikely (rc
!= MEMCACHED_SUCCESS
)
187 LIBMEMCACHED_MEMCACHED_DECREMENT_START();
188 if (ptr
->flags
.binary_protocol
)
190 rc
= binary_incr_decr(ptr
, PROTOCOL_BINARY_CMD_DECREMENT
,
191 master_key
, master_key_length
, key
, key_length
,
192 (uint64_t)offset
, 0, MEMCACHED_EXPIRATION_NOT_ADD
,
197 rc
= text_incr_decr(ptr
, "decr", master_key
, master_key_length
, key
, key_length
, offset
, value
);
200 LIBMEMCACHED_MEMCACHED_DECREMENT_END();
205 memcached_return_t
memcached_increment_with_initial(memcached_st
*ptr
,
213 return memcached_increment_with_initial_by_key(ptr
, key
, key_length
,
215 offset
, initial
, expiration
, value
);
218 memcached_return_t
memcached_increment_with_initial_by_key(memcached_st
*ptr
,
219 const char *master_key
,
220 size_t master_key_length
,
228 memcached_return_t rc
= memcached_validate_key_length(key_length
, ptr
->flags
.binary_protocol
);
229 unlikely (rc
!= MEMCACHED_SUCCESS
)
232 LIBMEMCACHED_MEMCACHED_INCREMENT_WITH_INITIAL_START();
233 if (ptr
->flags
.binary_protocol
)
234 rc
= binary_incr_decr(ptr
, PROTOCOL_BINARY_CMD_INCREMENT
,
235 master_key
, master_key_length
, key
, key_length
,
236 offset
, initial
, (uint32_t)expiration
,
239 rc
= MEMCACHED_PROTOCOL_ERROR
;
241 LIBMEMCACHED_MEMCACHED_INCREMENT_WITH_INITIAL_END();
246 memcached_return_t
memcached_decrement_with_initial(memcached_st
*ptr
,
254 return memcached_decrement_with_initial_by_key(ptr
, key
, key_length
,
256 offset
, initial
, expiration
, value
);
259 memcached_return_t
memcached_decrement_with_initial_by_key(memcached_st
*ptr
,
260 const char *master_key
,
261 size_t master_key_length
,
269 memcached_return_t rc
= memcached_validate_key_length(key_length
, ptr
->flags
.binary_protocol
);
270 unlikely (rc
!= MEMCACHED_SUCCESS
)
273 LIBMEMCACHED_MEMCACHED_INCREMENT_WITH_INITIAL_START();
274 if (ptr
->flags
.binary_protocol
)
276 rc
= binary_incr_decr(ptr
, PROTOCOL_BINARY_CMD_DECREMENT
,
277 master_key
, master_key_length
, key
, key_length
,
278 offset
, initial
, (uint32_t)expiration
,
283 rc
= MEMCACHED_PROTOCOL_ERROR
;
286 LIBMEMCACHED_MEMCACHED_INCREMENT_WITH_INITIAL_END();