70 lines
2.1 KiB
Python
70 lines
2.1 KiB
Python
|
#!/usr/bin/python3
|
||
|
|
|
||
|
|
########################################
|
||
|
|
# Boot for Local:
|
||
|
|
# sudo ./main (or: sudo ./main [master-ipaddr])
|
||
|
|
#
|
||
|
|
|
||
|
|
########################################
|
||
|
|
# Usage for Local:
|
||
|
|
# curl -F uuid="lxc-name1" http://0.0.0.0:1729/v1/cgroup/container/sample
|
||
|
|
#
|
||
|
|
|
||
|
|
import time, sys, signal, json, subprocess, os
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
if not subprocess.getoutput('lsb_release -r -s 2>/dev/null').startswith('16.04'):
|
||
|
|
raise Exception('Ubuntu 16.04 LTS is required.')
|
||
|
|
|
||
|
|
if not os.path.exists('/sys/fs/cgroup/memory/memory.memsw.usage_in_bytes'):
|
||
|
|
raise Exception('Please append "swapaccount=1" to kernel.')
|
||
|
|
|
||
|
|
if subprocess.getoutput('whoami') != 'root':
|
||
|
|
raise Exception('Root privilege is required.')
|
||
|
|
|
||
|
|
from daemon.http import *
|
||
|
|
if len(sys.argv) == 1:
|
||
|
|
sys.argv.append('disable-network')
|
||
|
|
|
||
|
|
def signal_handler(signal, frame):
|
||
|
|
if sys.argv[1] == 'master':
|
||
|
|
subprocess.getoutput('ovs-vsctl del-br ovs-master >/dev/null 2>&1')
|
||
|
|
else:
|
||
|
|
subprocess.getoutput('ovs-vsctl del-br ovs-minion >/dev/null 2>&1')
|
||
|
|
sys.exit(0)
|
||
|
|
signal.signal(signal.SIGINT, signal_handler)
|
||
|
|
|
||
|
|
if sys.argv[1] != 'master': # for minions
|
||
|
|
from intra.cgroup import cgroup_manager
|
||
|
|
cgroup_manager.auto_detect_prefix()
|
||
|
|
cgroup_manager.set_default_memory_limit(4)
|
||
|
|
|
||
|
|
from intra.system import system_manager
|
||
|
|
system_manager.set_db_prefix('/var/lib/docklet/meter')
|
||
|
|
# system_manager.extend_swap(32)
|
||
|
|
|
||
|
|
if sys.argv[1] != 'disable-network':
|
||
|
|
from connector.minion import minion_connector
|
||
|
|
minion_connector.start(sys.argv[1])
|
||
|
|
else:
|
||
|
|
print("(No network mode)")
|
||
|
|
|
||
|
|
from policy.quota import identify_policy
|
||
|
|
from intra.smart import smart_controller
|
||
|
|
|
||
|
|
smart_controller.set_policy(identify_policy)
|
||
|
|
smart_controller.start()
|
||
|
|
|
||
|
|
print("Minion REST Daemon Starts Listening ..")
|
||
|
|
http = http_daemon_listener(minion_http_handler)
|
||
|
|
http.listen()
|
||
|
|
|
||
|
|
else: # for master: sudo ./main master
|
||
|
|
from connector.master import master_connector
|
||
|
|
master_connector.start()
|
||
|
|
|
||
|
|
print("Master REST Daemon Starts Listening ..")
|
||
|
|
http = http_daemon_listener(master_http_handler, master_connector)
|
||
|
|
http.listen()
|
||
|
|
|