X-Git-Url: https://code.grnet.gr/git/ganeti-local/blobdiff_plain/3a2c7775b37d3bf96bdfb6f31f9ee2a612ad634b..2d54e29cd20fa61ff4337f5332a77075cb9fb7b5:/daemons/ganeti-masterd diff --git a/daemons/ganeti-masterd b/daemons/ganeti-masterd index 11b779e..4d783a8 100755 --- a/daemons/ganeti-masterd +++ b/daemons/ganeti-masterd @@ -1,4 +1,4 @@ -#!/usr/bin/python -u +#!/usr/bin/python # # Copyright (C) 2006, 2007 Google Inc. @@ -26,22 +26,22 @@ inheritance from parent classes requires it. """ +# pylint: disable-msg=C0103 +# C0103: Invalid name ganeti-masterd +import os import sys import SocketServer import time import collections -import Queue -import random import signal -import simplejson import logging -from cStringIO import StringIO from optparse import OptionParser from ganeti import config from ganeti import constants +from ganeti import daemon from ganeti import mcpu from ganeti import opcodes from ganeti import jqueue @@ -50,8 +50,10 @@ from ganeti import luxi from ganeti import utils from ganeti import errors from ganeti import ssconf -from ganeti import logger from ganeti import workerpool +from ganeti import rpc +from ganeti import bootstrap +from ganeti import serializer CLIENT_REQUEST_WORKERS = 16 @@ -61,6 +63,7 @@ EXIT_NODESETUP_ERROR = constants.EXIT_NODESETUP_ERROR class ClientRequestWorker(workerpool.BaseWorker): + # pylint: disable-msg=W0221 def RunTask(self, server, request, client_address): """Process the request. @@ -70,7 +73,7 @@ class ClientRequestWorker(workerpool.BaseWorker): try: server.finish_request(request, client_address) server.close_request(request) - except: + except: # pylint: disable-msg=W0702 server.handle_error(request, client_address) server.close_request(request) @@ -83,24 +86,21 @@ class IOServer(SocketServer.UnixStreamServer): cleanup at shutdown. """ - def __init__(self, address, rqhandler, context): + def __init__(self, address, rqhandler): """IOServer constructor - Args: - address: the address to bind this IOServer to - rqhandler: RequestHandler type object - context: Context Object common to all worker threads + @param address: the address to bind this IOServer to + @param rqhandler: RequestHandler type object """ SocketServer.UnixStreamServer.__init__(self, address, rqhandler) - self.context = context # We'll only start threads once we've forked. - self.jobqueue = None + self.context = None self.request_workers = None def setup_queue(self): - self.jobqueue = jqueue.JobQueue(self.context) + self.context = GanetiContext() self.request_workers = workerpool.WorkerPool(CLIENT_REQUEST_WORKERS, ClientRequestWorker) @@ -110,14 +110,17 @@ class IOServer(SocketServer.UnixStreamServer): """ self.request_workers.AddTask(self, request, client_address) - def serve_forever(self): + @utils.SignalHandled([signal.SIGINT, signal.SIGTERM]) + def serve_forever(self, signal_handlers=None): # pylint: disable-msg=W0221 """Handle one request at a time until told to quit.""" - sighandler = utils.SignalHandler([signal.SIGINT, signal.SIGTERM]) - try: - while not sighandler.called: - self.handle_request() - finally: - sighandler.Reset() + assert isinstance(signal_handlers, dict) and \ + len(signal_handlers) > 0, \ + "Broken SignalHandled decorator" + # Since we use SignalHandled only once, the resulting dict will map all + # signals to the same handler. We'll just use the first one. + sighandler = signal_handlers.values()[0] + while not sighandler.called: + self.handle_request() def server_cleanup(self): """Cleanup the server. @@ -128,12 +131,11 @@ class IOServer(SocketServer.UnixStreamServer): """ try: self.server_close() - utils.RemoveFile(constants.MASTER_SOCKET) finally: if self.request_workers: self.request_workers.TerminateWorkers() - if self.jobqueue: - self.jobqueue.Shutdown() + if self.context: + self.context.jobqueue.Shutdown() class ClientRqHandler(SocketServer.BaseRequestHandler): @@ -142,6 +144,8 @@ class ClientRqHandler(SocketServer.BaseRequestHandler): READ_SIZE = 4096 def setup(self): + # pylint: disable-msg=W0201 + # setup() is the api for initialising for this class self._buffer = "" self._msgs = collections.deque() self._ops = ClientOps(self.server) @@ -150,10 +154,10 @@ class ClientRqHandler(SocketServer.BaseRequestHandler): while True: msg = self.read_message() if msg is None: - logging.info("client closed connection") + logging.debug("client closed connection") break - request = simplejson.loads(msg) + request = serializer.LoadJson(msg) logging.debug("request: %s", request) if not isinstance(request, dict): logging.error("wrong request received: %s", msg) @@ -169,6 +173,9 @@ class ClientRqHandler(SocketServer.BaseRequestHandler): try: result = self._ops.handle_request(method, args) success = True + except errors.GenericError, err: + success = False + result = errors.EncodeException(err) except: logging.error("Unexpected exception", exc_info=True) err = sys.exc_info() @@ -179,7 +186,7 @@ class ClientRqHandler(SocketServer.BaseRequestHandler): luxi.KEY_RESULT: result, } logging.debug("response: %s", response) - self.send_message(simplejson.dumps(response)) + self.send_message(serializer.DumpJson(response)) def read_message(self): while not self._msgs: @@ -193,6 +200,7 @@ class ClientRqHandler(SocketServer.BaseRequestHandler): def send_message(self, msg): #print "sending", msg + # TODO: sendall is not guaranteed to send everything self.request.sendall(msg + self.EOM) @@ -201,33 +209,129 @@ class ClientOps: def __init__(self, server): self.server = server - def handle_request(self, method, args): - queue = self.server.jobqueue + def handle_request(self, method, args): # pylint: disable-msg=R0911 + queue = self.server.context.jobqueue # TODO: Parameter validation + # TODO: Rewrite to not exit in each 'if/elif' branch + if method == luxi.REQ_SUBMIT_JOB: + logging.info("Received new job") ops = [opcodes.OpCode.LoadOpCode(state) for state in args] - # we need to compute the node list here, since from now on all - # operations require locks on the queue or the storage, and we - # shouldn't get another lock - node_list = self.server.context.cfg.GetNodeList() - return queue.SubmitJob(ops, node_list) + return queue.SubmitJob(ops) + + if method == luxi.REQ_SUBMIT_MANY_JOBS: + logging.info("Received multiple jobs") + jobs = [] + for ops in args: + jobs.append([opcodes.OpCode.LoadOpCode(state) for state in ops]) + return queue.SubmitManyJobs(jobs) elif method == luxi.REQ_CANCEL_JOB: job_id = args + logging.info("Received job cancel request for %s", job_id) return queue.CancelJob(job_id) elif method == luxi.REQ_ARCHIVE_JOB: job_id = args + logging.info("Received job archive request for %s", job_id) return queue.ArchiveJob(job_id) + elif method == luxi.REQ_AUTOARCHIVE_JOBS: + (age, timeout) = args + logging.info("Received job autoarchive request for age %s, timeout %s", + age, timeout) + return queue.AutoArchiveJobs(age, timeout) + + elif method == luxi.REQ_WAIT_FOR_JOB_CHANGE: + (job_id, fields, prev_job_info, prev_log_serial, timeout) = args + logging.info("Received job poll request for %s", job_id) + return queue.WaitForJobChanges(job_id, fields, prev_job_info, + prev_log_serial, timeout) + elif method == luxi.REQ_QUERY_JOBS: (job_ids, fields) = args + if isinstance(job_ids, (tuple, list)) and job_ids: + msg = utils.CommaJoin(job_ids) + else: + msg = str(job_ids) + logging.info("Received job query request for %s", msg) return queue.QueryJobs(job_ids, fields) + elif method == luxi.REQ_QUERY_INSTANCES: + (names, fields, use_locking) = args + logging.info("Received instance query request for %s", names) + if use_locking: + raise errors.OpPrereqError("Sync queries are not allowed", + errors.ECODE_INVAL) + op = opcodes.OpQueryInstances(names=names, output_fields=fields, + use_locking=use_locking) + return self._Query(op) + + elif method == luxi.REQ_QUERY_NODES: + (names, fields, use_locking) = args + logging.info("Received node query request for %s", names) + if use_locking: + raise errors.OpPrereqError("Sync queries are not allowed", + errors.ECODE_INVAL) + op = opcodes.OpQueryNodes(names=names, output_fields=fields, + use_locking=use_locking) + return self._Query(op) + + elif method == luxi.REQ_QUERY_EXPORTS: + nodes, use_locking = args + if use_locking: + raise errors.OpPrereqError("Sync queries are not allowed", + errors.ECODE_INVAL) + logging.info("Received exports query request") + op = opcodes.OpQueryExports(nodes=nodes, use_locking=use_locking) + return self._Query(op) + + elif method == luxi.REQ_QUERY_CONFIG_VALUES: + fields = args + logging.info("Received config values query request for %s", fields) + op = opcodes.OpQueryConfigValues(output_fields=fields) + return self._Query(op) + + elif method == luxi.REQ_QUERY_CLUSTER_INFO: + logging.info("Received cluster info query request") + op = opcodes.OpQueryClusterInfo() + return self._Query(op) + + elif method == luxi.REQ_QUEUE_SET_DRAIN_FLAG: + drain_flag = args + logging.info("Received queue drain flag change request to %s", + drain_flag) + return queue.SetDrainFlag(drain_flag) + + elif method == luxi.REQ_SET_WATCHER_PAUSE: + (until, ) = args + + if until is None: + logging.info("Received request to no longer pause the watcher") + else: + if not isinstance(until, (int, float)): + raise TypeError("Duration must be an integer or float") + + if until < time.time(): + raise errors.GenericError("Unable to set pause end time in the past") + + logging.info("Received request to pause the watcher until %s", until) + + return _SetWatcherPause(until) + else: - raise ValueError("Invalid operation") + logging.info("Received invalid request '%s'", method) + raise ValueError("Invalid operation '%s'" % method) + + def _Query(self, op): + """Runs the specified opcode and returns the result. + + """ + # Queries don't have a job id + proc = mcpu.Processor(self.server.context, None) + return proc.ExecOpCode(op, None) class GanetiContext(object): @@ -236,6 +340,8 @@ class GanetiContext(object): This class creates and holds common objects shared by all threads. """ + # pylint: disable-msg=W0212 + # we do want to ensure a singleton here _instance = None def __init__(self): @@ -247,13 +353,17 @@ class GanetiContext(object): """ assert self.__class__._instance is None, "double GanetiContext instance" - # Create a ConfigWriter... + # Create global configuration object self.cfg = config.ConfigWriter() - # And a GanetiLockingManager... + + # Locking manager self.glm = locking.GanetiLockManager( self.cfg.GetNodeList(), self.cfg.GetInstanceList()) + # Job queue + self.jobqueue = jqueue.JobQueue(self) + # setting this also locks the class against attribute modifications self.__class__._instance = self @@ -264,84 +374,248 @@ class GanetiContext(object): assert self.__class__._instance is None, "Attempt to modify Ganeti Context" object.__setattr__(self, name, value) + def AddNode(self, node, ec_id): + """Adds a node to the configuration and lock manager. + + """ + # Add it to the configuration + self.cfg.AddNode(node, ec_id) + + # If preseeding fails it'll not be added + self.jobqueue.AddNode(node) -def CheckMaster(debug): - """Checks the node setup. + # Add the new node to the Ganeti Lock Manager + self.glm.add(locking.LEVEL_NODE, node.name) - If this is the master, the function will return. Otherwise it will - exit with an exit code based on the node status. + def ReaddNode(self, node): + """Updates a node that's already in the configuration + + """ + # Synchronize the queue again + self.jobqueue.AddNode(node) + + def RemoveNode(self, name): + """Removes a node from the configuration and lock manager. + + """ + # Remove node from configuration + self.cfg.RemoveNode(name) + + # Notify job queue + self.jobqueue.RemoveNode(name) + + # Remove the node from the Ganeti Lock Manager + self.glm.remove(locking.LEVEL_NODE, name) + + +def _SetWatcherPause(until): + """Creates or removes the watcher pause file. + + @type until: None or int + @param until: Unix timestamp saying until when the watcher shouldn't run """ - try: - ss = ssconf.SimpleStore() - master_name = ss.GetMasterNode() - except errors.ConfigurationError, err: - print "Cluster configuration incomplete: '%s'" % str(err) - sys.exit(EXIT_NODESETUP_ERROR) + if until is None: + utils.RemoveFile(constants.WATCHER_PAUSEFILE) + else: + utils.WriteFile(constants.WATCHER_PAUSEFILE, + data="%d\n" % (until, )) + + return until + + +def CheckAgreement(): + """Check the agreement on who is the master. + The function uses a very simple algorithm: we must get more positive + than negative answers. Since in most of the cases we are the master, + we'll use our own config file for getting the node list. In the + future we could collect the current node list from our (possibly + obsolete) known nodes. + + In order to account for cold-start of all nodes, we retry for up to + a minute until we get a real answer as the top-voted one. If the + nodes are more out-of-sync, for now manual startup of the master + should be attempted. + + Note that for a even number of nodes cluster, we need at least half + of the nodes (beside ourselves) to vote for us. This creates a + problem on two-node clusters, since in this case we require the + other node to be up too to confirm our status. + + """ + myself = utils.HostInfo().name + #temp instantiation of a config writer, used only to get the node list + cfg = config.ConfigWriter() + node_list = cfg.GetNodeList() + del cfg + retries = 6 + while retries > 0: + votes = bootstrap.GatherMasterVotes(node_list) + if not votes: + # empty node list, this is a one node cluster + return True + if votes[0][0] is None: + retries -= 1 + time.sleep(10) + continue + break + if retries == 0: + logging.critical("Cluster inconsistent, most of the nodes didn't answer" + " after multiple retries. Aborting startup") + return False + # here a real node is at the top of the list + all_votes = sum(item[1] for item in votes) + top_node, top_votes = votes[0] + + result = False + if top_node != myself: + logging.critical("It seems we are not the master (top-voted node" + " is %s with %d out of %d votes)", top_node, top_votes, + all_votes) + elif top_votes < all_votes - top_votes: + logging.critical("It seems we are not the master (%d votes for," + " %d votes against)", top_votes, all_votes - top_votes) + else: + result = True + + return result + + +def CheckAgreementWithRpc(): + rpc.Init() try: - myself = utils.HostInfo() - except errors.ResolverError, err: - sys.stderr.write("Cannot resolve my own name (%s)\n" % err.args[0]) - sys.exit(EXIT_NODESETUP_ERROR) + return CheckAgreement() + finally: + rpc.Shutdown() - if myself.name != master_name: - if debug: - sys.stderr.write("Not master, exiting.\n") - sys.exit(EXIT_NOTMASTER) +def _RunInSeparateProcess(fn): + """Runs a function in a separate process. -def ParseOptions(): - """Parse the command line options. + Note: Only boolean return values are supported. - Returns: - (options, args) as from OptionParser.parse_args() + @type fn: callable + @param fn: Function to be called + @rtype: bool """ - parser = OptionParser(description="Ganeti master daemon", - usage="%prog [-f] [-d]", - version="%%prog (ganeti) %s" % - constants.RELEASE_VERSION) + pid = os.fork() + if pid == 0: + # Child process + try: + # Call function + result = int(bool(fn())) + assert result in (0, 1) + except: # pylint: disable-msg=W0702 + logging.exception("Error while calling function in separate process") + # 0 and 1 are reserved for the return value + result = 33 - parser.add_option("-f", "--foreground", dest="fork", - help="Don't detach from the current terminal", - default=True, action="store_false") - parser.add_option("-d", "--debug", dest="debug", - help="Enable some debug messages", - default=False, action="store_true") - options, args = parser.parse_args() - return options, args + os._exit(result) # pylint: disable-msg=W0212 + # Parent process -def main(): - """Main function""" + # Avoid zombies and check exit code + (_, status) = os.waitpid(pid, 0) + + if os.WIFSIGNALED(status): + signum = os.WTERMSIG(status) + exitcode = None + else: + signum = None + exitcode = os.WEXITSTATUS(status) + + if not (exitcode in (0, 1) and signum is None): + logging.error("Child program failed (code=%s, signal=%s)", + exitcode, signum) + sys.exit(constants.EXIT_FAILURE) + + return bool(exitcode) + + +def CheckMasterd(options, args): + """Initial checks whether to run or exit with a failure. + + """ + if args: # masterd doesn't take any arguments + print >> sys.stderr, ("Usage: %s [-f] [-d]" % sys.argv[0]) + sys.exit(constants.EXIT_FAILURE) - options, args = ParseOptions() - utils.debug = options.debug - utils.no_fork = True + ssconf.CheckMaster(options.debug) - CheckMaster(options.debug) + # If CheckMaster didn't fail we believe we are the master, but we have to + # confirm with the other nodes. + if options.no_voting: + if options.yes_do_it: + return - master = IOServer(constants.MASTER_SOCKET, ClientRqHandler, GanetiContext()) + sys.stdout.write("The 'no voting' option has been selected.\n") + sys.stdout.write("This is dangerous, please confirm by" + " typing uppercase 'yes': ") + sys.stdout.flush() - # become a daemon - if options.fork: - utils.Daemonize(logfile=constants.LOG_MASTERDAEMON, - noclose_fds=[master.fileno()]) + confirmation = sys.stdin.readline().strip() + if confirmation != "YES": + print >> sys.stderr, "Aborting." + sys.exit(constants.EXIT_FAILURE) - utils.WritePidFile('ganeti-masterd') + return - logger.SetupDaemon(constants.LOG_MASTERDAEMON, debug=options.debug, - stderr_logging=not options.fork) + # CheckAgreement uses RPC and threads, hence it needs to be run in a separate + # process before we call utils.Daemonize in the current process. + if not _RunInSeparateProcess(CheckAgreementWithRpc): + sys.exit(constants.EXIT_FAILURE) - logging.info("ganeti master daemon startup") - master.setup_queue() +def ExecMasterd (options, args): # pylint: disable-msg=W0613 + """Main master daemon function, executed with the PID file held. + + """ + # This is safe to do as the pid file guarantees against + # concurrent execution. + utils.RemoveFile(constants.MASTER_SOCKET) + + master = IOServer(constants.MASTER_SOCKET, ClientRqHandler) try: - master.serve_forever() + rpc.Init() + try: + # activate ip + master_node = ssconf.SimpleStore().GetMasterNode() + result = rpc.RpcRunner.call_node_start_master(master_node, False, False) + msg = result.fail_msg + if msg: + logging.error("Can't activate master IP address: %s", msg) + + master.setup_queue() + try: + master.serve_forever() + finally: + master.server_cleanup() + finally: + rpc.Shutdown() finally: - master.server_cleanup() - utils.RemovePidFile('ganeti-masterd') + utils.RemoveFile(constants.MASTER_SOCKET) + + +def main(): + """Main function""" + parser = OptionParser(description="Ganeti master daemon", + usage="%prog [-f] [-d]", + version="%%prog (ganeti) %s" % + constants.RELEASE_VERSION) + parser.add_option("--no-voting", dest="no_voting", + help="Do not check that the nodes agree on this node" + " being the master and start the daemon unconditionally", + default=False, action="store_true") + parser.add_option("--yes-do-it", dest="yes_do_it", + help="Override interactive check for --no-voting", + default=False, action="store_true") + dirs = [(constants.RUN_GANETI_DIR, constants.RUN_DIRS_MODE), + (constants.SOCKET_DIR, constants.SOCKET_DIR_MODE), + ] + daemon.GenericMain(constants.MASTERD, parser, dirs, + CheckMasterd, ExecMasterd) if __name__ == "__main__":