5 # NodeManager plugin - first step of handling omf_controlled slices
8 Overwrites the 'resctl' tag of slivers controlled by OMF so slivermanager.py does the right thing
21 logger.log("omf_resctl: plugin starting up...")
23 def GetSlivers(data, conf = None, plc = None):
24 if 'accounts' not in data:
25 logger.log_missing_data("omf_resctl.GetSlivers",'accounts')
29 xmpp_server=data['xmpp']['server']
31 # we have the key but no value, just as bad
34 # disabled feature - bailing out
35 # xxx might need to clean up more deeply..
36 logger.log("PLC_OMF_ENABLED is not set -- plugin exiting")
39 # as hrn is set only at AddNode-time, upgraded myplcs might still miss this
40 # clue: just overwrite the hostname of all nodes
41 # for node in GetNodes(): UpdateNode(node['node_id'],{'hostname':node['hostname']})
43 node_hrn = data['hrn']
44 if not node_hrn: raise Exception
46 logger.log("Failed to read hrn from GetSlivers, using 'default' - *please upgrade PLCAPI*")
47 node_hrn='default # Failed to read hrn from GetSlivers, please upgrade PLCAPI'
49 for sliver in data['slivers']:
51 sliver_pub_key_dir=os.path.join("/home", name, ".ssh/")
52 sliver_private_key=os.path.join(sliver_pub_key_dir, "id_rsa")
53 for chunk in sliver['attributes']:
54 if chunk['tagname']=='omf_control':
55 # scan all versions of omf-resctl
56 etc_path="/vservers/%s/etc/"%name
57 pattern = etc_path + "omf-resctl-*/omf-resctl.yaml.in"
58 templates = glob.glob (pattern)
60 logger.log("WARNING: omf_resctl plugin, no template found for slice %s using pattern %s"\
63 for template in templates:
64 # remove the .in extension
66 # figure service name as subdir under etc/
67 service_name=os.path.split(template.replace(etc_path,''))[0]
68 # read template and replace
69 template_contents=file(template).read()
70 yaml_contents=template_contents\
71 .replace('@XMPP_SERVER@',xmpp_server)\
72 .replace('@NODE_HRN@',node_hrn)\
73 .replace('@SLICE_NAME@',name)\
74 .replace('@SLIVER_PRIVATE_KEY@',sliver_private_key)\
75 .replace('@SLIVER_PUB_KEY_DIR@',sliver_pub_key_dir)
76 changes=tools.replace_file_with_string(yaml,yaml_contents)
77 logger.log("yaml_contents length=%d, changes=%r"%(len(yaml_contents),changes))
79 sp=subprocess.Popen(['vserver',name,'exec','service',service_name,'restart'],
80 stdout=subprocess.PIPE,stderr=subprocess.STDOUT)
81 (output,retcod)=sp.communicate()
82 logger.log("omf_resctl: %s: restarted resource controller (retcod=%r)"%(name,retcod))
83 logger.log("omf_resctl: got output\n%s"%output)
85 logger.log("omf_resctl: %s: omf_control'ed sliver has no change" % name)