Bernd Worsch
5e4dc6c946
activities.py: - init_session_factory(url): module-level async_sessionmaker init, called once from worker.py before workers start - load_activity_definition(activity_id): queries activity_definitions by UUID, returns JSON-serialisable dict; raises ApplicationError (non_retryable=True) if row not found worker.py: - reads ACTCORE_DB_URL at startup, fails fast if missing - calls init_session_factory() before connecting to Temporal Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
72 lines
2.1 KiB
Python
72 lines
2.1 KiB
Python
"""Temporal worker entrypoint for activity-core.
|
|
|
|
Starts two workers (wired up in T20):
|
|
- orchestrator-tq: RunActivityWorkflow + its activities
|
|
- task-execution-tq: TaskExecutorWorkflow
|
|
|
|
Run with:
|
|
TEMPORAL_HOST=localhost:7233 \
|
|
ACTCORE_DB_URL=postgresql+asyncpg://actcore:actcore@localhost:5433/actcore \
|
|
python -m activity_core.worker
|
|
|
|
Environment variables:
|
|
TEMPORAL_HOST Temporal frontend address (default: localhost:7233)
|
|
TEMPORAL_NAMESPACE Temporal namespace (default: default)
|
|
ACTCORE_DB_URL App DB connection string (required)
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
import asyncio
|
|
import os
|
|
|
|
from temporalio.client import Client
|
|
from temporalio.worker import Worker
|
|
|
|
from activity_core.activities import (
|
|
init_session_factory,
|
|
load_activity_definition,
|
|
log_run,
|
|
resolve_context,
|
|
)
|
|
from activity_core.workflows import RunActivityWorkflow, TaskExecutorWorkflow
|
|
|
|
TEMPORAL_HOST = os.environ.get("TEMPORAL_HOST", "localhost:7233")
|
|
TEMPORAL_NAMESPACE = os.environ.get("TEMPORAL_NAMESPACE", "default")
|
|
|
|
ORCHESTRATOR_TASK_QUEUE = "orchestrator-tq"
|
|
TASK_EXECUTION_TASK_QUEUE = "task-execution-tq"
|
|
|
|
|
|
async def run() -> None:
|
|
db_url = os.environ.get("ACTCORE_DB_URL")
|
|
if not db_url:
|
|
raise RuntimeError("ACTCORE_DB_URL is required")
|
|
init_session_factory(db_url)
|
|
|
|
client = await Client.connect(TEMPORAL_HOST, namespace=TEMPORAL_NAMESPACE)
|
|
|
|
orchestrator_worker = Worker(
|
|
client,
|
|
task_queue=ORCHESTRATOR_TASK_QUEUE,
|
|
workflows=[RunActivityWorkflow],
|
|
activities=[load_activity_definition, resolve_context, log_run],
|
|
)
|
|
|
|
task_worker = Worker(
|
|
client,
|
|
task_queue=TASK_EXECUTION_TASK_QUEUE,
|
|
workflows=[TaskExecutorWorkflow],
|
|
activities=[],
|
|
)
|
|
|
|
async with orchestrator_worker, task_worker:
|
|
print(
|
|
f"Workers running — queues: {ORCHESTRATOR_TASK_QUEUE!r}, "
|
|
f"{TASK_EXECUTION_TASK_QUEUE!r} (namespace={TEMPORAL_NAMESPACE!r})"
|
|
)
|
|
await asyncio.Future() # run until cancelled
|
|
|
|
|
|
if __name__ == "__main__":
|
|
asyncio.run(run())
|