| #!/usr/bin/python |
| #pylint: disable-msg=C0111 |
| |
| """ |
| Autotest scheduler |
| """ |
| |
| import datetime, optparse, os, signal |
| import sys, time |
| import logging, gc |
| |
| import common |
| from autotest_lib.frontend import setup_django_environment |
| |
| import django.db |
| |
| from autotest_lib.client.common_lib import global_config |
| from autotest_lib.client.common_lib import utils |
| from autotest_lib.client.common_lib.cros.graphite import stats |
| from autotest_lib.frontend.afe import models, rpc_utils |
| from autotest_lib.scheduler import agent_task, drone_manager, drones |
| from autotest_lib.scheduler import email_manager, gc_stats, host_scheduler |
| from autotest_lib.scheduler import monitor_db_cleanup, prejob_task |
| from autotest_lib.scheduler import postjob_task |
| from autotest_lib.scheduler import query_managers |
| from autotest_lib.scheduler import scheduler_lib |
| from autotest_lib.scheduler import scheduler_models |
| from autotest_lib.scheduler import status_server, scheduler_config |
| from autotest_lib.scheduler import scheduler_lib |
| from autotest_lib.server import autoserv_utils |
| |
| BABYSITTER_PID_FILE_PREFIX = 'monitor_db_babysitter' |
| PID_FILE_PREFIX = 'monitor_db' |
| |
| RESULTS_DIR = '.' |
| AUTOTEST_PATH = os.path.join(os.path.dirname(__file__), '..') |
| |
| if os.environ.has_key('AUTOTEST_DIR'): |
| AUTOTEST_PATH = os.environ['AUTOTEST_DIR'] |
| AUTOTEST_SERVER_DIR = os.path.join(AUTOTEST_PATH, 'server') |
| AUTOTEST_TKO_DIR = os.path.join(AUTOTEST_PATH, 'tko') |
| |
| if AUTOTEST_SERVER_DIR not in sys.path: |
| sys.path.insert(0, AUTOTEST_SERVER_DIR) |
| |
| # error message to leave in results dir when an autoserv process disappears |
| # mysteriously |
| _LOST_PROCESS_ERROR = """\ |
| Autoserv failed abnormally during execution for this job, probably due to a |
| system error on the Autotest server. Full results may not be available. Sorry. |
| """ |
| |
| _db_manager = None |
| _shutdown = False |
| |
| # These 2 globals are replaced for testing |
| _autoserv_directory = autoserv_utils.autoserv_directory |
| _autoserv_path = autoserv_utils.autoserv_path |
| _testing_mode = False |
| _drone_manager = None |
| _inline_host_acquisition = global_config.global_config.get_config_value( |
| scheduler_config.CONFIG_SECTION, 'inline_host_acquisition', type=bool, |
| default=True) |
| |
| |
| def _parser_path_default(install_dir): |
| return os.path.join(install_dir, 'tko', 'parse') |
| _parser_path_func = utils.import_site_function( |
| __file__, 'autotest_lib.scheduler.site_monitor_db', |
| 'parser_path', _parser_path_default) |
| _parser_path = _parser_path_func(drones.AUTOTEST_INSTALL_DIR) |
| |
| |
| def _site_init_monitor_db_dummy(): |
| return {} |
| |
| |
| def _verify_default_drone_set_exists(): |
| if (models.DroneSet.drone_sets_enabled() and |
| not models.DroneSet.default_drone_set_name()): |
| raise scheduler_lib.SchedulerError( |
| 'Drone sets are enabled, but no default is set') |
| |
| |
| def _sanity_check(): |
| """Make sure the configs are consistent before starting the scheduler""" |
| _verify_default_drone_set_exists() |
| |
| |
| def main(): |
| try: |
| try: |
| main_without_exception_handling() |
| except SystemExit: |
| raise |
| except: |
| logging.exception('Exception escaping in monitor_db') |
| raise |
| finally: |
| utils.delete_pid_file_if_exists(PID_FILE_PREFIX) |
| |
| |
| def main_without_exception_handling(): |
| scheduler_lib.setup_logging( |
| os.environ.get('AUTOTEST_SCHEDULER_LOG_DIR', None), |
| os.environ.get('AUTOTEST_SCHEDULER_LOG_NAME', None)) |
| usage = 'usage: %prog [options] results_dir' |
| parser = optparse.OptionParser(usage) |
| parser.add_option('--recover-hosts', help='Try to recover dead hosts', |
| action='store_true') |
| parser.add_option('--test', help='Indicate that scheduler is under ' + |
| 'test and should use dummy autoserv and no parsing', |
| action='store_true') |
| (options, args) = parser.parse_args() |
| if len(args) != 1: |
| parser.print_usage() |
| return |
| |
| scheduler_enabled = global_config.global_config.get_config_value( |
| scheduler_config.CONFIG_SECTION, 'enable_scheduler', type=bool) |
| |
| if not scheduler_enabled: |
| logging.error("Scheduler not enabled, set enable_scheduler to true in " |
| "the global_config's SCHEDULER section to enable it. " |
| "Exiting.") |
| sys.exit(1) |
| |
| global RESULTS_DIR |
| RESULTS_DIR = args[0] |
| |
| site_init = utils.import_site_function(__file__, |
| "autotest_lib.scheduler.site_monitor_db", "site_init_monitor_db", |
| _site_init_monitor_db_dummy) |
| site_init() |
| |
| # Change the cwd while running to avoid issues incase we were launched from |
| # somewhere odd (such as a random NFS home directory of the person running |
| # sudo to launch us as the appropriate user). |
| os.chdir(RESULTS_DIR) |
| |
| # This is helpful for debugging why stuff a scheduler launches is |
| # misbehaving. |
| logging.info('os.environ: %s', os.environ) |
| |
| if options.test: |
| global _autoserv_path |
| _autoserv_path = 'autoserv_dummy' |
| global _testing_mode |
| _testing_mode = True |
| |
| server = status_server.StatusServer() |
| server.start() |
| |
| try: |
| initialize() |
| dispatcher = Dispatcher() |
| dispatcher.initialize(recover_hosts=options.recover_hosts) |
| |
| while not _shutdown and not server._shutdown_scheduler: |
| dispatcher.tick() |
| time.sleep(scheduler_config.config.tick_pause_sec) |
| except Exception: |
| email_manager.manager.log_stacktrace( |
| "Uncaught exception; terminating monitor_db") |
| |
| email_manager.manager.send_queued_emails() |
| server.shutdown() |
| _drone_manager.shutdown() |
| _db_manager.disconnect() |
| |
| |
| def handle_signal(signum, frame): |
| global _shutdown |
| _shutdown = True |
| logging.info("Shutdown request received.") |
| |
| |
| def initialize(): |
| logging.info("%s> dispatcher starting", time.strftime("%X %x")) |
| logging.info("My PID is %d", os.getpid()) |
| |
| if utils.program_is_alive(PID_FILE_PREFIX): |
| logging.critical("monitor_db already running, aborting!") |
| sys.exit(1) |
| utils.write_pid(PID_FILE_PREFIX) |
| |
| if _testing_mode: |
| global_config.global_config.override_config_value( |
| scheduler_lib.DB_CONFIG_SECTION, 'database', |
| 'stresstest_autotest_web') |
| |
| os.environ['PATH'] = AUTOTEST_SERVER_DIR + ':' + os.environ['PATH'] |
| global _db_manager |
| _db_manager = scheduler_lib.ConnectionManager() |
| logging.info("Setting signal handler") |
| signal.signal(signal.SIGINT, handle_signal) |
| signal.signal(signal.SIGTERM, handle_signal) |
| |
| initialize_globals() |
| scheduler_models.initialize() |
| |
| drones = global_config.global_config.get_config_value( |
| scheduler_config.CONFIG_SECTION, 'drones', default='localhost') |
| drone_list = [hostname.strip() for hostname in drones.split(',')] |
| results_host = global_config.global_config.get_config_value( |
| scheduler_config.CONFIG_SECTION, 'results_host', default='localhost') |
| _drone_manager.initialize(RESULTS_DIR, drone_list, results_host) |
| |
| logging.info("Connected! Running...") |
| |
| |
| def initialize_globals(): |
| global _drone_manager |
| _drone_manager = drone_manager.instance() |
| |
| |
| def _autoserv_command_line(machines, extra_args, job=None, queue_entry=None, |
| verbose=True): |
| """ |
| @returns The autoserv command line as a list of executable + parameters. |
| |
| @param machines - string - A machine or comma separated list of machines |
| for the (-m) flag. |
| @param extra_args - list - Additional arguments to pass to autoserv. |
| @param job - Job object - If supplied, -u owner, -l name, --test-retry, |
| and client -c or server -s parameters will be added. |
| @param queue_entry - A HostQueueEntry object - If supplied and no Job |
| object was supplied, this will be used to lookup the Job object. |
| """ |
| return autoserv_utils.autoserv_run_job_command(_autoserv_directory, |
| machines, results_directory=drone_manager.WORKING_DIRECTORY, |
| extra_args=extra_args, job=job, queue_entry=queue_entry, |
| verbose=verbose) |
| |
| |
| class BaseDispatcher(object): |
| |
| |
| def __init__(self): |
| self._agents = [] |
| self._last_clean_time = time.time() |
| user_cleanup_time = scheduler_config.config.clean_interval |
| self._periodic_cleanup = monitor_db_cleanup.UserCleanup( |
| _db_manager.get_connection(), user_cleanup_time) |
| self._24hr_upkeep = monitor_db_cleanup.TwentyFourHourUpkeep( |
| _db_manager.get_connection()) |
| self._host_agents = {} |
| self._queue_entry_agents = {} |
| self._tick_count = 0 |
| self._last_garbage_stats_time = time.time() |
| self._seconds_between_garbage_stats = 60 * ( |
| global_config.global_config.get_config_value( |
| scheduler_config.CONFIG_SECTION, |
| 'gc_stats_interval_mins', type=int, default=6*60)) |
| self._tick_debug = global_config.global_config.get_config_value( |
| scheduler_config.CONFIG_SECTION, 'tick_debug', type=bool, |
| default=False) |
| self._extra_debugging = global_config.global_config.get_config_value( |
| scheduler_config.CONFIG_SECTION, 'extra_debugging', type=bool, |
| default=False) |
| |
| # If _inline_host_acquisition is set the scheduler will acquire and |
| # release hosts against jobs inline, with the tick. Otherwise the |
| # scheduler will only focus on jobs that already have hosts, and |
| # will not explicitly unlease a host when a job finishes using it. |
| self._job_query_manager = query_managers.AFEJobQueryManager() |
| self._host_scheduler = (host_scheduler.BaseHostScheduler() |
| if _inline_host_acquisition else |
| host_scheduler.DummyHostScheduler()) |
| |
| |
| def initialize(self, recover_hosts=True): |
| self._periodic_cleanup.initialize() |
| self._24hr_upkeep.initialize() |
| |
| # always recover processes |
| self._recover_processes() |
| |
| if recover_hosts: |
| self._recover_hosts() |
| |
| |
| def _log_tick_msg(self, msg): |
| if self._tick_debug: |
| logging.debug(msg) |
| |
| |
| def _log_extra_msg(self, msg): |
| if self._extra_debugging: |
| logging.debug(msg) |
| |
| |
| def tick(self): |
| """ |
| This is an altered version of tick() where we keep track of when each |
| major step begins so we can try to figure out where we are using most |
| of the tick time. |
| """ |
| timer = stats.Timer('scheduler.tick') |
| self._log_tick_msg('Calling new tick, starting garbage collection().') |
| self._garbage_collection() |
| self._log_tick_msg('Calling _drone_manager.refresh().') |
| _drone_manager.refresh() |
| self._log_tick_msg('Calling _run_cleanup().') |
| self._run_cleanup() |
| self._log_tick_msg('Calling _find_aborting().') |
| self._find_aborting() |
| self._log_tick_msg('Calling _find_aborted_special_tasks().') |
| self._find_aborted_special_tasks() |
| self._log_tick_msg('Calling _process_recurring_runs().') |
| self._process_recurring_runs() |
| self._log_tick_msg('Calling _schedule_delay_tasks().') |
| self._schedule_delay_tasks() |
| self._log_tick_msg('Calling _schedule_running_host_queue_entries().') |
| self._schedule_running_host_queue_entries() |
| self._log_tick_msg('Calling _schedule_special_tasks().') |
| self._schedule_special_tasks() |
| self._log_tick_msg('Calling _schedule_new_jobs().') |
| self._schedule_new_jobs() |
| self._log_tick_msg('Calling _handle_agents().') |
| self._handle_agents() |
| self._log_tick_msg('Calling _host_scheduler.tick().') |
| self._host_scheduler.tick() |
| self._log_tick_msg('Calling _drone_manager.execute_actions().') |
| _drone_manager.execute_actions() |
| self._log_tick_msg('Calling ' |
| 'email_manager.manager.send_queued_emails().') |
| with timer.get_client('email_manager_send_queued_emails'): |
| email_manager.manager.send_queued_emails() |
| self._log_tick_msg('Calling django.db.reset_queries().') |
| with timer.get_client('django_db_reset_queries'): |
| django.db.reset_queries() |
| self._tick_count += 1 |
| |
| |
| def _run_cleanup(self): |
| self._periodic_cleanup.run_cleanup_maybe() |
| self._24hr_upkeep.run_cleanup_maybe() |
| |
| |
| def _garbage_collection(self): |
| threshold_time = time.time() - self._seconds_between_garbage_stats |
| if threshold_time < self._last_garbage_stats_time: |
| # Don't generate these reports very often. |
| return |
| |
| self._last_garbage_stats_time = time.time() |
| # Force a full level 0 collection (because we can, it doesn't hurt |
| # at this interval). |
| gc.collect() |
| logging.info('Logging garbage collector stats on tick %d.', |
| self._tick_count) |
| gc_stats._log_garbage_collector_stats() |
| |
| |
| def _register_agent_for_ids(self, agent_dict, object_ids, agent): |
| for object_id in object_ids: |
| agent_dict.setdefault(object_id, set()).add(agent) |
| |
| |
| def _unregister_agent_for_ids(self, agent_dict, object_ids, agent): |
| for object_id in object_ids: |
| assert object_id in agent_dict |
| agent_dict[object_id].remove(agent) |
| # If an ID has no more active agent associated, there is no need to |
| # keep it in the dictionary. Otherwise, scheduler will keep an |
| # unnecessarily big dictionary until being restarted. |
| if not agent_dict[object_id]: |
| agent_dict.pop(object_id) |
| |
| |
| def add_agent_task(self, agent_task): |
| """ |
| Creates and adds an agent to the dispatchers list. |
| |
| In creating the agent we also pass on all the queue_entry_ids and |
| host_ids from the special agent task. For every agent we create, we |
| add it to 1. a dict against the queue_entry_ids given to it 2. A dict |
| against the host_ids given to it. So theoritically, a host can have any |
| number of agents associated with it, and each of them can have any |
| special agent task, though in practice we never see > 1 agent/task per |
| host at any time. |
| |
| @param agent_task: A SpecialTask for the agent to manage. |
| """ |
| agent = Agent(agent_task) |
| self._agents.append(agent) |
| agent.dispatcher = self |
| self._register_agent_for_ids(self._host_agents, agent.host_ids, agent) |
| self._register_agent_for_ids(self._queue_entry_agents, |
| agent.queue_entry_ids, agent) |
| |
| |
| def get_agents_for_entry(self, queue_entry): |
| """ |
| Find agents corresponding to the specified queue_entry. |
| """ |
| return list(self._queue_entry_agents.get(queue_entry.id, set())) |
| |
| |
| def host_has_agent(self, host): |
| """ |
| Determine if there is currently an Agent present using this host. |
| """ |
| return bool(self._host_agents.get(host.id, None)) |
| |
| |
| def remove_agent(self, agent): |
| self._agents.remove(agent) |
| self._unregister_agent_for_ids(self._host_agents, agent.host_ids, |
| agent) |
| self._unregister_agent_for_ids(self._queue_entry_agents, |
| agent.queue_entry_ids, agent) |
| |
| |
| def _host_has_scheduled_special_task(self, host): |
| return bool(models.SpecialTask.objects.filter(host__id=host.id, |
| is_active=False, |
| is_complete=False)) |
| |
| |
| def _recover_processes(self): |
| agent_tasks = self._create_recovery_agent_tasks() |
| self._register_pidfiles(agent_tasks) |
| _drone_manager.refresh() |
| self._recover_tasks(agent_tasks) |
| self._recover_pending_entries() |
| self._check_for_unrecovered_verifying_entries() |
| self._reverify_remaining_hosts() |
| # reinitialize drones after killing orphaned processes, since they can |
| # leave around files when they die |
| _drone_manager.execute_actions() |
| _drone_manager.reinitialize_drones() |
| |
| |
| def _create_recovery_agent_tasks(self): |
| return (self._get_queue_entry_agent_tasks() |
| + self._get_special_task_agent_tasks(is_active=True)) |
| |
| |
| def _get_queue_entry_agent_tasks(self): |
| """ |
| Get agent tasks for all hqe in the specified states. |
| |
| Loosely this translates to taking a hqe in one of the specified states, |
| say parsing, and getting an AgentTask for it, like the FinalReparseTask, |
| through _get_agent_task_for_queue_entry. Each queue entry can only have |
| one agent task at a time, but there might be multiple queue entries in |
| the group. |
| |
| @return: A list of AgentTasks. |
| """ |
| # host queue entry statuses handled directly by AgentTasks (Verifying is |
| # handled through SpecialTasks, so is not listed here) |
| statuses = (models.HostQueueEntry.Status.STARTING, |
| models.HostQueueEntry.Status.RUNNING, |
| models.HostQueueEntry.Status.GATHERING, |
| models.HostQueueEntry.Status.PARSING, |
| models.HostQueueEntry.Status.ARCHIVING) |
| status_list = ','.join("'%s'" % status for status in statuses) |
| queue_entries = scheduler_models.HostQueueEntry.fetch( |
| where='status IN (%s)' % status_list) |
| stats.Gauge('scheduler.jobs_per_tick').send( |
| 'running', len(queue_entries)) |
| |
| agent_tasks = [] |
| used_queue_entries = set() |
| for entry in queue_entries: |
| if self.get_agents_for_entry(entry): |
| # already being handled |
| continue |
| if entry in used_queue_entries: |
| # already picked up by a synchronous job |
| continue |
| agent_task = self._get_agent_task_for_queue_entry(entry) |
| agent_tasks.append(agent_task) |
| used_queue_entries.update(agent_task.queue_entries) |
| return agent_tasks |
| |
| |
| def _get_special_task_agent_tasks(self, is_active=False): |
| special_tasks = models.SpecialTask.objects.filter( |
| is_active=is_active, is_complete=False) |
| return [self._get_agent_task_for_special_task(task) |
| for task in special_tasks] |
| |
| |
| def _get_agent_task_for_queue_entry(self, queue_entry): |
| """ |
| Construct an AgentTask instance for the given active HostQueueEntry. |
| |
| @param queue_entry: a HostQueueEntry |
| @return: an AgentTask to run the queue entry |
| """ |
| task_entries = queue_entry.job.get_group_entries(queue_entry) |
| self._check_for_duplicate_host_entries(task_entries) |
| |
| if queue_entry.status in (models.HostQueueEntry.Status.STARTING, |
| models.HostQueueEntry.Status.RUNNING): |
| if queue_entry.is_hostless(): |
| return HostlessQueueTask(queue_entry=queue_entry) |
| return QueueTask(queue_entries=task_entries) |
| if queue_entry.status == models.HostQueueEntry.Status.GATHERING: |
| return postjob_task.GatherLogsTask(queue_entries=task_entries) |
| if queue_entry.status == models.HostQueueEntry.Status.PARSING: |
| return postjob_task.FinalReparseTask(queue_entries=task_entries) |
| if queue_entry.status == models.HostQueueEntry.Status.ARCHIVING: |
| return postjob_task.ArchiveResultsTask(queue_entries=task_entries) |
| |
| raise scheduler_lib.SchedulerError( |
| '_get_agent_task_for_queue_entry got entry with ' |
| 'invalid status %s: %s' % (queue_entry.status, queue_entry)) |
| |
| |
| def _check_for_duplicate_host_entries(self, task_entries): |
| non_host_statuses = (models.HostQueueEntry.Status.PARSING, |
| models.HostQueueEntry.Status.ARCHIVING) |
| for task_entry in task_entries: |
| using_host = (task_entry.host is not None |
| and task_entry.status not in non_host_statuses) |
| if using_host: |
| self._assert_host_has_no_agent(task_entry) |
| |
| |
| def _assert_host_has_no_agent(self, entry): |
| """ |
| @param entry: a HostQueueEntry or a SpecialTask |
| """ |
| if self.host_has_agent(entry.host): |
| agent = tuple(self._host_agents.get(entry.host.id))[0] |
| raise scheduler_lib.SchedulerError( |
| 'While scheduling %s, host %s already has a host agent %s' |
| % (entry, entry.host, agent.task)) |
| |
| |
| def _get_agent_task_for_special_task(self, special_task): |
| """ |
| Construct an AgentTask class to run the given SpecialTask and add it |
| to this dispatcher. |
| |
| A special task is create through schedule_special_tasks, but only if |
| the host doesn't already have an agent. This happens through |
| add_agent_task. All special agent tasks are given a host on creation, |
| and a Null hqe. To create a SpecialAgentTask object, you need a |
| models.SpecialTask. If the SpecialTask used to create a SpecialAgentTask |
| object contains a hqe it's passed on to the special agent task, which |
| creates a HostQueueEntry and saves it as it's queue_entry. |
| |
| @param special_task: a models.SpecialTask instance |
| @returns an AgentTask to run this SpecialTask |
| """ |
| self._assert_host_has_no_agent(special_task) |
| |
| special_agent_task_classes = (prejob_task.CleanupTask, |
| prejob_task.VerifyTask, |
| prejob_task.RepairTask, |
| prejob_task.ResetTask, |
| prejob_task.ProvisionTask) |
| |
| for agent_task_class in special_agent_task_classes: |
| if agent_task_class.TASK_TYPE == special_task.task: |
| return agent_task_class(task=special_task) |
| |
| raise scheduler_lib.SchedulerError( |
| 'No AgentTask class for task', str(special_task)) |
| |
| |
| def _register_pidfiles(self, agent_tasks): |
| for agent_task in agent_tasks: |
| agent_task.register_necessary_pidfiles() |
| |
| |
| def _recover_tasks(self, agent_tasks): |
| orphans = _drone_manager.get_orphaned_autoserv_processes() |
| |
| for agent_task in agent_tasks: |
| agent_task.recover() |
| if agent_task.monitor and agent_task.monitor.has_process(): |
| orphans.discard(agent_task.monitor.get_process()) |
| self.add_agent_task(agent_task) |
| |
| self._check_for_remaining_orphan_processes(orphans) |
| |
| |
| def _get_unassigned_entries(self, status): |
| for entry in scheduler_models.HostQueueEntry.fetch(where="status = '%s'" |
| % status): |
| if entry.status == status and not self.get_agents_for_entry(entry): |
| # The status can change during iteration, e.g., if job.run() |
| # sets a group of queue entries to Starting |
| yield entry |
| |
| |
| def _check_for_remaining_orphan_processes(self, orphans): |
| if not orphans: |
| return |
| subject = 'Unrecovered orphan autoserv processes remain' |
| message = '\n'.join(str(process) for process in orphans) |
| email_manager.manager.enqueue_notify_email(subject, message) |
| |
| die_on_orphans = global_config.global_config.get_config_value( |
| scheduler_config.CONFIG_SECTION, 'die_on_orphans', type=bool) |
| |
| if die_on_orphans: |
| raise RuntimeError(subject + '\n' + message) |
| |
| |
| def _recover_pending_entries(self): |
| for entry in self._get_unassigned_entries( |
| models.HostQueueEntry.Status.PENDING): |
| logging.info('Recovering Pending entry %s', entry) |
| entry.on_pending() |
| |
| |
| def _check_for_unrecovered_verifying_entries(self): |
| queue_entries = scheduler_models.HostQueueEntry.fetch( |
| where='status = "%s"' % models.HostQueueEntry.Status.VERIFYING) |
| unrecovered_hqes = [] |
| for queue_entry in queue_entries: |
| special_tasks = models.SpecialTask.objects.filter( |
| task__in=(models.SpecialTask.Task.CLEANUP, |
| models.SpecialTask.Task.VERIFY), |
| queue_entry__id=queue_entry.id, |
| is_complete=False) |
| if special_tasks.count() == 0: |
| unrecovered_hqes.append(queue_entry) |
| |
| if unrecovered_hqes: |
| message = '\n'.join(str(hqe) for hqe in unrecovered_hqes) |
| raise scheduler_lib.SchedulerError( |
| '%d unrecovered verifying host queue entries:\n%s' % |
| (len(unrecovered_hqes), message)) |
| |
| |
| def _schedule_special_tasks(self): |
| """ |
| Execute queued SpecialTasks that are ready to run on idle hosts. |
| |
| Special tasks include PreJobTasks like verify, reset and cleanup. |
| They are created through _schedule_new_jobs and associated with a hqe |
| This method translates SpecialTasks to the appropriate AgentTask and |
| adds them to the dispatchers agents list, so _handle_agents can execute |
| them. |
| """ |
| # When the host scheduler is responsible for acquisition we only want |
| # to run tasks with leased hosts. All hqe tasks will already have |
| # leased hosts, and we don't want to run frontend tasks till the host |
| # scheduler has vetted the assignment. Note that this doesn't include |
| # frontend tasks with hosts leased by other active hqes. |
| for task in self._job_query_manager.get_prioritized_special_tasks( |
| only_tasks_with_leased_hosts=not _inline_host_acquisition): |
| if self.host_has_agent(task.host): |
| continue |
| self.add_agent_task(self._get_agent_task_for_special_task(task)) |
| |
| |
| def _reverify_remaining_hosts(self): |
| # recover active hosts that have not yet been recovered, although this |
| # should never happen |
| message = ('Recovering active host %s - this probably indicates a ' |
| 'scheduler bug') |
| self._reverify_hosts_where( |
| "status IN ('Repairing', 'Verifying', 'Cleaning', 'Provisioning')", |
| print_message=message) |
| |
| |
| def _reverify_hosts_where(self, where, |
| print_message='Reverifying host %s'): |
| full_where='locked = 0 AND invalid = 0 AND ' + where |
| for host in scheduler_models.Host.fetch(where=full_where): |
| if self.host_has_agent(host): |
| # host has already been recovered in some way |
| continue |
| if self._host_has_scheduled_special_task(host): |
| # host will have a special task scheduled on the next cycle |
| continue |
| if print_message: |
| logging.info(print_message, host.hostname) |
| models.SpecialTask.objects.create( |
| task=models.SpecialTask.Task.CLEANUP, |
| host=models.Host.objects.get(id=host.id)) |
| |
| |
| def _recover_hosts(self): |
| # recover "Repair Failed" hosts |
| message = 'Reverifying dead host %s' |
| self._reverify_hosts_where("status = 'Repair Failed'", |
| print_message=message) |
| |
| |
| def _refresh_pending_queue_entries(self): |
| """ |
| Lookup the pending HostQueueEntries and call our HostScheduler |
| refresh() method given that list. Return the list. |
| |
| @returns A list of pending HostQueueEntries sorted in priority order. |
| """ |
| queue_entries = self._job_query_manager.get_pending_queue_entries( |
| only_hostless=not _inline_host_acquisition) |
| if not queue_entries: |
| return [] |
| return queue_entries |
| |
| |
| def _schedule_hostless_job(self, queue_entry): |
| """Schedule a hostless (suite) job. |
| |
| @param queue_entry: The queue_entry representing the hostless job. |
| """ |
| self.add_agent_task(HostlessQueueTask(queue_entry)) |
| queue_entry.set_status(models.HostQueueEntry.Status.STARTING) |
| |
| |
| def _schedule_host_job(self, host, queue_entry): |
| """Schedules a job on the given host. |
| |
| 1. Assign the host to the hqe, if it isn't already assigned. |
| 2. Create a SpecialAgentTask for the hqe. |
| 3. Activate the hqe. |
| |
| @param queue_entry: The job to schedule. |
| @param host: The host to schedule the job on. |
| """ |
| if self.host_has_agent(host): |
| host_agent_task = list(self._host_agents.get(host.id))[0].task |
| subject = 'Host with agents assigned to an HQE' |
| message = ('HQE: %s assigned host %s, but the host has ' |
| 'agent: %s for queue_entry %s. The HQE ' |
| 'will have to try and acquire a host next tick ' % |
| (queue_entry, host.hostname, host_agent_task, |
| host_agent_task.queue_entry)) |
| email_manager.manager.enqueue_notify_email(subject, message) |
| else: |
| self._host_scheduler.schedule_host_job(host, queue_entry) |
| |
| |
| def _schedule_new_jobs(self): |
| """ |
| Find any new HQEs and call schedule_pre_job_tasks for it. |
| |
| This involves setting the status of the HQE and creating a row in the |
| db corresponding the the special task, through |
| scheduler_models._queue_special_task. The new db row is then added as |
| an agent to the dispatcher through _schedule_special_tasks and |
| scheduled for execution on the drone through _handle_agents. |
| """ |
| queue_entries = self._refresh_pending_queue_entries() |
| |
| key = 'scheduler.jobs_per_tick' |
| new_hostless_jobs = 0 |
| new_jobs_with_hosts = 0 |
| new_jobs_need_hosts = 0 |
| host_jobs = [] |
| logging.debug('Processing %d queue_entries', len(queue_entries)) |
| |
| for queue_entry in queue_entries: |
| if queue_entry.is_hostless(): |
| self._schedule_hostless_job(queue_entry) |
| new_hostless_jobs = new_hostless_jobs + 1 |
| else: |
| host_jobs.append(queue_entry) |
| new_jobs_need_hosts = new_jobs_need_hosts + 1 |
| |
| stats.Gauge(key).send('new_hostless_jobs', new_hostless_jobs) |
| if not host_jobs: |
| return |
| if not _inline_host_acquisition: |
| message = ('Found %s jobs that need hosts though ' |
| '_inline_host_acquisition=%s. Will acquire hosts.' % |
| ([str(job) for job in host_jobs], |
| _inline_host_acquisition)) |
| email_manager.manager.enqueue_notify_email( |
| 'Processing unexpected host acquisition requests', message) |
| jobs_with_hosts = self._host_scheduler.find_hosts_for_jobs(host_jobs) |
| for host_assignment in jobs_with_hosts: |
| self._schedule_host_job(host_assignment.host, host_assignment.job) |
| new_jobs_with_hosts = new_jobs_with_hosts + 1 |
| |
| stats.Gauge(key).send('new_jobs_with_hosts', new_jobs_with_hosts) |
| stats.Gauge(key).send('new_jobs_without_hosts', |
| new_jobs_need_hosts - new_jobs_with_hosts) |
| |
| |
| def _schedule_running_host_queue_entries(self): |
| """ |
| Adds agents to the dispatcher. |
| |
| Any AgentTask, like the QueueTask, is wrapped in an Agent. The |
| QueueTask for example, will have a job with a control file, and |
| the agent will have methods that poll, abort and check if the queue |
| task is finished. The dispatcher runs the agent_task, as well as |
| other agents in it's _agents member, through _handle_agents, by |
| calling the Agents tick(). |
| |
| This method creates an agent for each HQE in one of (starting, running, |
| gathering, parsing, archiving) states, and adds it to the dispatcher so |
| it is handled by _handle_agents. |
| """ |
| for agent_task in self._get_queue_entry_agent_tasks(): |
| self.add_agent_task(agent_task) |
| |
| |
| def _schedule_delay_tasks(self): |
| for entry in scheduler_models.HostQueueEntry.fetch( |
| where='status = "%s"' % models.HostQueueEntry.Status.WAITING): |
| task = entry.job.schedule_delayed_callback_task(entry) |
| if task: |
| self.add_agent_task(task) |
| |
| |
| def _find_aborting(self): |
| """ |
| Looks through the afe_host_queue_entries for an aborted entry. |
| |
| The aborted bit is set on an HQE in many ways, the most common |
| being when a user requests an abort through the frontend, which |
| results in an rpc from the afe to abort_host_queue_entries. |
| """ |
| jobs_to_stop = set() |
| for entry in scheduler_models.HostQueueEntry.fetch( |
| where='aborted=1 and complete=0'): |
| logging.info('Aborting %s', entry) |
| |
| # The task would have started off with both is_complete and |
| # is_active = False. Aborted tasks are neither active nor complete. |
| # For all currently active tasks this will happen through the agent, |
| # but we need to manually update the special tasks that haven't |
| # started yet, because they don't have agents. |
| models.SpecialTask.objects.filter(is_active=False, |
| queue_entry_id=entry.id).update(is_complete=True) |
| |
| for agent in self.get_agents_for_entry(entry): |
| agent.abort() |
| entry.abort(self) |
| jobs_to_stop.add(entry.job) |
| logging.debug('Aborting %d jobs this tick.', len(jobs_to_stop)) |
| for job in jobs_to_stop: |
| job.stop_if_necessary() |
| |
| |
| def _find_aborted_special_tasks(self): |
| """ |
| Find SpecialTasks that have been marked for abortion. |
| |
| Poll the database looking for SpecialTasks that are active |
| and have been marked for abortion, then abort them. |
| """ |
| |
| # The completed and active bits are very important when it comes |
| # to scheduler correctness. The active bit is set through the prolog |
| # of a special task, and reset through the cleanup method of the |
| # SpecialAgentTask. The cleanup is called both through the abort and |
| # epilog. The complete bit is set in several places, and in general |
| # a hanging job will have is_active=1 is_complete=0, while a special |
| # task which completed will have is_active=0 is_complete=1. To check |
| # aborts we directly check active because the complete bit is set in |
| # several places, including the epilog of agent tasks. |
| aborted_tasks = models.SpecialTask.objects.filter(is_active=True, |
| is_aborted=True) |
| for task in aborted_tasks: |
| # There are 2 ways to get the agent associated with a task, |
| # through the host and through the hqe. A special task |
| # always needs a host, but doesn't always need a hqe. |
| for agent in self._host_agents.get(task.host.id, []): |
| if isinstance(agent.task, agent_task.SpecialAgentTask): |
| |
| # The epilog preforms critical actions such as |
| # queueing the next SpecialTask, requeuing the |
| # hqe etc, however it doesn't actually kill the |
| # monitor process and set the 'done' bit. Epilogs |
| # assume that the job failed, and that the monitor |
| # process has already written an exit code. The |
| # done bit is a necessary condition for |
| # _handle_agents to schedule any more special |
| # tasks against the host, and it must be set |
| # in addition to is_active, is_complete and success. |
| agent.task.epilog() |
| agent.task.abort() |
| |
| |
| def _can_start_agent(self, agent, num_started_this_cycle, |
| have_reached_limit): |
| # always allow zero-process agents to run |
| if agent.task.num_processes == 0: |
| return True |
| # don't allow any nonzero-process agents to run after we've reached a |
| # limit (this avoids starvation of many-process agents) |
| if have_reached_limit: |
| return False |
| # total process throttling |
| max_runnable_processes = _drone_manager.max_runnable_processes( |
| agent.task.owner_username, |
| agent.task.get_drone_hostnames_allowed()) |
| if agent.task.num_processes > max_runnable_processes: |
| return False |
| # if a single agent exceeds the per-cycle throttling, still allow it to |
| # run when it's the first agent in the cycle |
| if num_started_this_cycle == 0: |
| return True |
| # per-cycle throttling |
| if (num_started_this_cycle + agent.task.num_processes > |
| scheduler_config.config.max_processes_started_per_cycle): |
| return False |
| return True |
| |
| |
| def _handle_agents(self): |
| """ |
| Handles agents of the dispatcher. |
| |
| Appropriate Agents are added to the dispatcher through |
| _schedule_running_host_queue_entries. These agents each |
| have a task. This method runs the agents task through |
| agent.tick() leading to: |
| agent.start |
| prolog -> AgentTasks prolog |
| For each queue entry: |
| sets host status/status to Running |
| set started_on in afe_host_queue_entries |
| run -> AgentTasks run |
| Creates PidfileRunMonitor |
| Queues the autoserv command line for this AgentTask |
| via the drone manager. These commands are executed |
| through the drone managers execute actions. |
| poll -> AgentTasks/BaseAgentTask poll |
| checks the monitors exit_code. |
| Executes epilog if task is finished. |
| Executes AgentTasks _finish_task |
| finish_task is usually responsible for setting the status |
| of the HQE/host, and updating it's active and complete fileds. |
| |
| agent.is_done |
| Removed the agent from the dispatchers _agents queue. |
| Is_done checks the finished bit on the agent, that is |
| set based on the Agents task. During the agents poll |
| we check to see if the monitor process has exited in |
| it's finish method, and set the success member of the |
| task based on this exit code. |
| """ |
| num_started_this_cycle = 0 |
| have_reached_limit = False |
| # iterate over copy, so we can remove agents during iteration |
| logging.debug('Handling %d Agents', len(self._agents)) |
| for agent in list(self._agents): |
| self._log_extra_msg('Processing Agent with Host Ids: %s and ' |
| 'queue_entry ids:%s' % (agent.host_ids, |
| agent.queue_entry_ids)) |
| if not agent.started: |
| if not self._can_start_agent(agent, num_started_this_cycle, |
| have_reached_limit): |
| have_reached_limit = True |
| logging.debug('Reached Limit of allowed running Agents.') |
| continue |
| num_started_this_cycle += agent.task.num_processes |
| self._log_extra_msg('Starting Agent') |
| agent.tick() |
| self._log_extra_msg('Agent tick completed.') |
| if agent.is_done(): |
| self._log_extra_msg("Agent finished") |
| self.remove_agent(agent) |
| logging.info('%d running processes. %d added this cycle.', |
| _drone_manager.total_running_processes(), |
| num_started_this_cycle) |
| |
| |
| def _process_recurring_runs(self): |
| recurring_runs = models.RecurringRun.objects.filter( |
| start_date__lte=datetime.datetime.now()) |
| for rrun in recurring_runs: |
| # Create job from template |
| job = rrun.job |
| info = rpc_utils.get_job_info(job) |
| options = job.get_object_dict() |
| |
| host_objects = info['hosts'] |
| one_time_hosts = info['one_time_hosts'] |
| metahost_objects = info['meta_hosts'] |
| dependencies = info['dependencies'] |
| atomic_group = info['atomic_group'] |
| |
| for host in one_time_hosts or []: |
| this_host = models.Host.create_one_time_host(host.hostname) |
| host_objects.append(this_host) |
| |
| try: |
| rpc_utils.create_new_job(owner=rrun.owner.login, |
| options=options, |
| host_objects=host_objects, |
| metahost_objects=metahost_objects, |
| atomic_group=atomic_group) |
| |
| except Exception, ex: |
| logging.exception(ex) |
| #TODO send email |
| |
| if rrun.loop_count == 1: |
| rrun.delete() |
| else: |
| if rrun.loop_count != 0: # if not infinite loop |
| # calculate new start_date |
| difference = datetime.timedelta(seconds=rrun.loop_period) |
| rrun.start_date = rrun.start_date + difference |
| rrun.loop_count -= 1 |
| rrun.save() |
| |
| |
| SiteDispatcher = utils.import_site_class( |
| __file__, 'autotest_lib.scheduler.site_monitor_db', |
| 'SiteDispatcher', BaseDispatcher) |
| |
| class Dispatcher(SiteDispatcher): |
| pass |
| |
| |
| class Agent(object): |
| """ |
| An agent for use by the Dispatcher class to perform a task. An agent wraps |
| around an AgentTask mainly to associate the AgentTask with the queue_entry |
| and host ids. |
| |
| The following methods are required on all task objects: |
| poll() - Called periodically to let the task check its status and |
| update its internal state. If the task succeeded. |
| is_done() - Returns True if the task is finished. |
| abort() - Called when an abort has been requested. The task must |
| set its aborted attribute to True if it actually aborted. |
| |
| The following attributes are required on all task objects: |
| aborted - bool, True if this task was aborted. |
| success - bool, True if this task succeeded. |
| queue_entry_ids - A sequence of HostQueueEntry ids this task handles. |
| host_ids - A sequence of Host ids this task represents. |
| """ |
| |
| |
| def __init__(self, task): |
| """ |
| @param task: An instance of an AgentTask. |
| """ |
| self.task = task |
| |
| # This is filled in by Dispatcher.add_agent() |
| self.dispatcher = None |
| |
| self.queue_entry_ids = task.queue_entry_ids |
| self.host_ids = task.host_ids |
| |
| self.started = False |
| self.finished = False |
| |
| |
| def tick(self): |
| self.started = True |
| if not self.finished: |
| self.task.poll() |
| if self.task.is_done(): |
| self.finished = True |
| |
| |
| def is_done(self): |
| return self.finished |
| |
| |
| def abort(self): |
| if self.task: |
| self.task.abort() |
| if self.task.aborted: |
| # tasks can choose to ignore aborts |
| self.finished = True |
| |
| |
| class AbstractQueueTask(agent_task.AgentTask, agent_task.TaskWithJobKeyvals): |
| """ |
| Common functionality for QueueTask and HostlessQueueTask |
| """ |
| def __init__(self, queue_entries): |
| super(AbstractQueueTask, self).__init__() |
| self.job = queue_entries[0].job |
| self.queue_entries = queue_entries |
| |
| |
| def _keyval_path(self): |
| return os.path.join(self._working_directory(), self._KEYVAL_FILE) |
| |
| |
| def _write_control_file(self, execution_path): |
| control_path = _drone_manager.attach_file_to_execution( |
| execution_path, self.job.control_file) |
| return control_path |
| |
| |
| # TODO: Refactor into autoserv_utils. crbug.com/243090 |
| def _command_line(self): |
| execution_path = self.queue_entries[0].execution_path() |
| control_path = self._write_control_file(execution_path) |
| hostnames = ','.join(entry.host.hostname |
| for entry in self.queue_entries |
| if not entry.is_hostless()) |
| |
| execution_tag = self.queue_entries[0].execution_tag() |
| params = _autoserv_command_line( |
| hostnames, |
| ['-P', execution_tag, '-n', |
| _drone_manager.absolute_path(control_path)], |
| job=self.job, verbose=False) |
| if self.job.is_image_update_job(): |
| params += ['--image', self.job.update_image_path] |
| |
| return params |
| |
| |
| @property |
| def num_processes(self): |
| return len(self.queue_entries) |
| |
| |
| @property |
| def owner_username(self): |
| return self.job.owner |
| |
| |
| def _working_directory(self): |
| return self._get_consistent_execution_path(self.queue_entries) |
| |
| |
| def prolog(self): |
| queued_key, queued_time = self._job_queued_keyval(self.job) |
| keyval_dict = self.job.keyval_dict() |
| keyval_dict[queued_key] = queued_time |
| group_name = self.queue_entries[0].get_group_name() |
| if group_name: |
| keyval_dict['host_group_name'] = group_name |
| self._write_keyvals_before_job(keyval_dict) |
| for queue_entry in self.queue_entries: |
| queue_entry.set_status(models.HostQueueEntry.Status.RUNNING) |
| queue_entry.set_started_on_now() |
| |
| |
| def _write_lost_process_error_file(self): |
| error_file_path = os.path.join(self._working_directory(), 'job_failure') |
| _drone_manager.write_lines_to_file(error_file_path, |
| [_LOST_PROCESS_ERROR]) |
| |
| |
| def _finish_task(self): |
| if not self.monitor: |
| return |
| |
| self._write_job_finished() |
| |
| if self.monitor.lost_process: |
| self._write_lost_process_error_file() |
| |
| |
| def _write_status_comment(self, comment): |
| _drone_manager.write_lines_to_file( |
| os.path.join(self._working_directory(), 'status.log'), |
| ['INFO\t----\t----\t' + comment], |
| paired_with_process=self.monitor.get_process()) |
| |
| |
| def _log_abort(self): |
| if not self.monitor or not self.monitor.has_process(): |
| return |
| |
| # build up sets of all the aborted_by and aborted_on values |
| aborted_by, aborted_on = set(), set() |
| for queue_entry in self.queue_entries: |
| if queue_entry.aborted_by: |
| aborted_by.add(queue_entry.aborted_by) |
| t = int(time.mktime(queue_entry.aborted_on.timetuple())) |
| aborted_on.add(t) |
| |
| # extract some actual, unique aborted by value and write it out |
| # TODO(showard): this conditional is now obsolete, we just need to leave |
| # it in temporarily for backwards compatibility over upgrades. delete |
| # soon. |
| assert len(aborted_by) <= 1 |
| if len(aborted_by) == 1: |
| aborted_by_value = aborted_by.pop() |
| aborted_on_value = max(aborted_on) |
| else: |
| aborted_by_value = 'autotest_system' |
| aborted_on_value = int(time.time()) |
| |
| self._write_keyval_after_job("aborted_by", aborted_by_value) |
| self._write_keyval_after_job("aborted_on", aborted_on_value) |
| |
| aborted_on_string = str(datetime.datetime.fromtimestamp( |
| aborted_on_value)) |
| self._write_status_comment('Job aborted by %s on %s' % |
| (aborted_by_value, aborted_on_string)) |
| |
| |
| def abort(self): |
| super(AbstractQueueTask, self).abort() |
| self._log_abort() |
| self._finish_task() |
| |
| |
| def epilog(self): |
| super(AbstractQueueTask, self).epilog() |
| self._finish_task() |
| |
| |
| class QueueTask(AbstractQueueTask): |
| def __init__(self, queue_entries): |
| super(QueueTask, self).__init__(queue_entries) |
| self._set_ids(queue_entries=queue_entries) |
| |
| |
| def prolog(self): |
| self._check_queue_entry_statuses( |
| self.queue_entries, |
| allowed_hqe_statuses=(models.HostQueueEntry.Status.STARTING, |
| models.HostQueueEntry.Status.RUNNING), |
| allowed_host_statuses=(models.Host.Status.PENDING, |
| models.Host.Status.RUNNING)) |
| |
| super(QueueTask, self).prolog() |
| |
| for queue_entry in self.queue_entries: |
| self._write_host_keyvals(queue_entry.host) |
| queue_entry.host.set_status(models.Host.Status.RUNNING) |
| queue_entry.host.update_field('dirty', 1) |
| |
| |
| def _finish_task(self): |
| super(QueueTask, self)._finish_task() |
| |
| for queue_entry in self.queue_entries: |
| queue_entry.set_status(models.HostQueueEntry.Status.GATHERING) |
| queue_entry.host.set_status(models.Host.Status.RUNNING) |
| |
| |
| def _command_line(self): |
| invocation = super(QueueTask, self)._command_line() |
| return invocation + ['--verify_job_repo_url'] |
| |
| |
| class HostlessQueueTask(AbstractQueueTask): |
| def __init__(self, queue_entry): |
| super(HostlessQueueTask, self).__init__([queue_entry]) |
| self.queue_entry_ids = [queue_entry.id] |
| |
| |
| def prolog(self): |
| self.queue_entries[0].update_field('execution_subdir', 'hostless') |
| super(HostlessQueueTask, self).prolog() |
| |
| |
| def _finish_task(self): |
| super(HostlessQueueTask, self)._finish_task() |
| |
| # When a job is added to database, its initial status is always |
| # Starting. In a scheduler tick, scheduler finds all jobs in Starting |
| # status, check if any of them can be started. If scheduler hits some |
| # limit, e.g., max_hostless_jobs_per_drone, max_jobs_started_per_cycle, |
| # scheduler will leave these jobs in Starting status. Otherwise, the |
| # jobs' status will be changed to Running, and an autoserv process will |
| # be started in drone for each of these jobs. |
| # If the entry is still in status Starting, the process has not started |
| # yet. Therefore, there is no need to parse and collect log. Without |
| # this check, exception will be raised by scheduler as execution_subdir |
| # for this queue entry does not have a value yet. |
| hqe = self.queue_entries[0] |
| if hqe.status != models.HostQueueEntry.Status.STARTING: |
| hqe.set_status(models.HostQueueEntry.Status.PARSING) |
| |
| |
| if __name__ == '__main__': |
| main() |