From e331261585bce06aa1c979d33f3932572115ba01 Mon Sep 17 00:00:00 2001 From: Dana Powers Date: Sun, 31 Mar 2019 10:12:05 -0700 Subject: Avoid race condition on client._conns in send() --- kafka/client_async.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) (limited to 'kafka/client_async.py') diff --git a/kafka/client_async.py b/kafka/client_async.py index b6adb77..3b5ca46 100644 --- a/kafka/client_async.py +++ b/kafka/client_async.py @@ -516,14 +516,15 @@ class KafkaClient(object): Returns: Future: resolves to Response struct or Error """ - if not self._can_send_request(node_id): + conn = self._conns.get(node_id) + if not conn or not self._can_send_request(node_id): self.maybe_connect(node_id, wakeup=wakeup) return Future().failure(Errors.NodeNotReadyError(node_id)) # conn.send will queue the request internally # we will need to call send_pending_requests() # to trigger network I/O - future = self._conns[node_id].send(request, blocking=False) + future = conn.send(request, blocking=False) # Wakeup signal is useful in case another thread is # blocked waiting for incoming network traffic while holding -- cgit v1.2.1