From 9091250d93c9aefc50c863a60c7c9d3a8d446122 Mon Sep 17 00:00:00 2001
From: Dana Powers <dana.powers@rd.io>
Date: Sat, 10 Mar 2018 14:15:12 -0500
Subject: KAFKA-5512; Awake the heartbeat thread when timetoNextHeartbeat is
 equal to 0

---
 kafka/coordinator/base.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/kafka/coordinator/base.py b/kafka/coordinator/base.py
index 57da971..bff6286 100644
--- a/kafka/coordinator/base.py
+++ b/kafka/coordinator/base.py
@@ -314,6 +314,10 @@ class BaseCoordinator(object):
                     cause = self._heartbeat_thread.failed
                     self._heartbeat_thread = None
                     raise cause  # pylint: disable-msg=raising-bad-type
+
+                # Awake the heartbeat thread if needed
+                if self.heartbeat.should_heartbeat():
+                    self._lock.notify()
                 self.heartbeat.poll()
 
     def time_to_next_heartbeat(self):
-- 
cgit v1.2.1