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()