1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
|
/* LibMemcached
* Copyright (C) 2006-2009 Brian Aker
* All rights reserved.
*
* Use and distribution licensed under the BSD license. See
* the COPYING file in the parent directory for full text.
*
* Summary:
*
*/
/*
Execute a memcached_set() a set of pairs.
Return the number of rows set.
*/
#include <config.h>
#include "clients/execute.h"
unsigned int execute_set(memcached_st *memc, pairs_st *pairs, unsigned int number_of)
{
unsigned int x;
unsigned int pairs_sent;
for (x= 0, pairs_sent= 0; x < number_of; x++)
{
memcached_return_t rc= memcached_set(memc, pairs[x].key, pairs[x].key_length,
pairs[x].value, pairs[x].value_length,
0, 0);
if (memcached_failed(rc))
{
fprintf(stderr, "%s:%d Failure on insert (%s) of %.*s\n",
__FILE__, __LINE__,
memcached_last_error_message(memc),
(unsigned int)pairs[x].key_length, pairs[x].key);
// We will try to reconnect and see if that fixes the issue
memcached_quit(memc);
}
else
{
pairs_sent++;
}
}
return pairs_sent;
}
/*
Execute a memcached_get() on a set of pairs.
Return the number of rows retrieved.
*/
unsigned int execute_get(memcached_st *memc, pairs_st *pairs, unsigned int number_of)
{
unsigned int x;
unsigned int retrieved;
for (retrieved= 0,x= 0; x < number_of; x++)
{
size_t value_length;
uint32_t flags;
unsigned int fetch_key= (unsigned int)((unsigned int)random() % number_of);
memcached_return_t rc;
char *value= memcached_get(memc, pairs[fetch_key].key, pairs[fetch_key].key_length,
&value_length, &flags, &rc);
if (memcached_failed(rc))
{
fprintf(stderr, "%s:%d Failure on read(%s) of %.*s\n",
__FILE__, __LINE__,
memcached_last_error_message(memc),
(unsigned int)pairs[fetch_key].key_length, pairs[fetch_key].key);
}
else
{
retrieved++;
}
::free(value);
}
return retrieved;
}
/**
* Callback function to count the number of results
*/
static memcached_return_t callback_counter(const memcached_st *ptr,
memcached_result_st *result,
void *context)
{
(void)ptr;
(void)result;
unsigned int *counter= (unsigned int *)context;
*counter= *counter + 1;
return MEMCACHED_SUCCESS;
}
/**
* Try to run a large mget to get all of the keys
* @param memc memcached handle
* @param keys the keys to get
* @param key_length the length of the keys
* @param number_of the number of keys to try to get
* @return the number of keys received
*/
unsigned int execute_mget(memcached_st *memc,
const char * const *keys,
size_t *key_length,
unsigned int number_of)
{
unsigned int retrieved= 0;
memcached_execute_fn callbacks[]= { callback_counter };
memcached_return_t rc;
rc= memcached_mget_execute(memc, keys, key_length,
(size_t)number_of, callbacks, &retrieved, 1);
if (rc == MEMCACHED_SUCCESS || rc == MEMCACHED_NOTFOUND ||
rc == MEMCACHED_BUFFERED || rc == MEMCACHED_END)
{
rc= memcached_fetch_execute(memc, callbacks, (void *)&retrieved, 1);
if (rc != MEMCACHED_SUCCESS && rc != MEMCACHED_NOTFOUND && rc != MEMCACHED_END)
{
fprintf(stderr, "%s:%d Failed to execute mget: %s\n",
__FILE__, __LINE__,
memcached_strerror(memc, rc));
memcached_quit(memc);
return 0;
}
}
else
{
fprintf(stderr, "%s:%d Failed to execute mget: %s\n",
__FILE__, __LINE__,
memcached_strerror(memc, rc));
memcached_quit(memc);
return 0;
}
return retrieved;
}
|
