mirror of
https://github.com/Hestia-Homes/Model.git
synced 2026-07-12 13:29:04 +00:00
30 lines
1.2 KiB
Python
30 lines
1.2 KiB
Python
import os
|
|
from collections.abc import Generator
|
|
from contextlib import contextmanager
|
|
|
|
from sqlalchemy.pool import NullPool
|
|
from sqlmodel import Session
|
|
|
|
from infrastructure.postgres.config import PostgresConfig
|
|
from infrastructure.postgres.engine import make_engine
|
|
from orchestration.task_orchestrator import TaskOrchestrator
|
|
from repositories.tasks.subtask_postgres_repository import SubTaskPostgresRepository
|
|
from repositories.tasks.task_postgres_repository import TaskPostgresRepository
|
|
|
|
|
|
@contextmanager
|
|
def default_orchestrator() -> Generator[TaskOrchestrator, None, None]:
|
|
"""Yield a TaskOrchestrator wired to a fresh Postgres session.
|
|
|
|
Connection params come from os.environ via PostgresConfig.from_env. Each
|
|
handler invocation gets its own session, cleaned up on context exit.
|
|
|
|
NullPool is intentional: a new engine is created on every invocation, so
|
|
pooling would accumulate idle connections across warm Lambda containers.
|
|
"""
|
|
engine = make_engine(PostgresConfig.from_env(dict(os.environ)), poolclass=NullPool)
|
|
with Session(engine) as session:
|
|
yield TaskOrchestrator(
|
|
task_repo=TaskPostgresRepository(session=session),
|
|
subtask_repo=SubTaskPostgresRepository(session=session),
|
|
)
|