Synnefo » snf-ganeti

#!/usr/bin/python

#

import sys

import optparse

from ganeti import opcodes

from ganeti import mcpu

from ganeti import objects

from ganeti import constants

from ganeti import cli

from ganeti import logger

from ganeti import errors

from ganeti import utils

USAGE = ("\tburnin -o OS_NAME [options...] instance_name ...")

def Usage():

  """Shows program usage information and exits the program."""

  print >> sys.stderr, "Usage:"

  print >> sys.stderr, USAGE

  sys.exit(2)

def Feedback(msg):

  """Simple function that prints out its argument.

  """

  print msg

def ParseOptions():

  """Parses the command line options.

  In case of command line errors, it will show the usage and exit the

  program.

  Returns:

    (options, args), as returned by OptionParser.parse_args

  """

  parser = optparse.OptionParser(usage="\n%s" % USAGE,

                                 version="%%prog (ganeti) %s" %

                                 constants.RELEASE_VERSION,

                                 option_class=cli.CliOption)

  parser.add_option("-o", "--os", dest="os", default=None,

                    help="OS to use during burnin",

                    metavar="<OS>")

  parser.add_option("--os-size", dest="os_size", help="Disk size",

                    default=4 * 1024, type="unit", metavar="<size>")

  parser.add_option("--swap-size", dest="swap_size", help="Swap size",

                    default=4 * 1024, type="unit", metavar="<size>")

  parser.add_option("-v", "--verbose",

                    action="store_true", dest="verbose", default=False,

                    help="print command execution messages to stdout")

  parser.add_option("--no-replace1", dest="do_replace1",

                    help="Do disk replacement with the same secondary",

                    action="store_false", default=True)

  parser.add_option("--no-replace2", dest="do_replace2",

                    help="Do disk replacement with a different secondary",

                    action="store_false", default=True)

  parser.add_option("--no-failover", dest="do_failover",

                    help="Do instance failovers", action="store_false",

                    default=True)

  parser.add_option("-t", "--disk-template", dest="disk_template",

                    choices=("remote_raid1", "drbd8"), default="remote_raid1",

                    help="Template type for network mirroring (remote_raid1"

                    " or drbd8) [remote_raid1]")

  options, args = parser.parse_args()

  if len(args) < 1 or options.os is None:

    Usage()

  return options, args

def BurninCluster(opts, args):

  """Test a cluster intensively.

  This will create instances and then start/stop/failover them.

  It is safe for existing instances but could impact performance.

  """

  logger.SetupLogging(debug=True, program="ganeti/burnin")

  proc = mcpu.Processor(feedback=Feedback)

  result = proc.ExecOpCode(opcodes.OpQueryNodes(output_fields=["name"],

                                                names=[]))

  nodelist = [data[0] for data in result]

  Feedback("- Testing global parameters")

  result = proc.ExecOpCode(opcodes.OpDiagnoseOS())

  if not result:

    Feedback("Can't get the OS list")

    return 1

  # filter non-valid OS-es

  oses = {}

  for node_name in result:

    oses[node_name] = [obj for obj in result[node_name]

                       if isinstance(obj, objects.OS)]

  fnode = oses.keys()[0]

  os_set = set([os_inst.name for os_inst in oses[fnode]])

  del oses[fnode]

  for node in oses:

    os_set &= set([os_inst.name for os_inst in oses[node]])

  if opts.os not in os_set:

    Feedback("OS '%s' not found" % opts.os)

    return 1

  to_remove = []

  if opts.disk_template == "remote_raid1":

    disk_template = constants.DT_REMOTE_RAID1

  elif opts.disk_template == "drbd8":

    disk_template = constants.DT_DRBD8

  else:

    Feedback("Unknown disk template '%s'" % opts.disk_template)

    return 1

  try:

    idx = 0

    for instance_name in args:

      next_idx = idx + 1

      if next_idx >= len(nodelist):

        next_idx = 0

      pnode = nodelist[idx]

      snode = nodelist[next_idx]

      if len(nodelist) > 1:

        tplate = disk_template

      else:

        tplate = constants.DT_PLAIN

      op = opcodes.OpCreateInstance(instance_name=instance_name, mem_size=128,

                                    disk_size=opts.os_size,

                                    swap_size=opts.swap_size,

                                    disk_template=tplate,

                                    mode=constants.INSTANCE_CREATE,

                                    os_type=opts.os, pnode=pnode,

                                    snode=snode, vcpus=1,

                                    start=True,

                                    ip_check=True,

                                    wait_for_sync=True)

      Feedback("- Add instance %s on node %s" % (instance_name, pnode))

      result = proc.ExecOpCode(op)

      to_remove.append(instance_name)

      idx = next_idx

    if opts.do_replace1:

      if len(nodelist) > 1:

        # failover

        for instance_name in args:

          op = opcodes.OpReplaceDisks(instance_name=instance_name,

                                      remote_node=None,

                                      mode=constants.REPLACE_DISK_ALL,

                                      disks=["sda", "sdb"])

          Feedback("- Replace disks for instance %s" % (instance_name))

          result = proc.ExecOpCode(op)

      else:

        Feedback("- Can't run replace1, not enough nodes")

    if opts.do_failover:

      if len(nodelist) > 1:

        # failover

        for instance_name in args:

          op = opcodes.OpFailoverInstance(instance_name=instance_name,

                                          ignore_consistency=True)

          Feedback("- Failover instance %s" % (instance_name))

          result = proc.ExecOpCode(op)

      else:

        Feedback("- Can't run failovers, not enough nodes")

    # stop / start

    for instance_name in args:

      op = opcodes.OpShutdownInstance(instance_name=instance_name)

      Feedback("- Shutdown instance %s" % instance_name)

      result = proc.ExecOpCode(op)

      op = opcodes.OpStartupInstance(instance_name=instance_name, force=False)

      Feedback("- Start instance %s" % instance_name)

      result = proc.ExecOpCode(op)

  finally:

    # remove

    for instance_name in to_remove:

      op = opcodes.OpRemoveInstance(instance_name=instance_name)

      Feedback("- Remove instance %s" % instance_name)

      result = proc.ExecOpCode(op)

  return 0

def main():

  """Main function"""

  opts, args = ParseOptions()

  try:

    utils.Lock('cmd', max_retries=15, debug=True)

  except errors.LockError, err:

    logger.ToStderr(str(err))

    return 1

  try:

    retval = BurninCluster(opts, args)

  finally:

    utils.Unlock('cmd')

    utils.LockCleanup()

  return retval

if __name__ == "__main__":

  main()