feat(proxy): PUBSUB command should send to first write resource (#279)

netease-im · Aug 7, 2024 · e5ee2dc · e5ee2dc
1 parent 2bb5d4e
commit e5ee2dc
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/.../main/java/com/netease/nim/camellia/redis/proxy/upstream/UpstreamRedisClientTemplate.java b/.../main/java/com/netease/nim/camellia/redis/proxy/upstream/UpstreamRedisClientTemplate.java
@@ -266,6 +266,13 @@ public List<CompletableFuture<Reply>> sendCommand(int db, List<Command> commands
                     futureList.add(future);
                     continue;
                 }
+                if (redisCommand == RedisCommand.PUBSUB) {
+                    //send to first write resource
+                    List<Resource> writeResources = resourceSelector.getWriteResources(Utils.EMPTY_ARRAY);
+                    CompletableFuture<Reply> future = doWrite(Collections.singletonList(writeResources.get(0)), commandFlusher, command);
+                    futureList.add(future);
+                    continue;
+                }
                 //other commands
             }