server/usr/sbin/invirt-remote-create

   1 #!/usr/bin/python
   2
   3 """
   4 Picks a host to "create" (boot) a VM on, and does so.
   5
   6 Current load-balancing algorithm: wherever there's more free RAM.
   7
   8 TODO: use a lock to avoid creating the same VM twice in a race
   9 """
  10
  11 from invirt.remote import bcast
  12 from subprocess import PIPE, Popen, call
  13 import sys
  14 import yaml
  15
  16 def choose_host():
  17     # Query each of the hosts.
  18     # XXX will the output of 'xm info' always be parseable YAML?
  19     results = bcast('availability')
  20     return max((int(o), s) for (s, o) in results)[1]
  21
  22 def main(argv):
  23     if len(argv) < 3:
  24         print >> sys.stderr, "usage: invirt-remote-create <operation> <machine> [<other args...>]"
  25         return 2
  26     operation = argv[1]
  27     machine_name = argv[2]
  28     args = argv[3:]
  29
  30     if operation == 'install':
  31         options = dict(arg.split('=', 1) for arg in args)
  32         valid_keys = set(('mirror', 'dist', 'arch', 'imagesize', 'noinstall'))
  33         if not set(options.keys()).issubset(valid_keys):
  34             print >> sys.stderr, "Invalid argument. Use the help command to see valid arguments to install"
  35             return 1
  36         if any(' ' in val for val in options.values()):
  37             print >> sys.stderr, "Arguments to the autoinstaller cannot contain spaces"
  38             return 1
  39
  40     p = Popen(['/usr/sbin/invirt-remote-proxy-web', 'listvms'], stdout=PIPE)
  41     output = p.communicate()[0]
  42     if p.returncode != 0:
  43         raise RuntimeError("Command '%s' returned non-zero exit status %d"
  44                            % ('invirt-remote-proxy-web', p.returncode))
  45     vms = yaml.load(output, yaml.CSafeLoader)
  46
  47     if machine_name in vms:
  48         host = vms[machine_name]['host']
  49         print >> sys.stderr, ("machine '%s' is already running on host %s"
  50                               % (machine_name, host))
  51         return 1
  52
  53     host = choose_host()
  54     print 'Creating on host %s...' % host
  55     sys.stdout.flush()
  56     return call(['remctl', host, 'remote', 'control',
  57                  machine_name, operation] + args)
  58
  59 if __name__ == '__main__':
  60     sys.exit(main(sys.argv))
  61
  62 # vim:et:sw=4:ts=4