From 405a3a8f385dad753c5559540727a71c3365840e Mon Sep 17 00:00:00 2001
From: Joshua Boniface <joshua@boniface.me>
Date: Wed, 27 Jun 2018 23:37:39 -0400
Subject: [PATCH] Make cleanup function more robust

---
 pvcd.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/pvcd.py b/pvcd.py
index 12788468..e0fe191f 100755
--- a/pvcd.py
+++ b/pvcd.py
@@ -148,7 +148,11 @@ def cleanup(signum, frame):
     # Set stop state in Zookeeper
     zkhandler.writedata(zk_conn, { '/nodes/{}/daemonstate'.format(myhostname): 'stop' })
     # Close the Zookeeper connection
-    zk_conn.close()
+    try:
+        zk_conn.stop()
+        zk_conn.close()
+    except:
+        pass
     # Stop keepalive thread
     stopKeepaliveTimer(update_timer)
     # Exit