bugfix: catch KeyError in check_version and retry

aiven · Feb 28, 2024 · c662d94 · c662d94
1 parent 1a6bf01
commit c662d94
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/kafka/client_async.py b/kafka/client_async.py
@@ -912,7 +912,13 @@ def check_version(self, node_id=None, timeout=2, strict=False):
                 self._lock.release()
                 raise Errors.NoBrokersAvailable()
             self._maybe_connect(try_node)
-            conn = self._conns[try_node]
+            try:
+                conn = self._conns[try_node]
+            except KeyError:
+                if node_id is not None:
+                    self._lock.release()
+                    raise Errors.NodeNotReadyError()
+                continue
 
             # We will intentionally cause socket failures
             # These should not trigger metadata refresh