kenmirrors
/
freeswitch
mirror of https://github.com/signalwire/freeswitch.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

make increment/decrement work if the key hasn't been set yet 

git-svn-id: http://svn.freeswitch.org/svn/freeswitch/trunk@14461 d0543943-73ff-0310-b7d9-9358b9ac24b2
This commit is contained in:
Rupa Schomaker 2009-08-07 20:59:23 +00:00
parent 55bfb7d55a
commit 57dc9c2636
1 changed files with 26 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
src/mod/applications/mod_memcache/mod_memcache.c
View File

@ -230,14 +230,40 @@ SWITCH_STANDARD_API(memcache_function)
key = argv[1]; key = argv[1];
uint64_t ivalue; uint64_t ivalue;
unsigned int offset = 1; unsigned int offset = 1;
switch_bool_t increment = SWITCH_TRUE;
char *svalue = NULL;
if(argc > 2) { if(argc > 2) {
offset = (unsigned int)strtol(argv[2], NULL, 10); offset = (unsigned int)strtol(argv[2], NULL, 10);
svalue = argv[2];
} else {
svalue = "1";
} }
if (!strcasecmp(subcmd, "increment")) { if (!strcasecmp(subcmd, "increment")) {
increment = SWITCH_TRUE;
rc = memcached_increment(memcached, key, strlen(key), offset, &ivalue); rc = memcached_increment(memcached, key, strlen(key), offset, &ivalue);
} else if (!strcasecmp(subcmd, "decrement")) { } else if (!strcasecmp(subcmd, "decrement")) {
increment = SWITCH_FALSE;
rc = memcached_decrement(memcached, key, strlen(key), offset, &ivalue); rc = memcached_decrement(memcached, key, strlen(key), offset, &ivalue);
} }
if (rc == MEMCACHED_NOTFOUND) {
/* ok, trying to incr / decr a value that doesn't exist yet.
Try to add an appropriate initial value. If someone else beat
us to it, then redo incr/decr. Otherwise we're good.
*/
rc = memcached_add(memcached, key, strlen(key), (increment) ? svalue : "0", strlen(svalue), 0, 0);
if (rc == MEMCACHED_SUCCESS) {
ivalue = (increment) ? offset : 0;
switch_log_printf(SWITCH_CHANNEL_LOG, SWITCH_LOG_DEBUG, "Initialized inc/dec memcache key: %s to value %d\n", key, offset);
} else {
if (increment) {
rc = memcached_increment(memcached, key, strlen(key), offset, &ivalue);
} else {
rc = memcached_decrement(memcached, key, strlen(key), offset, &ivalue);
}
switch_log_printf(SWITCH_CHANNEL_LOG, SWITCH_LOG_DEBUG, "Someone else created incr/dec memcache key, resubmitting inc/dec request.\n");
}
}
if (rc == MEMCACHED_SUCCESS) { if (rc == MEMCACHED_SUCCESS) {
stream->write_function(stream, "%ld", ivalue); stream->write_function(stream, "%ld", ivalue);
} else { } else {