fix(agent): co-locate ansible candidate producer
Some checks failed
CD Pipeline / workflow-shape (push) Successful in 0s
CD Pipeline / cancel-stale-cd (push) Has been skipped
CD Pipeline / tests (push) Successful in 2m35s
CD Pipeline / post-deploy-checks (push) Has been cancelled
CD Pipeline / build-and-deploy (push) Has been cancelled

This commit is contained in:
ogt
2026-07-10 17:51:21 +08:00
parent aceee39b4a
commit 125b3a3126
4 changed files with 44 additions and 2 deletions

View File

@@ -608,6 +608,21 @@ async def _main() -> None:
# Initialize and start worker
worker = await init_signal_worker()
ansible_candidate_backfill_task: asyncio.Task[Any] | None = None
if settings.ENABLE_AWOOOP_ANSIBLE_CANDIDATE_BACKFILL_WORKER:
from src.jobs.awooop_ansible_candidate_backfill_job import (
run_awooop_ansible_candidate_backfill_loop,
)
ansible_candidate_backfill_task = asyncio.create_task(
run_awooop_ansible_candidate_backfill_loop(),
name="run_awooop_ansible_candidate_backfill_loop",
)
logger.info(
"signal_worker_ansible_candidate_backfill_loop_started",
interval_seconds=settings.AWOOOP_ANSIBLE_CANDIDATE_BACKFILL_INTERVAL_SECONDS,
batch_limit=settings.AWOOOP_ANSIBLE_CANDIDATE_BACKFILL_BATCH_LIMIT,
)
ansible_check_mode_task: asyncio.Task[Any] | None = None
if settings.ENABLE_AWOOOP_ANSIBLE_CHECK_MODE_WORKER:
from src.jobs.awooop_ansible_check_mode_job import (
@@ -649,6 +664,12 @@ async def _main() -> None:
# Graceful shutdown
logger.info("signal_worker_shutting_down")
if ansible_candidate_backfill_task is not None:
ansible_candidate_backfill_task.cancel()
try:
await ansible_candidate_backfill_task
except asyncio.CancelledError:
pass
if ansible_check_mode_task is not None:
ansible_check_mode_task.cancel()
try: