diff options
-rw-r--r-- | kazoo/client.py | 6 |
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.""" |