154 lines
5.7 KiB
Python
154 lines
5.7 KiB
Python
"""画像生成的数据访问层（Repository）。
|
||
|
||
按 MVC+S 边界：**Service 不得直接建 Session / 直接查 Model**，所有库交互收口在这里。
|
||
|
||
只做两件事：
|
||
1. **只读**组装画像所需的权威事实（`docs/00` §6.4.6：画像生成只允许读取**已提交的**业务事实
|
||
和状态为**有效**的中期记忆）；
|
||
2. **写入**画像版本与两条跨存储同步事件（同事务，由调用方管理事务）。
|
||
"""
|
||
|
||
from __future__ import annotations
|
||
|
||
from datetime import datetime
|
||
from typing import Any
|
||
|
||
from sqlalchemy import text
|
||
from sqlalchemy.ext.asyncio import AsyncSession
|
||
|
||
from app.model.memory import MemorySyncOutbox
|
||
|
||
#: 画像快照：查询客户主表。
|
||
_SQL_PROFILE = text("""
|
||
SELECT customer_id, investor_type, investment_horizon, preferred_asset_class,
|
||
trading_frequency, total_asset, behavior_score, risk_tags, last_active_at
|
||
FROM fin_customer_profile
|
||
WHERE customer_id = :customer_id
|
||
""")
|
||
|
||
#: 风险测评：取**最新一条**，含有效期（用于判定是否过期）。
|
||
_SQL_ASSESSMENT = text("""
|
||
SELECT id, questionnaire_version, investor_type, total_score, assessed_at, valid_until
|
||
FROM fin_risk_assessment
|
||
WHERE customer_id = :customer_id
|
||
ORDER BY assessed_at DESC, id DESC
|
||
LIMIT 1
|
||
""")
|
||
|
||
#: 当前画像版本（取 `is_current=1` 那一条）。
|
||
_SQL_CURRENT_SNAPSHOT = text("""
|
||
SELECT id, profile_uuid, version, snapshot, snapshot_hash, generation_basis
|
||
FROM profile_snapshots
|
||
WHERE customer_id = :customer_id AND is_current = 1
|
||
LIMIT 1
|
||
""")
|
||
|
||
#: 下一版本号 = 当前最大版本 + 1
|
||
#: （`uk_profile_snapshot_version (customer_id, version)` 要求版本号按客户唯一）。
|
||
_SQL_NEXT_VERSION = text("""
|
||
SELECT COALESCE(MAX(version), 0) + 1
|
||
FROM profile_snapshots WHERE customer_id = :customer_id
|
||
""")
|
||
|
||
#: 旧画像置非当前（`current_customer_id` 有唯一键，必须先置 0 才能插新的当前版本）。
|
||
_SQL_CLEAR_CURRENT = text("""
|
||
UPDATE profile_snapshots
|
||
SET is_current = 0, current_customer_id = NULL, updated_at = :now
|
||
WHERE customer_id = :customer_id AND is_current = 1
|
||
""")
|
||
|
||
_SQL_INSERT_SNAPSHOT = text("""
|
||
INSERT INTO profile_snapshots
|
||
(profile_uuid, customer_id, version, snapshot, generation_basis, snapshot_hash,
|
||
is_current, current_customer_id, generated_at, created_at, updated_at)
|
||
VALUES
|
||
(:profile_uuid, :customer_id, :version, :snapshot, :generation_basis, :snapshot_hash,
|
||
1, :customer_id, :now, :now, :now)
|
||
""")
|
||
|
||
|
||
class ProfileRepository:
|
||
def __init__(self, session: AsyncSession) -> None:
|
||
self._session = session
|
||
|
||
# --- 只读事实 ---------------------------------------------------------
|
||
|
||
async def profile_row(self, customer_id: int) -> dict[str, Any] | None:
|
||
result = await self._session.execute(_SQL_PROFILE, {"customer_id": customer_id})
|
||
row = result.mappings().first()
|
||
return dict(row) if row is not None else None
|
||
|
||
async def latest_assessment(self, customer_id: int) -> dict[str, Any] | None:
|
||
result = await self._session.execute(_SQL_ASSESSMENT, {"customer_id": customer_id})
|
||
row = result.mappings().first()
|
||
return dict(row) if row is not None else None
|
||
|
||
async def current_snapshot(self, customer_id: int) -> dict[str, Any] | None:
|
||
result = await self._session.execute(_SQL_CURRENT_SNAPSHOT, {"customer_id": customer_id})
|
||
row = result.mappings().first()
|
||
return dict(row) if row is not None else None
|
||
|
||
async def next_version(self, customer_id: int) -> int:
|
||
return int(await self._session.scalar(_SQL_NEXT_VERSION, {"customer_id": customer_id}))
|
||
|
||
# --- 写入（调用方管事务） ---------------------------------------------
|
||
|
||
async def clear_current(self, customer_id: int, *, now: datetime) -> None:
|
||
"""把旧画像的 `is_current` 置 0 —— 必须在插入新版本**之前**调用。
|
||
|
||
`uk_profile_snapshot_current` 建在 `current_customer_id` 上（唯一），
|
||
所以同一客户同时只能有一条 `is_current=1`；顺序颠倒会撞唯一键。
|
||
"""
|
||
await self._session.execute(_SQL_CLEAR_CURRENT, {"customer_id": customer_id, "now": now})
|
||
|
||
async def insert_snapshot(
|
||
self,
|
||
*,
|
||
customer_id: int,
|
||
version: int,
|
||
profile_uuid: str,
|
||
snapshot_json: str,
|
||
generation_basis_json: str,
|
||
snapshot_hash: str,
|
||
now: datetime,
|
||
) -> None:
|
||
await self._session.execute(_SQL_INSERT_SNAPSHOT, {
|
||
"profile_uuid": profile_uuid,
|
||
"customer_id": customer_id,
|
||
"version": version,
|
||
"snapshot": snapshot_json,
|
||
"generation_basis": generation_basis_json,
|
||
"snapshot_hash": snapshot_hash,
|
||
"now": now,
|
||
})
|
||
|
||
def add_sync_event(
|
||
self,
|
||
*,
|
||
event_uuid: str,
|
||
target_store: str,
|
||
aggregate_uuid: str,
|
||
aggregate_version: int,
|
||
payload: dict[str, Any],
|
||
now: datetime,
|
||
status: str,
|
||
operation: str,
|
||
) -> None:
|
||
"""追加一条跨存储同步事件。
|
||
|
||
唯一键 `uk_memory_sync_event (event_uuid, target_store)`：**同一 `event_uuid` 允许
|
||
对两个目标库各写一条**，因此 Milvus 与 Neo4j 共用同一个 `event_uuid`。
|
||
"""
|
||
self._session.add(MemorySyncOutbox(
|
||
event_uuid=event_uuid,
|
||
aggregate_type="profile",
|
||
aggregate_uuid=aggregate_uuid,
|
||
aggregate_version=aggregate_version,
|
||
target_store=target_store,
|
||
operation=operation,
|
||
payload=payload,
|
||
status=status,
|
||
retry_count=0,
|
||
created_at=now,
|
||
))
|