1 /* vim:expandtab:shiftwidth=2:tabstop=2:smarttab:
5 * Copyright (C) 2011-2012 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 char *memcached_fetch(memcached_st
*ptr
, char *key
, size_t *key_length
,
43 memcached_return_t
*error
)
45 memcached_return_t unused
;
51 if (memcached_is_udp(ptr
))
73 *error
= MEMCACHED_NOT_SUPPORTED
;
77 memcached_result_st
*result_buffer
= &ptr
->result
;
78 result_buffer
= memcached_fetch_result(ptr
, result_buffer
, error
);
79 if (result_buffer
== NULL
or memcached_failed(*error
))
81 WATCHPOINT_ASSERT(result_buffer
== NULL
);
107 *value_length
= memcached_string_length(&result_buffer
->value
);
112 if (result_buffer
->key_length
> MEMCACHED_MAX_KEY
)
114 *error
= MEMCACHED_KEY_TOO_BIG
;
138 strncpy(key
, result_buffer
->item_key
, result_buffer
->key_length
); // For the binary protocol we will cut off the key :(
141 *key_length
= result_buffer
->key_length
;
147 *flags
= result_buffer
->item_flags
;
150 return memcached_string_take_value(&result_buffer
->value
);
153 memcached_result_st
*memcached_fetch_result(memcached_st
*ptr
,
154 memcached_result_st
*result
,
155 memcached_return_t
*error
)
157 memcached_return_t unused
;
165 *error
= MEMCACHED_INVALID_ARGUMENTS
;
169 if (memcached_is_udp(ptr
))
171 *error
= MEMCACHED_NOT_SUPPORTED
;
177 // If we have already initialized (ie it is in use) our internal, we
179 if (memcached_is_initialized(&ptr
->result
))
181 if ((result
= memcached_result_create(ptr
, NULL
)) == NULL
)
183 *error
= MEMCACHED_MEMORY_ALLOCATION_FAILURE
;
189 result
= memcached_result_create(ptr
, &ptr
->result
);
193 *error
= MEMCACHED_MAXIMUM_RETURN
; // We use this to see if we ever go into the loop
194 org::libmemcached::Instance
*server
;
195 memcached_return_t read_ret
= MEMCACHED_SUCCESS
;
196 while ((server
= memcached_io_get_readable_server(ptr
, read_ret
)))
198 char buffer
[MEMCACHED_DEFAULT_COMMAND_SIZE
];
199 *error
= memcached_response(server
, buffer
, sizeof(buffer
), result
);
201 if (*error
== MEMCACHED_IN_PROGRESS
)
205 else if (*error
== MEMCACHED_SUCCESS
)
210 else if (*error
== MEMCACHED_END
)
212 memcached_server_response_reset(server
);
214 else if (*error
!= MEMCACHED_NOTFOUND
)
220 if (*error
== MEMCACHED_NOTFOUND
and result
->count
)
222 *error
= MEMCACHED_END
;
224 else if (*error
== MEMCACHED_MAXIMUM_RETURN
and result
->count
)
226 *error
= MEMCACHED_END
;
228 else if (*error
== MEMCACHED_MAXIMUM_RETURN
) // while() loop was never entered
230 *error
= MEMCACHED_NOTFOUND
;
232 else if (*error
== MEMCACHED_SUCCESS
)
234 *error
= MEMCACHED_END
;
236 else if (result
->count
== 0)
238 *error
= MEMCACHED_NOTFOUND
;
241 /* We have completed reading data */
242 if (memcached_is_allocated(result
))
244 memcached_result_free(result
);
249 memcached_string_reset(&result
->value
);
255 memcached_return_t
memcached_fetch_execute(memcached_st
*ptr
,
256 memcached_execute_fn
*callback
,
258 uint32_t number_of_callbacks
)
260 memcached_result_st
*result
= &ptr
->result
;
261 memcached_return_t rc
;
262 bool some_errors
= false;
264 while ((result
= memcached_fetch_result(ptr
, result
, &rc
)))
266 if (memcached_failed(rc
) and rc
== MEMCACHED_NOTFOUND
)
270 else if (memcached_failed(rc
))
272 memcached_set_error(*ptr
, rc
, MEMCACHED_AT
);
277 for (uint32_t x
= 0; x
< number_of_callbacks
; x
++)
279 memcached_return_t ret
= (*callback
[x
])(ptr
, result
, context
);
280 if (memcached_failed(ret
))
283 memcached_set_error(*ptr
, ret
, MEMCACHED_AT
);
291 return MEMCACHED_SOME_ERRORS
;
294 // If we were able to run all keys without issue we return
296 if (memcached_success(rc
))
298 return MEMCACHED_SUCCESS
;