wooo/awoooi
1
1
Fork 0
Code Issues 2 Pull Requests 1 Actions 2 Packages Projects Releases Wiki Activity

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

Browse Source
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
Download Patch File Download Diff File Expand all files Collapse all files

8
apps/api/src/main.py
View File

@@ -638,10 +638,14 @@ async def lifespan(_app: FastAPI) -> AsyncGenerator[None, None]:
from src.jobs.awooop_ansible_candidate_backfill_job import (
run_awooop_ansible_candidate_backfill_loop,
)
schedule_api_background_task(run_awooop_ansible_candidate_backfill_loop())
scheduled_task = schedule_api_background_task(
run_awooop_ansible_candidate_backfill_loop()
)
logger.info(
"awooop_ansible_candidate_backfill_worker_scheduled",
"awooop_ansible_candidate_backfill_worker_schedule_evaluated",
enabled=settings.ENABLE_AWOOOP_ANSIBLE_CANDIDATE_BACKFILL_WORKER,
scheduled_in_api_process=scheduled_task is not None,
production_process_owner="awoooi-worker",
interval_seconds=settings.AWOOOP_ANSIBLE_CANDIDATE_BACKFILL_INTERVAL_SECONDS,
)
except Exception as e:

21
apps/api/src/workers/signal_worker.py
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: