DATAREDIS-1231 - Use Jedis command routing if multiple BRPOP keys map…

… to a single slot. Original pull request: #568.
spring-projects · Oct 13, 2020 · ead954a · ead954a
1 parent 41136c6
commit ead954a
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/src/main/java/org/springframework/data/redis/connection/jedis/JedisClusterListCommands.java b/src/main/java/org/springframework/data/redis/connection/jedis/JedisClusterListCommands.java
@@ -303,6 +303,14 @@ public List<byte[]> bRPop(int timeout, byte[]... keys) {
 		Assert.notNull(keys, "Key must not be null!");
 		Assert.noNullElements(keys, "Keys must not contain null elements!");
 
+		if (ClusterSlotHashUtil.isSameSlotForAllKeys(keys)) {
+			try {
+				return connection.getCluster().brpop(timeout,keys);
+			} catch (Exception ex) {
+				throw convertJedisAccessException(ex);
+			}
+		}
+
 		return connection.getClusterCommandExecutor()
 				.executeMultiKeyCommand(
 						(JedisMultiKeyClusterCommandCallback<List<byte[]>>) (client, key) -> client.brpop(timeout, key),