Changeset 1ae1aa2 for fedd/federation/emulab_segment.py

Timestamp:

Nov 30, 2011 5:55:13 PM (12 years ago)

Author:

Ted Faber <faber@…>

Branches:

compt_changes, info-ops, master

Children:

f671ef7

Parents:

b709861

Message:

Reload interface works

File:

• fedd/federation/emulab_segment.py (modified) (8 diffs)

fedd/federation/emulab_segment.py

@@ -3 +3 @@
 import logging 
 import util
+import topdl
 
 from M2Crypto import SSL
@@ -17 +18 @@
 
 class emulab_segment:
+    class node_info:
+        def __init__(self, name, pname, status=None, osname=None, 
+                osversion=None, image=None):
+            self.name = name
+            self.pname = pname
+            self.status = status
+            self.osname = osname
+            self.osversion = osversion
+            self.image = image
+        def getOS(self):
+            rv = None
+            if self.osname or self.osversion:
+                rv = topdl.OperatingSystem(name=self.osname, 
+                        version=self.osversion)
+            if self.image and rv:
+                rv.set_attribute('image', self.image)
+            return rv
+
     def __init__(self, boss, cert):
         self.ctxt = fedd_ssl_context(my_cert=cert)
@@ -34 +53 @@
         self.node = { }
         self.status = [ ]
+        self.node_info = emulab_segment.node_info
 
     def emulab_call(self, method, params):
@@ -207 +227 @@
             return code == 0
 
+    def get_osid_map(self):
+        oses = { }
+        code, osids = self.emulab_call('osid.getlist', {})
+        for key, val in osids.items():
+            val['imageid'] = key
+            oses[val['osid']] = val
+        return oses
 
     def get_mapping(self, pid, eid):
@@ -223 +250 @@
             if self.log:
                 self.log.info("[get_mapping] Generating mapping")
+
+            osidmap = self.get_osid_map()
 
             params = {
@@ -235 +264 @@
                         st = v.get('status', 'up')
                         ev = v.get('eventstatus', 'ISUP')
+                        os = v.get('osid', None)
 
                         if st == 'up' and ev in ev_ok: st = 'active'
                         else: st = 'failed'
-                        self.node[k] = (v['pnode'], st)
+
+                        if os and os in osidmap:
+                           osname = osidmap[os].get('OS', None)
+                           osversion = osidmap[os].get('version', None)
+                           osimage = "%s/%s" % \
+                                   ( osidmap[os].get('pid', ''), 
+                                           osidmap[os].get('imageid', ''))
+                        else:
+                            osname = osversion = osimage = None
+
+                        self.node[k] = self.node_info(k, v['pnode'], st,
+                                osname, osversion, osimage)
                 if self.log:
                     self.log.info("Mapping complete")
@@ -245 +286 @@
                 raise service_error(service_error.internal,
                         "Cannot get node mapping of segment:%s/%s" % (pid, eid))
-
-    def do_operation(self, op, lnode, pnode, params):
+    def get_image(self, node, top):
+        for e in top.elements:
+            if isinstance(e, topdl.Computer):
+                if node == e.name:
+                    if e.os: 
+                        return e.os[0].get_attribute('image')
+        return None
+
+
+    def do_operation(self, op, lnode, pnode, params, top):
         """
         Carry out operation on node in the given experiment.
@@ -268 +317 @@
                 return False
             elif state == 'initial': 
-                self.status.append(operation_status(lnode,
-                        operation_status.unsupp, 'Image load unsupported'))
-                return False
+                image = self.get_image(lnode, top)
+                if image:
+                    pid, iid = image.split('/')
+                    p = {'nodes': pnode, 'imagename': iid, 'imageproj': pid, 
+                            'wait': False}
+                    code, result = self.emulab_call('node.reload', p)
+                    if code == 0: 
+                        self.status.append(operation_status(lnode,
+                            operation_status.success, 'reloading'))
+                        return True
+                    else:
+                        self.status.append(operation_status(lnode,
+                                operation_status.federant, 
+                                'Error code: %d' % code))
+                        return False
+                else:
+                    self.status.append(operation_status(lnode,
+                            operation_status.federant, 
+                            'cannot find imageid??'))
+                    return False
             elif state == 'boot': 
                 p = {'nodes': pnode, 'wait': False}