Add proper exit

This commit is contained in:
Joshua Boniface 2018-06-16 22:32:16 -04:00
parent 261d9394db
commit 313181f48f
1 changed files with 2 additions and 1 deletions

View File

@ -116,10 +116,11 @@ zk.add_listener(zk_listener)
@atexit.register
def cleanup():
update_timer.shutdown()
zk.set('/nodes/{}/daemonstate'.format(myhostname), 'stop'.encode('ascii'))
zk.stop()
zk.close()
update_timer.shutdown()
exit(0)
# Handle SIGTERM gracefully
signal.signal(signal.SIGTERM, cleanup)