KAFKA-5512; Awake the heartbeat thread when timetoNextHeartbeat is eq…

…ual to 0 (dpkp#1439)
52388fa · Mar 11, 2018 · 9998604 · 9998604
1 parent c920a2a
commit 9998604
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/kafka/coordinator/base.py b/kafka/coordinator/base.py
@@ -314,6 +314,10 @@ def poll_heartbeat(self):
                     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):