summaryrefslogtreecommitdiff
path: root/redis/commands/cluster.py
diff options
context:
space:
mode:
authordvora-h <67596500+dvora-h@users.noreply.github.com>2022-10-30 09:14:00 +0200
committerGitHub <noreply@github.com>2022-10-30 09:14:00 +0200
commit9f3791fa781f979664edc0869eb2c02562abc4bb (patch)
treef2ae1001f8120e55136d6e4f5bdd6a222e44a514 /redis/commands/cluster.py
parent51797765d6c9cf72776eaa031b0ebb1e6a5ed77a (diff)
downloadredis-py-9f3791fa781f979664edc0869eb2c02562abc4bb.tar.gz
Fix KeyError in async cluster - initialize before execute multi key commands (#2439)
* Fix KeyError in async cluster * link to issue * typo
Diffstat (limited to 'redis/commands/cluster.py')
-rw-r--r--redis/commands/cluster.py19
1 files changed, 19 insertions, 0 deletions
diff --git a/redis/commands/cluster.py b/redis/commands/cluster.py
index 8e4c8ef..f0eaaf7 100644
--- a/redis/commands/cluster.py
+++ b/redis/commands/cluster.py
@@ -316,6 +316,25 @@ class AsyncClusterMultiKeyCommands(ClusterMultiKeyCommands):
# Sum up the reply from each command
return sum(await self._execute_pipeline_by_slot(command, slots_to_keys))
+ async def _execute_pipeline_by_slot(
+ self, command: str, slots_to_args: Mapping[int, Iterable[EncodableT]]
+ ) -> List[Any]:
+ if self._initialize:
+ await self.initialize()
+ read_from_replicas = self.read_from_replicas and command in READ_COMMANDS
+ pipe = self.pipeline()
+ [
+ pipe.execute_command(
+ command,
+ *slot_args,
+ target_nodes=[
+ self.nodes_manager.get_node_from_slot(slot, read_from_replicas)
+ ],
+ )
+ for slot, slot_args in slots_to_args.items()
+ ]
+ return await pipe.execute()
+
class ClusterManagementCommands(ManagementCommands):
"""