Fix connection disconnected before idel check on cluster pool

Derived from NoneGG/aredis#192
alisaifee · Jan 17, 2022 · 34f90a9 · 34f90a9
1 parent 13867d7
commit 34f90a9
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/coredis/pool.py b/coredis/pool.py
@@ -562,9 +562,10 @@ async def disconnect_on_idle_time_exceeded(self, connection):
             ):
                 connection.disconnect()
                 node = connection.node
-                self._available_connections[node["name"]].remove(connection)
+                connections = self._available_connections.get(node["name"])
+                if connections:
+                    connections.remove(connection)
                 self._created_connections_per_node[node["name"]] -= 1
-
                 break
             await asyncio.sleep(self.idle_check_interval)