Archived
feat(db-async): 站点数据异步读写仓储 + FastAPI 生命周期
- core/repo.py:经由 core.db(aiosqlite) 读写 auth_system 的**同一份** site.db (settings/announcements/users 只读/ledger 只读),字段口径对齐 auth_system; 读并发、写串行、WAL 读写同时进行 - web/routers/site.py:/api/v1/site/announcements、/api/v1/site/stats(异步多表并发读) - web/app.py:lifespan 关闭时释放 aiohttp 连接池(含按 loop 隔离的 session) 写路径说明:迁移期写操作仍归 auth_system 的 _DB_LOCK 事务口径,避免两套写 路径互相踩踏;本层先只做读。 验证:pytest 22/22(新增 6 项含「异步读 + 同步写者并发 WAL」); /api/v1/site/* 实测读到共享库(3 用户/1 管理员);ruff/pyright 干净。
This commit is contained in:
1 parent
ce7ea66d6a
commit
0f2e72db44
4 files changed
+219
No files matched your search
@@ -0,0 +1,61 @@
|
||||
# -*- coding: utf-8 -*-
|
||||
"""站点数据异步读写仓储(FastAPI 栈用)。
|
||||
|
||||
直接读写 auth_system 建立的同一份 SQLite(同一 DB_PATH / 同一 schema),
|
||||
但走 core.db 的 aiosqlite 异步层——读并发、写串行、WAL 读写同时进行。
|
||||
|
||||
设计约束:本模块**不改动** auth_system 既有 schema,只做读写;字段口径与
|
||||
auth_system 保持一致(键名、默认值)。迁移期两条栈共存、共库。
|
||||
"""
|
||||
from typing import Any
|
||||
|
||||
from core import db
|
||||
|
||||
# ---------- settings ----------
|
||||
|
||||
async def get_setting(key: str, default: str = '') -> str:
|
||||
"""读单个站点设置(对应 auth_system.get_setting)。"""
|
||||
row = await db.fetch_one('SELECT value FROM settings WHERE key = ?', (key,))
|
||||
return row['value'] if row is not None else default
|
||||
|
||||
|
||||
async def set_setting(key: str, value: str) -> None:
|
||||
"""写站点设置(upsert)。"""
|
||||
await db.execute(
|
||||
'INSERT INTO settings (key, value) VALUES (?, ?) '
|
||||
'ON CONFLICT(key) DO UPDATE SET value = excluded.value',
|
||||
(key, value),
|
||||
)
|
||||
|
||||
|
||||
async def all_settings() -> dict[str, str]:
|
||||
rows = await db.fetch_all('SELECT key, value FROM settings')
|
||||
return {r['key']: r['value'] for r in rows}
|
||||
|
||||
|
||||
# ---------- announcements ----------
|
||||
|
||||
async def list_announcements(limit: int = 20) -> list[dict[str, Any]]:
|
||||
rows = await db.fetch_all(
|
||||
'SELECT * FROM announcements ORDER BY rowid DESC LIMIT ?', (limit,))
|
||||
return [dict(r) for r in rows]
|
||||
|
||||
|
||||
# ---------- users(只读统计;写操作请走 auth_system 的事务口径) ----------
|
||||
|
||||
async def count_users(role: str | None = None) -> int:
|
||||
if role:
|
||||
return int(await db.fetch_val('SELECT COUNT(*) FROM users WHERE role = ?', (role,), 0))
|
||||
return int(await db.fetch_val('SELECT COUNT(*) FROM users', (), 0))
|
||||
|
||||
|
||||
async def get_user_by_name(username: str) -> dict[str, Any] | None:
|
||||
row = await db.fetch_one('SELECT * FROM users WHERE username = ?', (username,))
|
||||
return dict(row) if row is not None else None
|
||||
|
||||
|
||||
# ---------- billing ledger(只读对账) ----------
|
||||
|
||||
async def ledger_open_count() -> int:
|
||||
return int(await db.fetch_val(
|
||||
"SELECT COUNT(*) FROM billing_ledger WHERE status = 'open'", (), 0))
|
||||
Reference in new issue
Block a user