pvc/pvcd.py

91 lines
2.0 KiB
Python
Raw Normal View History

2018-05-31 20:26:44 -04:00
#!/usr/bin/env python3
from kazoo.client import KazooClient
from kazoo.client import KazooState
import libvirt
import sys
2018-05-31 21:49:23 -04:00
import socket
2018-05-31 20:26:44 -04:00
import uuid
import VMInstance
2018-05-31 21:49:23 -04:00
import NodeInstance
2018-05-31 20:26:44 -04:00
import time
2018-05-31 21:49:23 -04:00
import threading
import atexit
2018-05-31 20:26:44 -04:00
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:
2018-05-31 22:31:20 -04:00
cleanup()
exit(2)
2018-05-31 20:26:44 -04:00
elif state == KazooState.SUSPENDED:
2018-05-31 22:31:20 -04:00
cleanup()
exit(2)
2018-05-31 20:26:44 -04:00
else:
# Handle being connected/reconnected to Zookeeper
pass
zk.add_listener(zk_listener)
2018-05-31 22:31:20 -04:00
myhostname = socket.gethostname()
2018-05-31 21:49:23 -04:00
def cleanup():
2018-05-31 22:31:20 -04:00
for node in node_list:
t_node[node].stop()
2018-05-31 21:49:23 -04:00
zk.stop()
2018-05-31 20:26:44 -04:00
2018-05-31 21:49:23 -04:00
atexit.register(cleanup)
2018-05-31 20:26:44 -04:00
2018-05-31 22:55:44 -04:00
# Check if our node exists in Zookeeper, and create it if not
mynodestring = '/nodes/%s' % myhostname
if not zk.exists('%s' + mynodestring):
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'))
time.sleep(1)
2018-05-31 23:01:22 -04:00
def updatenodes():
node_list = zk.get_children('/nodes')
print(node_list)
for node in node_list:
if t_node[node] is None:
t_node[node] = NodeInstance.NodeInstance(node, node_list, zk);
else:
t_node[node].updatenodelist(node_list)
2018-05-31 21:49:23 -04:00
node_list = zk.get_children('/nodes')
2018-05-31 20:26:44 -04:00
2018-05-31 21:49:23 -04:00
domain_list = zk.get_children('/domains')
print(domain_list)
2018-05-31 20:26:44 -04:00
2018-05-31 21:49:23 -04:00
t_node = dict()
s_domain = dict()
2018-05-31 20:26:44 -04:00
2018-05-31 22:55:44 -04:00
time.sleep(1)
2018-05-31 21:49:23 -04:00
for domain in domain_list:
2018-05-31 22:31:20 -04:00
s_domain[domain] = VMInstance.VMInstance(domain, zk, t_node[myhostname]);
2018-05-31 20:26:44 -04:00
while True:
# Tick loop
2018-05-31 21:49:23 -04:00
try:
2018-05-31 22:31:20 -04:00
time.sleep(0.1)
2018-05-31 21:49:23 -04:00
except:
cleanup()
exit(0)