106 lines
2.7 KiB
Python
Executable File
106 lines
2.7 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()
|
|
time.sleep(0.2)
|
|
try:
|
|
zk.set('/nodes/' + myhostname + '/state', 'stop'.encode('ascii'))
|
|
zk.stop()
|
|
zk.close()
|
|
except:
|
|
pass
|
|
|
|
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'))
|
|
zk.create('%s/runningdomains' % mynodestring, ''.encode('ascii'))
|
|
|
|
t_node = dict()
|
|
s_domain = dict()
|
|
node_list = []
|
|
|
|
@zk.ChildrenWatch('/nodes')
|
|
def updatenodes(new_node_list):
|
|
node_list = new_node_list
|
|
print('Node list: %s' % node_list)
|
|
for node in node_list:
|
|
if node in t_node:
|
|
t_node[node].updatenodelist(t_node)
|
|
else:
|
|
t_node[node] = NodeInstance.NodeInstance(node, t_node, s_domain, zk);
|
|
if node == myhostname:
|
|
t_node[node].start()
|
|
time.sleep(0.2)
|
|
for node in node_list:
|
|
if node in t_node:
|
|
t_node[node].updatenodelist(t_node)
|
|
|
|
@zk.ChildrenWatch('/domains')
|
|
def updatedomains(new_domain_list):
|
|
domain_list = new_domain_list
|
|
print('Domain list: %s' % domain_list)
|
|
for domain in domain_list:
|
|
if not domain in s_domain:
|
|
s_domain[domain] = VMInstance.VMInstance(domain, zk, t_node[myhostname]);
|
|
for node in node_list:
|
|
if node in t_node:
|
|
t_node[node].updatedomainlist(s_domain)
|
|
|
|
while True:
|
|
# Tick loop
|
|
try:
|
|
time.sleep(0.1)
|
|
except:
|
|
cleanup()
|
|
exit(0)
|