Function task_loop

src/python/bot/startup/run_bot.py:164–254

Executes tasks indefinitely.

()

def task_loop():
  """Executes tasks indefinitely."""
  # Defer heavy task imports to prevent issues with multiprocessing.Process
  from clusterfuzz._internal.bot.tasks import commands

  clean_exit = False
  execution_count = 0
  max_task_executions = _get_max_task_executions()

  while True:
    stacktrace = ''
    exception_occurred = False
    task = None
    # This caches the current environment on first run. Don't move this.
    environment.reset_environment()
    try:
      if update_task_enabled():
        logs.info("Running update task.")
        # Run regular updates.
        # TODO(metzman): Move this after utask_main execution
        # so that utasks can't be updated on subsequent attempts.
        update_task.run()
        update_task.track_revision()
      else:
        logs.info(
            "Update task not enabled. Running environment cleanup and platform "
            "init scripts.")
        update_task.prepare_environment_for_new_task()
        update_task.run_platform_init_scripts()

      if environment.is_uworker():
        # Batch/Swarming tasks only run one at a time.
        sys.exit(utasks.uworker_bot_main())

      if environment.get_value('SCHEDULE_UTASK_MAINS'):
        # If the bot is configured to schedule utask_mains, don't run any other
        # tasks because scheduling these tasks is more important than executing
        # any one other task.

        # TODO(metzman): Convert this to a k8s cron.
        schedule_utask_mains()
        continue

      if environment.is_tworker():
        task = tasks.tworker_get_task(
            override_queue=_get_tworker_queue_override())
      else:
        task = tasks.get_task()

      if not task:
        continue

      with _Monitor(task):
        with task.lease():
          # Execute the command and delete the task.
          commands.process_command(task)
    except SystemExit as e:
      exception_occurred = True