parallelvirtualcluster
/
pvc
1
0
Fork 0
Code Issues 11 Pull Requests Projects Releases 111 Wiki Activity

Handle client connection falures gracefully

This commit is contained in:
Joshua Boniface 2019-07-12 00:44:25 -04:00
parent 383bc958fe
commit e765ec7f49
1 changed files with 8 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
client-common/common.py
View File

@ -46,7 +46,14 @@ def validateUUID(dom_uuid):
# #
def startZKConnection(zk_host): def startZKConnection(zk_host):
zk_conn = kazoo.client.KazooClient(hosts=zk_host) zk_conn = kazoo.client.KazooClient(hosts=zk_host)
zk_conn.start() try:
zk_conn.start()
except kazoo.handlers.threading.KazooTimeoutError:
print('Timed out connecting to Zookeeper at "{}".'.format(zk_host))
exit(1)
except Exception as e:
print('Failed to connect to Zookeeper at "{}": {}'.format(zk_host, e))
exit(1)
return zk_conn return zk_conn
def stopZKConnection(zk_conn): def stopZKConnection(zk_conn):