"""平台设置快照(Phase 4 Task 5A)——resolver 平台默认回退链的数据源。 PlatformSetting 是自由 KV 表(config_routes.settings_router 的 CRUD); lifespan 启动时全量快照进进程内存,PUT /api/platform/settings 提交后 refresh 热刷新——平台默认回退链即时生效,无需重启进程。 - 快照键名约定对齐 admin 平台设置页的推荐键(ai.api_key / ai.base_url / ai.model / smtp.host / smtp.port / smtp.user / smtp.password),但快照 本身不限定键集合——自由 KV 未识别的键照存照读(消费侧只取自己认识的); - 值语义: value 为 JSON Text 直存,读回还原类型(json.loads 失败的手工 裸文本回退原串——与 config_routes._decode_setting 同款容错);空串/ None 视同未设置,消费侧(config_views)的 ``or`` 空槽链条天然跳过; - 单进程纪律: 内存快照与 token 缓存/登录限流同约定,uvicorn 必须 --workers 1——多 worker 下各进程快照各自为政,PUT 后互相不可见; - 密钥值只在本模块与库之间流转,不进日志、不进异常信息。 """ import json from sqlalchemy import select from sqlalchemy.ext.asyncio import AsyncEngine, AsyncSession from app.platform.db import PlatformSetting # 进程内存快照(模块级单例,同 auth.token_store 模式)。启动为空 dict—— # 未 load 过(测试直调 resolver)时 get 一律 None,resolver 直接落到 env 兜底, # 行为与接线前一致。 _snapshot: dict = {} def _decode(raw: str): """Text 列还原类型:json.loads 失败(手工 SQL 塞的裸文本)回退原串。""" try: return json.loads(raw) except ValueError: return raw async def load_snapshot(engine: AsyncEngine) -> dict: """全表读入内存快照(lifespan 启动装配)。整表替换而非逐键合并—— 库里删掉的键(未来 DELETE 入口)不会在快照里残留。""" async with AsyncSession(engine, expire_on_commit=False) as session: rows = (await session.execute( select(PlatformSetting))).scalars().all() _snapshot.clear() _snapshot.update({row.key: _decode(row.value) for row in rows}) return dict(_snapshot) def get(key: str): """快照取值:缺键/未加载/空值由调用方按"未设置"处理(返回原样, None/空串/0 的取舍归消费侧 ``or`` 链条)。""" return _snapshot.get(key) async def refresh(engine: AsyncEngine) -> None: """重读全表(PUT settings 提交后调用,即时生效)。""" await load_snapshot(engine) def _reset_for_test() -> None: """单例复位(测试隔离):快照清空。conftest autouse 前后各跑一次, 防 resolver 直调测试之间串快照(同 auth._reset_for_test 纪律)。""" _snapshot.clear()