Always call _safe_close() on KazooClient.stop() so we don't leak work…

…er threads
ceache · Mar 1, 2017 · fa47be4 · fa47be4
1 parent b77d8de
commit fa47be4
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/kazoo/client.py b/kazoo/client.py
@@ -595,8 +595,10 @@ def stop(self):
 
         self._stopped.set()
         self._queue.append((CloseInstance, None))
-        self._connection._write_sock.send(b'\0')
-        self._safe_close()
+        try:
+            self._connection._write_sock.send(b'\0')
+        finally:
+            self._safe_close()
 
     def restart(self):
         """Stop and restart the Zookeeper session."""