variadic LPUSH/RPUSH
This commit is contained in:
parent
7c0e1b53c4
commit
fb2feae599
@ -85,8 +85,8 @@ struct redisCommand redisCommandTable[] = {
|
|||||||
{"incr",incrCommand,2,REDIS_CMD_DENYOOM,NULL,1,1,1,0,0},
|
{"incr",incrCommand,2,REDIS_CMD_DENYOOM,NULL,1,1,1,0,0},
|
||||||
{"decr",decrCommand,2,REDIS_CMD_DENYOOM,NULL,1,1,1,0,0},
|
{"decr",decrCommand,2,REDIS_CMD_DENYOOM,NULL,1,1,1,0,0},
|
||||||
{"mget",mgetCommand,-2,0,NULL,1,-1,1,0,0},
|
{"mget",mgetCommand,-2,0,NULL,1,-1,1,0,0},
|
||||||
{"rpush",rpushCommand,3,REDIS_CMD_DENYOOM,NULL,1,1,1,0,0},
|
{"rpush",rpushCommand,-3,REDIS_CMD_DENYOOM,NULL,1,1,1,0,0},
|
||||||
{"lpush",lpushCommand,3,REDIS_CMD_DENYOOM,NULL,1,1,1,0,0},
|
{"lpush",lpushCommand,-3,REDIS_CMD_DENYOOM,NULL,1,1,1,0,0},
|
||||||
{"rpushx",rpushxCommand,3,REDIS_CMD_DENYOOM,NULL,1,1,1,0,0},
|
{"rpushx",rpushxCommand,3,REDIS_CMD_DENYOOM,NULL,1,1,1,0,0},
|
||||||
{"lpushx",lpushxCommand,3,REDIS_CMD_DENYOOM,NULL,1,1,1,0,0},
|
{"lpushx",lpushxCommand,3,REDIS_CMD_DENYOOM,NULL,1,1,1,0,0},
|
||||||
{"linsert",linsertCommand,5,REDIS_CMD_DENYOOM,NULL,1,1,1,0,0},
|
{"linsert",linsertCommand,5,REDIS_CMD_DENYOOM,NULL,1,1,1,0,0},
|
||||||
|
41
src/t_list.c
41
src/t_list.c
@ -259,30 +259,35 @@ void listTypeConvert(robj *subject, int enc) {
|
|||||||
*----------------------------------------------------------------------------*/
|
*----------------------------------------------------------------------------*/
|
||||||
|
|
||||||
void pushGenericCommand(redisClient *c, int where) {
|
void pushGenericCommand(redisClient *c, int where) {
|
||||||
|
int j, addlen = 0, pushed = 0;
|
||||||
robj *lobj = lookupKeyWrite(c->db,c->argv[1]);
|
robj *lobj = lookupKeyWrite(c->db,c->argv[1]);
|
||||||
c->argv[2] = tryObjectEncoding(c->argv[2]);
|
int may_have_waiting_clients = (lobj == NULL);
|
||||||
if (lobj == NULL) {
|
|
||||||
if (handleClientsWaitingListPush(c,c->argv[1],c->argv[2])) {
|
if (lobj && lobj->type != REDIS_LIST) {
|
||||||
addReply(c,shared.cone);
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
lobj = createZiplistObject();
|
|
||||||
dbAdd(c->db,c->argv[1],lobj);
|
|
||||||
} else {
|
|
||||||
if (lobj->type != REDIS_LIST) {
|
|
||||||
addReply(c,shared.wrongtypeerr);
|
addReply(c,shared.wrongtypeerr);
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
if (handleClientsWaitingListPush(c,c->argv[1],c->argv[2])) {
|
|
||||||
signalModifiedKey(c->db,c->argv[1]);
|
for (j = 2; j < c->argc; j++) {
|
||||||
addReply(c,shared.cone);
|
c->argv[j] = tryObjectEncoding(c->argv[j]);
|
||||||
return;
|
if (may_have_waiting_clients) {
|
||||||
|
if (handleClientsWaitingListPush(c,c->argv[1],c->argv[j])) {
|
||||||
|
addlen++;
|
||||||
|
continue;
|
||||||
|
} else {
|
||||||
|
may_have_waiting_clients = 0;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
listTypePush(lobj,c->argv[2],where);
|
if (!lobj) {
|
||||||
addReplyLongLong(c,listTypeLength(lobj));
|
lobj = createZiplistObject();
|
||||||
signalModifiedKey(c->db,c->argv[1]);
|
dbAdd(c->db,c->argv[1],lobj);
|
||||||
server.dirty++;
|
}
|
||||||
|
listTypePush(lobj,c->argv[j],where);
|
||||||
|
pushed++;
|
||||||
|
}
|
||||||
|
addReplyLongLong(c,addlen + (lobj ? listTypeLength(lobj) : 0));
|
||||||
|
if (pushed) signalModifiedKey(c->db,c->argv[1]);
|
||||||
|
server.dirty += pushed;
|
||||||
}
|
}
|
||||||
|
|
||||||
void lpushCommand(redisClient *c) {
|
void lpushCommand(redisClient *c) {
|
||||||
|
Loading…
x
Reference in New Issue
Block a user