def main_without_exception_handling(options): scheduler_lib.setup_logging(os.environ.get('AUTOTEST_SCHEDULER_LOG_DIR', None), None, timestamped_logfile_prefix='shard_client') logging.info("Setting signal handler.") signal.signal(signal.SIGINT, handle_signal) signal.signal(signal.SIGTERM, handle_signal) logging.info("Starting shard client.") global _heartbeat_client _heartbeat_client = get_shard_client() _heartbeat_client.loop(options.lifetime_hours)
def main_without_exception_handling(): parser = argparse.ArgumentParser(description='Shard client.') options = parser.parse_args() scheduler_lib.setup_logging(os.environ.get('AUTOTEST_SCHEDULER_LOG_DIR', None), None, timestamped_logfile_prefix='shard_client') logging.info("Setting signal handler.") signal.signal(signal.SIGINT, handle_signal) signal.signal(signal.SIGTERM, handle_signal) logging.info("Starting shard client.") global _heartbeat_client _heartbeat_client = get_shard_client() _heartbeat_client.loop()
def initialize(testing=False): """Initialize the host scheduler.""" if testing: # Don't import testing utilities unless we're in testing mode, # as the database imports have side effects. from autotest_lib.scheduler import rdb_testing_utils rdb_testing_utils.FileDatabaseHelper().initialize_database_for_testing( db_file_path=rdb_testing_utils.FileDatabaseHelper.DB_FILE) global _db_manager _db_manager = scheduler_lib.ConnectionManager() scheduler_lib.setup_logging(os.environ.get('AUTOTEST_SCHEDULER_LOG_DIR', None), None, timestamped_logfile_prefix='host_scheduler') logging.info("Setting signal handler") signal.signal(signal.SIGINT, handle_signal) signal.signal(signal.SIGTERM, handle_signal) scheduler_models.initialize()