pvc/pvcd.py

112 lines
2.8 KiB
Python
Executable File

#!/usr/bin/env python3
from kazoo.client import KazooClient
from kazoo.client import KazooState
import libvirt
import sys
import socket
import uuid
import VMInstance
import NodeInstance
import time
import threading
import atexit
def help():
print("pvcd - Parallel Virtual Cluster management daemon")
# exit(0)
help()
# Connect to zookeeper
zk = KazooClient(hosts='127.0.0.1:2181')
try:
zk.start()
except:
print("Failed to connect to local Zookeeper instance")
exit(1)
def zk_listener(state):
if state == KazooState.LOST:
cleanup()
exit(2)
elif state == KazooState.SUSPENDED:
cleanup()
exit(2)
else:
# Handle being connected/reconnected to Zookeeper
pass
zk.add_listener(zk_listener)
myhostname = socket.gethostname()
mynodestring = '/nodes/%s' % myhostname
def cleanup():
t_node[myhostname].stop()
try:
zk.set('/nodes/' + myhostname + '/state', 'stop'.encode('ascii'))
except:
pass
zk.stop()
atexit.register(cleanup)
# Check if our node exists in Zookeeper, and create it if not
if zk.exists('%s' % mynodestring):
print("Node is present in Zookeeper")
else:
zk.create('%s' % mynodestring, 'hypervisor'.encode('ascii'))
zk.create('%s/state' % mynodestring, 'stop'.encode('ascii'))
zk.create('%s/cpucount' % mynodestring, '0'.encode('ascii'))
zk.create('%s/memfree' % mynodestring, '0'.encode('ascii'))
zk.create('%s/cpuload' % mynodestring, '0.0'.encode('ascii'))
t_node = dict()
s_domain = dict()
node_list = []
@zk.ChildrenWatch('/nodes')
def updatenodes(new_node_list):
node_list = new_node_list
active_node_list = []
flushed_node_list = []
inactive_node_list = []
print('Node list: %s' % node_list)
for node in node_list:
if node in t_node:
t_node[node].updatenodelist(node_list)
time.sleep(0.1)
else:
print("starting" + node)
t_node[node] = NodeInstance.NodeInstance(node, node_list, zk);
t_node[node].start()
time.sleep(0.1)
node_state = t_node[node].getstate()
print(node_state)
if node_state == 'start':
active_node_list.append(t_node[node].getname())
elif node_state == 'flush':
flushed_node_list.append(t_node[node].getname())
else:
inactive_node_list.append(t_node[node].getname())
print('Active nodes: %s' % active_node_list)
print('Flushed nodes: %s' % flushed_node_list)
print('Inactive nodes: %s' % inactive_node_list)
domain_list = zk.get_children('/domains')
print('Domain list: %s' % domain_list)
for domain in domain_list:
s_domain[domain] = VMInstance.VMInstance(domain, zk, t_node[myhostname]);
while True:
# Tick loop
try:
time.sleep(0.1)
except:
cleanup()
exit(0)