Some cleanup of event handling here

This commit is contained in:
Joshua Boniface 2018-06-04 16:34:41 -04:00
parent df4f80cfb1
commit f1f6d10b0c
1 changed files with 15 additions and 15 deletions

View File

@ -12,6 +12,9 @@ class NodeInstance(threading.Thread):
self.state = 'stop' self.state = 'stop'
self.stop_thread = threading.Event() self.stop_thread = threading.Event()
self.t_node = t_node self.t_node = t_node
self.active_node_list = []
self.flushed_node_list = []
self.inactive_node_list = []
self.s_domain = s_domain self.s_domain = s_domain
self.domain_list = [] self.domain_list = []
@ -19,6 +22,10 @@ class NodeInstance(threading.Thread):
@zk.DataWatch(self.zkey + '/state') @zk.DataWatch(self.zkey + '/state')
def watch_hypervisor_state(data, stat, event=""): def watch_hypervisor_state(data, stat, event=""):
self.state = data.decode('ascii') self.state = data.decode('ascii')
if self.state == 'flush':
self.flush()
elif self.state == 'unflush':
self.unflush()
@zk.DataWatch(self.zkey + '/memfree') @zk.DataWatch(self.zkey + '/memfree')
def watch_hypervisor_memfree(data, stat, event=""): def watch_hypervisor_memfree(data, stat, event=""):
@ -56,12 +63,12 @@ class NodeInstance(threading.Thread):
self.stop_thread.set() self.stop_thread.set()
# Flush all VMs on the host # Flush all VMs on the host
def flush(self, active_node_list): def flush(self):
for domain in self.domain_list: for domain in self.domain_list:
# Determine the best target hypervisor # Determine the best target hypervisor
least_mem = 2**64 least_mem = 2**64
least_host = None least_host = None
for node_name in active_node_list: for node_name in self.t_node if self.t_node[node_name].getstate() == 'start':
# It should never include itself, but just in case # It should never include itself, but just in case
if node_name == self.name: if node_name == self.name:
continue continue
@ -170,29 +177,22 @@ class NodeInstance(threading.Thread):
print(">>> %s - Free memory: %s | Load: %s" % ( time.strftime("%d/%m/%Y %H:%M:%S"), self.memfree, self.cpuload )) print(">>> %s - Free memory: %s | Load: %s" % ( time.strftime("%d/%m/%Y %H:%M:%S"), self.memfree, self.cpuload ))
print("Active domains: %s" % self.domain_list) print("Active domains: %s" % self.domain_list)
active_node_list = []
flushed_node_list = []
inactive_node_list = []
for node_name in self.t_node: for node_name in self.t_node:
state, stat = self.zk.get('/nodes/%s/state' % node_name) state, stat = self.zk.get('/nodes/%s/state' % node_name)
node_state = state.decode('ascii') node_state = state.decode('ascii')
if node_state == 'start': if node_state == 'start':
active_node_list.append(node_name) self.active_node_list.append(node_name)
elif node_state == 'flush': elif node_state == 'flush':
flushed_node_list.append(node_name) self.flushed_node_list.append(node_name)
else: else:
inactive_node_list.append(node_name) self.inactive_node_list.append(node_name)
print('Active nodes: %s' % active_node_list) print('Active nodes: %s' % self.active_node_list)
print('Flushed nodes: %s' % flushed_node_list) print('Flushed nodes: %s' % self.flushed_node_list)
print('Inactive nodes: %s' % inactive_node_list) print('Inactive nodes: %s' % self.inactive_node_list)
# Do any actions my node requires # Do any actions my node requires
if self.state == 'flush':
self.flush(active_node_list)
elif self.state == 'unflush':
self.unflush()
# Sleep for 10s but with quick interruptability # Sleep for 10s but with quick interruptability
for x in range(0,100): for x in range(0,100):