"""幂等键数据访问。"""

from sqlalchemy import func, select
from sqlalchemy.ext.asyncio import AsyncSession

from app.models.idempotency import IdempotencyKey


async def get_by_scope_key(session: AsyncSession, *, scope: str, key: str) -> IdempotencyKey | None:
    result = await session.execute(
        select(IdempotencyKey).where(IdempotencyKey.scope == scope, IdempotencyKey.key == key)
    )
    return result.scalar_one_or_none()


async def list_by_trace(session: AsyncSession, trace_id: str) -> list[IdempotencyKey]:
    """链路追踪：按 trace_id 查询幂等键（请求级），按 created_at 升序。"""
    result = await session.execute(
        select(IdempotencyKey)
        .where(IdempotencyKey.trace_id == trace_id)
        .order_by(IdempotencyKey.created_at.asc())
    )
    return list(result.scalars().all())


async def summarize_by_trace(session: AsyncSession):
    """GROUP BY trace_id：(trace_id, 计数, 最近时间)，仅 trace_id 非空。供链路追踪列表聚合。"""
    result = await session.execute(
        select(IdempotencyKey.trace_id, func.count(), func.max(IdempotencyKey.created_at))
        .where(IdempotencyKey.trace_id.is_not(None))
        .group_by(IdempotencyKey.trace_id)
    )
    return [(row[0], int(row[1]), row[2]) for row in result.all()]