初始化提交

backend/app/repositories/idempotency.py

@@ -0,0 +1,33 @@
+"""幂等键数据访问。"""
+
+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()]