Shutdown worker droid when zk conn is lost

vishalg0wda · Jun 15, 2017 · a34f948 · a34f948
1 parent 5ba4151
commit a34f948
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/hedroid/worker/zk_droid.py b/hedroid/worker/zk_droid.py
@@ -1,4 +1,5 @@
 import json
+import sys
 
 from kazoo.client import KazooClient, KazooState
 from kazoo.recipe.watchers import DataWatch
@@ -26,6 +27,7 @@ def conn_listener(state):
             logger.debug('connected...')
         elif state == KazooState.LOST:
             logger.debug('connection lost...')
+            sys.exit()
         elif state == KazooState.SUSPENDED:
             logger.debug('connection suspended...')