Always call _safe_close() on KazooClient.stop() so we don't leak worker threads

1 files changed, 4 insertions, 2 deletions

diff --git a/kazoo/client.py b/kazoo/client.py
index 99d8bb7..2e760f3 100644
--- a/kazoo/client.py
+++ b/kazoo/client.py
@@ -595,8 +595,10 @@ class KazooClient(object):
 
         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."""