plugins/omf_resctl.py

   1 #
   2 # $Id$
   3 # $URL$
   4 #
   5 # NodeManager plugin - first step of handling omf_controlled slices
   6
   7 """
   8 Overwrites the 'resctl' tag of slivers controlled by OMF so slivermanager.py does the right thing
   9 """
  10
  11 import os, os.path
  12 import glob
  13 import subprocess
  14
  15 import tools
  16 import logger
  17
  18 priority = 50
  19
  20 def start():
  21     logger.log("omf_resctl: plugin starting up...")
  22
  23 def GetSlivers(data, conf = None, plc = None):
  24     if 'accounts' not in data:
  25         logger.log_missing_data("omf_resctl.GetSlivers",'accounts')
  26         return
  27
  28     try:
  29         xmpp_server=data['xmpp']['server']
  30         if not xmpp_server:
  31             # we have the key but no value, just as bad
  32             raise Exception
  33     except:
  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")
  37         return
  38
  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']})
  42     try:
  43         node_hrn = data['hrn']
  44         if not node_hrn: raise Exception
  45     except:
  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'
  48
  49     for sliver in data['slivers']:
  50         name=sliver['name']
  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)
  59                 if not templates:
  60                     logger.log("WARNING: omf_resctl plugin, no template found for slice %s using pattern %s"\
  61                                    %(name,pattern))
  62                     continue
  63                 for template in templates:
  64                     # remove the .in extension
  65                     yaml=template[:-3]
  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))
  78                     if 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)
  84                     else:
  85                         logger.log("omf_resctl: %s: omf_control'ed sliver has no change" % name)