rewrite BRPOPLPUSH as RPOPLPUSH to propagate

huangboran123 · Mar 12, 2019 · 89f828f · 89f828f
1 parent 864f8d2
commit 89f828f
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 1 deletion.
diff --git a/src/server.c b/src/server.c
@@ -1376,6 +1376,7 @@ void createSharedObjects(void) {
     shared.rpop = createStringObject("RPOP", 4);
     shared.lpop = createStringObject("LPOP", 4);
     shared.lpush = createStringObject("LPUSH", 5);
+	shared.rpoplpush = createStringObject("RPOPLPUSH",9);
     for (j = 0; j < OBJ_SHARED_INTEGERS; j++) {
         shared.integers[j] =
             makeObjectShared(createObject(OBJ_STRING, (void*) (PORT_LONG) j));

diff --git a/src/server.h b/src/server.h
@@ -764,7 +764,7 @@ struct sharedObjectsStruct {
     *masterdownerr, *roslaveerr, *execaborterr, *noautherr, *noreplicaserr,
     *busykeyerr, *oomerr, *plus, *messagebulk, *pmessagebulk, *subscribebulk,
     *unsubscribebulk, *psubscribebulk, *punsubscribebulk, *del, *unlink,
-    *rpop, *lpop, *lpush, *emptyscan,
+    *rpop, *lpop, *lpush, *rpoplpush, *emptyscan,
     *select[PROTO_SHARED_SELECT_CMDS],
     *integers[OBJ_SHARED_INTEGERS],
     *mbulkhdr[OBJ_SHARED_BULKHDR_LEN], /* "*<value>\r\n" */

diff --git a/src/t_list.c b/src/t_list.c
@@ -596,6 +596,9 @@ void rpoplpushCommand(client *c) {
         signalModifiedKey(c->db,touchedkey);
         decrRefCount(touchedkey);
         server.dirty++;
+		if (c->cmd->proc == brpoplpushCommand) {
+			rewriteClientCommandVector(c,3,shared.rpoplpush,c->argv[1],c->argv[2]);
+		}
     }
 }