Fix up some more bugs

This commit is contained in:
2018-06-01 00:35:39 -04:00
parent ae445d7770
commit 7bcb2e0b00
2 changed files with 14 additions and 8 deletions

11
pvcd.py
View File

@ -43,9 +43,11 @@ myhostname = socket.gethostname()
mynodestring = '/nodes/%s' % myhostname
def cleanup():
zk.set('%s/state' % mynodestring, 'stop'.encode('ascii'))
for node in node_list:
t_node[node].stop()
t_node[myhostname].stop()
try:
zk.set('/nodes/' + myhostname + '/state', 'stop'.encode('ascii'))
except:
pass
zk.stop()
atexit.register(cleanup)
@ -76,9 +78,8 @@ def updatenodes(new_node_list):
t_node[node].updatenodelist(node_list)
else:
t_node[node] = NodeInstance.NodeInstance(node, node_list, zk);
if t_node[node].name == myhostname:
if node == myhostname:
t_node[node].start()
time.sleep(1)
node_state = t_node[node].getstate()
if node_state == 'start':