Stopping daemonized threads when references to these threads are lost

dpkp · Mar 2, 2015 · 0497924 · 0497924
1 parent 9ad0be6
commit 0497924
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/kafka/producer/base.py b/kafka/producer/base.py
@@ -127,6 +127,7 @@ def __init__(self, client, async=False,
         self.async = async
         self.req_acks = req_acks
         self.ack_timeout = ack_timeout
+        self.stopped = False
 
         if codec is None:
             codec = CODEC_NONE
@@ -212,3 +213,8 @@ def stop(self, timeout=1):
 
             if self.proc.is_alive():
                 self.proc.terminate()
+        self.stopped = True
+
+    def __del__(self):
+        if not self.stopped:
+            self.stop()
diff --git a/kafka/util.py b/kafka/util.py
@@ -151,3 +151,6 @@ def stop(self):
         # noinspection PyAttributeOutsideInit
         self.timer = None
         self.fn = None
+
+    def __del__(self):
+        self.stop()