do not block on heartbeat thread

cenkalti · Mar 9, 2018 · 4dfde9c · 4dfde9c
1 parent c32b883
commit 4dfde9c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/kuyruk/heartbeat.py b/kuyruk/heartbeat.py
@@ -27,10 +27,10 @@ def stop(self) -> None:
         self._thread.join()
 
     def _run(self) -> None:
-        while not self._stop.is_set():
+        while not self._stop.wait(timeout=1):
             try:
                 self._connection.heartbeat_tick()
-                self._connection.drain_events(timeout=1)
+                self._connection.drain_events(timeout=0)
             except socket.timeout:
                 pass
             except Exception as e: