Changeset 816 for trunk/packages/sipb-xen-remote-server/files/usr/sbin

Timestamp:

Aug 2, 2008, 8:28:18 PM (16 years ago)

Author:

y_z

Message:

• use invirt.config to get hostnames
• refactoring: extracted bcast() function into invirt.remote package
• fixed os.rename import bug
• using correct default paths in invirt-getconf

Location:

trunk/packages/sipb-xen-remote-server/files/usr/sbin

Files:

• sipb-xen-remote-create (modified) (3 diffs)
• sipb-xen-remote-listvms (modified) (2 diffs)

trunk/packages/sipb-xen-remote-server/files/usr/sbin/sipb-xen-remote-create

@@ -1 +1 @@
 #!/usr/bin/python
+
 """
 Picks a host to "create" (boot) a VM on, and does so.
@@ -8 +9 @@
 """
 
-
+from invirt.remote import bcast
 from subprocess import PIPE, Popen, call
 import sys
 import yaml
 
-
 def choose_host():
     # Query each of the hosts.
-    # TODO get `servers` from a real list of all the VM hosts (instead of
-    # hardcoding the list here)
-    servers = ['black-mesa.mit.edu', 'sx-blade-2.mit.edu']
-    pipes = [(server,
-              Popen(['remctl', server, 'remote', 'web', 'info'], stdout=PIPE))
-             for server in servers]
-    outputs = [(s, p.communicate()[0]) for (s, p) in pipes]
-    for (s, p) in pipes:
-        if p.returncode != 0:
-            raise RuntimeError("remctl to host %s returned non-zero exit status %d"
-                               % (s, p.returncode)) 
-    results = [(s, yaml.load(o, yaml.CSafeLoader)) for (s, o) in outputs]
     # XXX will the output of 'xm info' always be parseable YAML?
-
-    return max( (int(o['free_memory']), s) for (s, o) in results )[1]
-
+    results = bcast('info')
+    return max((int(o['free_memory']), s) for (s, o) in results)[1]
 
 def main(argv):
     if len(argv) < 2:
-        print >>sys.stderr, "usage: sipb-xen-remote-create <machine> [<other args...>]"
+        print >> sys.stderr, "usage: sipb-xen-remote-create <machine> [<other args...>]"
         return 2
     machine_name = argv[1]
     args = argv[2:]
-
 
     p = Popen(['/usr/sbin/sipb-xen-remote-proxy-web', 'listvms'], stdout=PIPE)
@@ -50 +36 @@
     if machine_name in vms:
         host = vms[machine_name]['host']
-        print >>sys.stderr, ("machine '%s' is already running on host %s"
-                             % (machine_name, host))
+        print >> sys.stderr, ("machine '%s' is already running on host %s"
+                              % (machine_name, host))
         return 1
-
 
     host = choose_host()

trunk/packages/sipb-xen-remote-server/files/usr/sbin/sipb-xen-remote-listvms

@@ -6 +6 @@
 """
 
-from subprocess import PIPE, Popen
+from invirt.remote import bcast
 import sys
 import yaml
 
 def main(argv):
-    # Query each of the server for their VMs.
-    # TODO get `servers` from a real list of all the VM hosts (instead of
-    # hardcoding the list here)
-    servers = ['black-mesa.mit.edu', 'sx-blade-2.mit.edu']
-    # XXX
-    pipes = [(server,
-              Popen(['remctl', server, 'remote', 'web', 'listvms'], stdout=PIPE))
-             for server in servers]
-    outputs = [(s, p.communicate()[0]) for (s, p) in pipes]
-    for (s, p) in pipes:
-        if p.returncode != 0:
-            raise RuntimeError("remctl to host %s returned non-zero exit status %d"
-                               % (s, p.returncode)) 
-    results = [(s, yaml.load(o, yaml.CSafeLoader)) for (s, o) in outputs]
-    results = filter(lambda (_, x): x is not None, results)
+    # Query each of the hosts.
+    results = filter(lambda (_, x): x is not None, bcast('listvms'))
 
     # Merge the results and print.
@@ -37 +24 @@
 
 if __name__ == '__main__':
-    main(sys.argv)
+    sys.exit(main(sys.argv))
 
 # vim:et:sw=4:ts=4