From 9f3791fa781f979664edc0869eb2c02562abc4bb Mon Sep 17 00:00:00 2001
From: dvora-h <67596500+dvora-h@users.noreply.github.com>
Date: Sun, 30 Oct 2022 09:14:00 +0200
Subject: Fix KeyError in async cluster - initialize before execute multi key
 commands (#2439)

* Fix KeyError in async cluster

* link to issue

* typo
---
 redis/commands/cluster.py | 19 +++++++++++++++++++
 1 file changed, 19 insertions(+)

(limited to 'redis/commands/cluster.py')

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):
     """
-- 
cgit v1.2.1