X-Git-Url: https://code.grnet.gr/git/ganeti-local/blobdiff_plain/aa089b6599bfc5b0bc82d86fe0beac565f296923..035566e317b53ea031bc515ed4d1397b862ace22:/tools/burnin diff --git a/tools/burnin b/tools/burnin index 84b506b..d0ef877 100755 --- a/tools/burnin +++ b/tools/burnin @@ -28,13 +28,11 @@ import sys import optparse import time import socket -import urllib2 -import errno +import urllib from itertools import izip, islice, cycle from cStringIO import StringIO from ganeti import opcodes -from ganeti import mcpu from ganeti import constants from ganeti import cli from ganeti import errors @@ -77,16 +75,36 @@ def Err(msg, exit_code=1): sys.stderr.flush() sys.exit(exit_code) + +class SimpleOpener(urllib.FancyURLopener): + """A simple url opener""" + + def prompt_user_passwd(self, host, realm, clear_cache = 0): + """No-interaction version of prompt_user_passwd.""" + return None, None + + def http_error_default(self, url, fp, errcode, errmsg, headers): + """Custom error handling""" + # make sure sockets are not left in CLOSE_WAIT, this is similar + # but with a different exception to the BasicURLOpener class + _ = fp.read() # throw away data + fp.close() + raise InstanceDown("HTTP error returned: code %s, msg %s" % + (errcode, errmsg)) + + class Burner(object): """Burner class.""" def __init__(self): """Constructor.""" utils.SetupLogging(constants.LOG_BURNIN, debug=False, stderr_logging=True) + self.url_opener = SimpleOpener() self._feed_buf = StringIO() self.nodes = [] self.instances = [] self.to_rem = [] + self.queued_ops = [] self.opts = None self.ParseOptions() self.cl = cli.GetClient() @@ -107,25 +125,52 @@ class Burner(object): if self.opts.verbose: Log(msg, indent=3) - def ExecOp(self, op): + def ExecOp(self, *ops): + """Execute one or more opcodes and manage the exec buffer. + + @result: if only opcode has been passed, we return its result; + otherwise we return the list of results + + """ + job_id = cli.SendJob(ops, cl=self.cl) + results = cli.PollJob(job_id, cl=self.cl, feedback_fn=self.Feedback) + if len(ops) == 1: + return results[0] + else: + return results + + def ExecOrQueue(self, name, *ops): """Execute an opcode and manage the exec buffer.""" - self.ClearFeedbackBuf() - return cli.SubmitOpCode(op, feedback_fn=self.Feedback, cl=self.cl) + if self.opts.parallel: + self.queued_ops.append((ops, name)) + else: + return self.ExecOp(*ops) + + def CommitQueue(self): + """Execute all submitted opcodes in case of parallel burnin""" + if not self.opts.parallel: + return + + try: + results = self.ExecJobSet(self.queued_ops) + finally: + self.queued_ops = [] + return results def ExecJobSet(self, jobs): """Execute a set of jobs and return once all are done. The method will return the list of results, if all jobs are - successfull. Otherwise, OpExecError will be raised from within + successful. Otherwise, OpExecError will be raised from within cli.py. """ self.ClearFeedbackBuf() - job_ids = [cli.SendJob(job, cl=self.cl) for job in jobs] - Log("Submitted job IDs %s" % ", ".join(job_ids), indent=1) + job_ids = [cli.SendJob(row[0], cl=self.cl) for row in jobs] + Log("Submitted job ID(s) %s" % ", ".join(job_ids), indent=1) results = [] - for jid in job_ids: - Log("Waiting for job %s" % jid, indent=2) + for jid, (_, iname) in zip(job_ids, jobs): + Log("waiting for job %s for %s" % (jid, iname), indent=2) results.append(cli.PollJob(jid, cl=self.cl, feedback_fn=self.Feedback)) return results @@ -204,7 +249,7 @@ class Burner(object): parser.add_option("-n", "--nodes", dest="nodes", default="", help="Comma separated list of nodes to perform" " the burnin on (defaults to all nodes)") - parser.add_option("--iallocator", dest="iallocator", + parser.add_option("-I", "--iallocator", dest="iallocator", default=None, type="string", help="Perform the allocation using an iallocator" " instead of fixed node spread (node restrictions no" @@ -278,12 +323,13 @@ class Burner(object): else: names = [] try: - op = opcodes.OpQueryNodes(output_fields=["name", "offline"], names=names) + op = opcodes.OpQueryNodes(output_fields=["name", "offline", "drained"], + names=names, use_locking=True) result = self.ExecOp(op) except errors.GenericError, err: err_code, msg = cli.FormatError(err) Err(msg, exit_code=err_code) - self.nodes = [data[0] for data in result if not data[1]] + self.nodes = [data[0] for data in result if not (data[1] or data[2])] result = self.ExecOp(opcodes.OpDiagnoseOS(output_fields=["name", "valid"], names=[])) @@ -297,7 +343,7 @@ class Burner(object): if self.opts.os not in os_set: Err("OS '%s' not found" % self.opts.os) - def CreateInstances(self): + def BurnCreateInstances(self): """Create the given instances. """ @@ -305,7 +351,6 @@ class Burner(object): mytor = izip(cycle(self.nodes), islice(cycle(self.nodes), 1, None), self.instances) - jobset = [] Log("Creating instances") for pnode, snode, instance in mytor: @@ -340,21 +385,15 @@ class Burner(object): hvparams=self.hvp, ) - if self.opts.parallel: - jobset.append([op]) - # FIXME: here we should not append to to_rem uncoditionally, - # but only when the job is successful - self.to_rem.append(instance) - else: - self.ExecOp(op) - self.to_rem.append(instance) - if self.opts.parallel: - self.ExecJobSet(jobset) + self.ExecOrQueue(instance, op) + self.to_rem.append(instance) + + self.CommitQueue() for instance in self.instances: self._CheckInstanceAlive(instance) - def GrowDisks(self): + def BurnGrowDisks(self): """Grow both the os and the swap disks by the requested amount, if any.""" Log("Growing disks") for instance in self.instances: @@ -364,21 +403,25 @@ class Burner(object): op = opcodes.OpGrowDisk(instance_name=instance, disk=idx, amount=growth, wait_for_sync=True) Log("increase disk/%s by %s MB" % (idx, growth), indent=2) - self.ExecOp(op) + self.ExecOrQueue(instance, op) + self.CommitQueue() - def ReplaceDisks1D8(self): + def BurnReplaceDisks1D8(self): """Replace disks on primary and secondary for drbd8.""" Log("Replacing disks on the same nodes") for instance in self.instances: Log("instance %s" % instance, indent=1) + ops = [] for mode in constants.REPLACE_DISK_SEC, constants.REPLACE_DISK_PRI: op = opcodes.OpReplaceDisks(instance_name=instance, mode=mode, disks=[i for i in range(self.disk_count)]) Log("run %s" % mode, indent=2) - self.ExecOp(op) + ops.append(op) + self.ExecOrQueue(instance, *ops) + self.CommitQueue() - def ReplaceDisks2(self): + def BurnReplaceDisks2(self): """Replace secondary node.""" Log("Changing the secondary node") mode = constants.REPLACE_DISK_CHG @@ -398,9 +441,10 @@ class Burner(object): iallocator=self.opts.iallocator, disks=[i for i in range(self.disk_count)]) Log("run %s %s" % (mode, msg), indent=2) - self.ExecOp(op) + self.ExecOrQueue(instance, op) + self.CommitQueue() - def Failover(self): + def BurnFailover(self): """Failover the instances.""" Log("Failing over instances") for instance in self.instances: @@ -408,26 +452,26 @@ class Burner(object): op = opcodes.OpFailoverInstance(instance_name=instance, ignore_consistency=False) - self.ExecOp(op) + self.ExecOrQueue(instance, op) + self.CommitQueue() for instance in self.instances: self._CheckInstanceAlive(instance) - def Migrate(self): + def BurnMigrate(self): """Migrate the instances.""" Log("Migrating instances") for instance in self.instances: Log("instance %s" % instance, indent=1) - op = opcodes.OpMigrateInstance(instance_name=instance, live=True, - cleanup=False) + op1 = opcodes.OpMigrateInstance(instance_name=instance, live=True, + cleanup=False) - Log("migration", indent=2) - self.ExecOp(op) - op = opcodes.OpMigrateInstance(instance_name=instance, live=True, - cleanup=True) - Log("migration cleanup", indent=2) - self.ExecOp(op) + op2 = opcodes.OpMigrateInstance(instance_name=instance, live=True, + cleanup=True) + Log("migration and migration cleanup", indent=2) + self.ExecOrQueue(instance, op1, op2) + self.CommitQueue() - def ImportExport(self): + def BurnImportExport(self): """Export the instance, delete it, and import it back. """ @@ -439,6 +483,11 @@ class Burner(object): for pnode, snode, enode, instance in mytor: Log("instance %s" % instance, indent=1) + # read the full name of the instance + nam_op = opcodes.OpQueryInstances(output_fields=["name"], + names=[instance], use_locking=True) + full_name = self.ExecOp(nam_op)[0][0] + if self.opts.iallocator: pnode = snode = None import_log_msg = ("import from %s" @@ -457,9 +506,6 @@ class Burner(object): shutdown=True) rem_op = opcodes.OpRemoveInstance(instance_name=instance, ignore_failures=True) - nam_op = opcodes.OpQueryInstances(output_fields=["name"], - names=[instance]) - full_name = self.ExecOp(nam_op)[0][0] imp_dir = os.path.join(constants.EXPORT_DIR, full_name) imp_op = opcodes.OpCreateInstance(instance_name=instance, disks = [ {"size": size} @@ -484,125 +530,131 @@ class Burner(object): erem_op = opcodes.OpRemoveExport(instance_name=instance) Log("export to node %s" % enode, indent=2) - self.ExecOp(exp_op) Log("remove instance", indent=2) - self.ExecOp(rem_op) - self.to_rem.remove(instance) Log(import_log_msg, indent=2) - self.ExecOp(imp_op) Log("remove export", indent=2) - self.ExecOp(erem_op) - - self.to_rem.append(instance) + self.ExecOrQueue(instance, exp_op, rem_op, imp_op, erem_op) + self.CommitQueue() for instance in self.instances: self._CheckInstanceAlive(instance) - def StopInstance(self, instance): + def StopInstanceOp(self, instance): """Stop given instance.""" - op = opcodes.OpShutdownInstance(instance_name=instance) - Log("shutdown", indent=2) - self.ExecOp(op) + return opcodes.OpShutdownInstance(instance_name=instance) - def StartInstance(self, instance): + def StartInstanceOp(self, instance): """Start given instance.""" - op = opcodes.OpStartupInstance(instance_name=instance, force=False) - Log("startup", indent=2) - self.ExecOp(op) + return opcodes.OpStartupInstance(instance_name=instance, force=False) - def RenameInstance(self, instance, instance_new): + def RenameInstanceOp(self, instance, instance_new): """Rename instance.""" - op = opcodes.OpRenameInstance(instance_name=instance, - new_name=instance_new) - Log("rename to %s" % instance_new, indent=2) - self.ExecOp(op) + return opcodes.OpRenameInstance(instance_name=instance, + new_name=instance_new) - def StopStart(self): + def BurnStopStart(self): """Stop/start the instances.""" Log("Stopping and starting instances") for instance in self.instances: Log("instance %s" % instance, indent=1) - self.StopInstance(instance) - self.StartInstance(instance) + op1 = self.StopInstanceOp(instance) + op2 = self.StartInstanceOp(instance) + self.ExecOrQueue(instance, op1, op2) + + self.CommitQueue() for instance in self.instances: self._CheckInstanceAlive(instance) - def Remove(self): + def BurnRemove(self): """Remove the instances.""" Log("Removing instances") for instance in self.to_rem: Log("instance %s" % instance, indent=1) op = opcodes.OpRemoveInstance(instance_name=instance, ignore_failures=True) - self.ExecOp(op) + self.ExecOrQueue(instance, op) + + self.CommitQueue() + + def BurnRename(self): + """Rename the instances. - def Rename(self): - """Rename the instances.""" + Note that this function will not execute in parallel, since we + only have one target for rename. + + """ Log("Renaming instances") rename = self.opts.rename for instance in self.instances: Log("instance %s" % instance, indent=1) - self.StopInstance(instance) - self.RenameInstance(instance, rename) - self.StartInstance(rename) + op_stop1 = self.StopInstanceOp(instance) + op_stop2 = self.StopInstanceOp(rename) + op_rename1 = self.RenameInstanceOp(instance, rename) + op_rename2 = self.RenameInstanceOp(rename, instance) + op_start1 = self.StartInstanceOp(rename) + op_start2 = self.StartInstanceOp(instance) + self.ExecOp(op_stop1, op_rename1, op_start1) self._CheckInstanceAlive(rename) - self.StopInstance(rename) - self.RenameInstance(rename, instance) - self.StartInstance(instance) - - for instance in self.instances: + self.ExecOp(op_stop2, op_rename2, op_start2) self._CheckInstanceAlive(instance) - def Reinstall(self): + def BurnReinstall(self): """Reinstall the instances.""" Log("Reinstalling instances") for instance in self.instances: Log("instance %s" % instance, indent=1) - self.StopInstance(instance) - op = opcodes.OpReinstallInstance(instance_name=instance) + op1 = self.StopInstanceOp(instance) + op2 = opcodes.OpReinstallInstance(instance_name=instance) Log("reinstall without passing the OS", indent=2) - self.ExecOp(op) - op = opcodes.OpReinstallInstance(instance_name=instance, - os_type=self.opts.os) + op3 = opcodes.OpReinstallInstance(instance_name=instance, + os_type=self.opts.os) Log("reinstall specifying the OS", indent=2) - self.ExecOp(op) - self.StartInstance(instance) + op4 = self.StartInstanceOp(instance) + self.ExecOrQueue(instance, op1, op2, op3, op4) + + self.CommitQueue() + for instance in self.instances: self._CheckInstanceAlive(instance) - def Reboot(self): + def BurnReboot(self): """Reboot the instances.""" Log("Rebooting instances") for instance in self.instances: Log("instance %s" % instance, indent=1) + ops = [] for reboot_type in constants.REBOOT_TYPES: op = opcodes.OpRebootInstance(instance_name=instance, reboot_type=reboot_type, ignore_secondaries=False) Log("reboot with type '%s'" % reboot_type, indent=2) - self.ExecOp(op) - self._CheckInstanceAlive(instance) + ops.append(op) + self.ExecOrQueue(instance, *ops) - def ActivateDisks(self): + self.CommitQueue() + + for instance in self.instances: + self._CheckInstanceAlive(instance) + + def BurnActivateDisks(self): """Activate and deactivate disks of the instances.""" Log("Activating/deactivating disks") for instance in self.instances: Log("instance %s" % instance, indent=1) + op_start = self.StartInstanceOp(instance) op_act = opcodes.OpActivateInstanceDisks(instance_name=instance) op_deact = opcodes.OpDeactivateInstanceDisks(instance_name=instance) + op_stop = self.StopInstanceOp(instance) Log("activate disks when online", indent=2) - self.ExecOp(op_act) - self.StopInstance(instance) Log("activate disks when offline", indent=2) - self.ExecOp(op_act) Log("deactivate disks (when offline)", indent=2) - self.ExecOp(op_deact) - self.StartInstance(instance) + self.ExecOrQueue(instance, op_act, op_stop, op_act, op_deact, op_start) + self.CommitQueue() for instance in self.instances: self._CheckInstanceAlive(instance) - def AddRemoveDisks(self): + def BurnAddRemoveDisks(self): """Add and remove an extra disk for the instances.""" Log("Adding and removing disks") for instance in self.instances: @@ -612,16 +664,16 @@ class Burner(object): disks=[(constants.DDM_ADD, {"size": self.disk_size[0]})]) op_rem = opcodes.OpSetInstanceParams(\ instance_name=instance, disks=[(constants.DDM_REMOVE, {})]) + op_stop = self.StopInstanceOp(instance) + op_start = self.StartInstanceOp(instance) Log("adding a disk", indent=2) - self.ExecOp(op_add) - self.StopInstance(instance) Log("removing last disk", indent=2) - self.ExecOp(op_rem) - self.StartInstance(instance) + self.ExecOrQueue(instance, op_add, op_stop, op_rem, op_start) + self.CommitQueue() for instance in self.instances: self._CheckInstanceAlive(instance) - def AddRemoveNICs(self): + def BurnAddRemoveNICs(self): """Add and remove an extra NIC for the instances.""" Log("Adding and removing NICs") for instance in self.instances: @@ -631,9 +683,9 @@ class Burner(object): op_rem = opcodes.OpSetInstanceParams(\ instance_name=instance, nics=[(constants.DDM_REMOVE, {})]) Log("adding a NIC", indent=2) - self.ExecOp(op_add) Log("removing last NIC", indent=2) - self.ExecOp(op_rem) + self.ExecOrQueue(instance, op_add, op_rem) + self.CommitQueue() def _CheckInstanceAlive(self, instance): """Check if an instance is alive by doing http checks. @@ -646,18 +698,18 @@ class Burner(object): """ if not self.opts.http_check: return - try: - for retries in range(self.opts.net_timeout): - try: - url = urllib2.urlopen("http://%s/hostname.txt" % instance) - except urllib2.URLError, err: - if err.args[0][0] == errno.ECONNREFUSED: - time.sleep(1) - continue - raise - except urllib2.URLError, err: - raise InstanceDown(instance, str(err)) + end_time = time.time() + self.opts.net_timeout + url = None + while time.time() < end_time and url is None: + try: + url = self.url_opener.open("http://%s/hostname.txt" % instance) + except IOError: + # here we can have connection refused, no route to host, etc. + time.sleep(1) + if url is None: + raise InstanceDown(instance, "Cannot contact instance") hostname = url.read().strip() + url.close() if hostname != instance: raise InstanceDown(instance, ("Hostname mismatch, expected %s, got %s" % (instance, hostname))) @@ -682,48 +734,48 @@ class Burner(object): has_err = True try: - self.CreateInstances() + self.BurnCreateInstances() if opts.do_replace1 and opts.disk_template in constants.DTS_NET_MIRROR: - self.ReplaceDisks1D8() + self.BurnReplaceDisks1D8() if (opts.do_replace2 and len(self.nodes) > 2 and opts.disk_template in constants.DTS_NET_MIRROR) : - self.ReplaceDisks2() + self.BurnReplaceDisks2() if (opts.disk_template != constants.DT_DISKLESS and utils.any(self.disk_growth, lambda n: n > 0)): - self.GrowDisks() + self.BurnGrowDisks() if opts.do_failover and opts.disk_template in constants.DTS_NET_MIRROR: - self.Failover() + self.BurnFailover() if opts.do_migrate and opts.disk_template == constants.DT_DRBD8: - self.Migrate() + self.BurnMigrate() if (opts.do_importexport and opts.disk_template not in (constants.DT_DISKLESS, constants.DT_FILE)): - self.ImportExport() + self.BurnImportExport() if opts.do_reinstall: - self.Reinstall() + self.BurnReinstall() if opts.do_reboot: - self.Reboot() + self.BurnReboot() if opts.do_addremove_disks: - self.AddRemoveDisks() + self.BurnAddRemoveDisks() if opts.do_addremove_nics: - self.AddRemoveNICs() + self.BurnAddRemoveNICs() if opts.do_activate_disks: - self.ActivateDisks() + self.BurnActivateDisks() if opts.rename: - self.Rename() + self.BurnRename() if opts.do_startstop: - self.StopStart() + self.BurnStopStart() has_err = False finally: @@ -732,7 +784,7 @@ class Burner(object): Log(self.GetFeedbackBuf()) Log("\n\n") if not self.opts.keep_instances: - self.Remove() + self.BurnRemove() return 0