Files
micro_servers/db/README.md
T
qing 6692c9c4ee refactor(db): 清理遗留死代码并对齐文档与测试
删除 dbv2 从未挂载的 log_module.py(Flask /log/* 遗留)与纯 TCP 服务
用不到的 security_middleware.py 副本;db/README.md 按 pymsg 实现重写,
requirements 去掉 flask/waitress;黑盒测试对 pymsg 服务改用 TCP 探测,
修复 check_service HTTP 探测导致的 db 测试整体假 skip。
2026-07-28 22:21:36 +08:00

6.9 KiB
Raw Blame History

通用数据库服务 db(pymsg/TCP + SQLAlchemy Core)

一个安全的数据库访问服务。各业务服务通过自研 pymsg 协议(TCP 长度前缀帧 + JSON)发送结构化查询意图,db 翻译为 SQL 执行,全程参数绑定,业务侧无需直连数据库、无需手写 SQL。

功能特点

  • 跨数据库:基于 SQLAlchemy Core 表达式语言,一套接口适配 SQLite / PostgreSQL / MySQL
  • 零 SQL 注入:所有条件、值均通过 SQLAlchemy 参数绑定处理
  • 统一 JSON 接口:_route + action + table + where 模式
  • 支持增删改查:select / insert / update / delete 全覆盖
  • 内置日志表:启动自动建 Logs 表,供全系统集中写日志;带后台自动清理
  • 仅监听本机:默认 127.0.0.1:8005,不对外暴露

快速开始

安装依赖

./_set.sh     # 建 venv 并安装 requirements.txt(SQLAlchemy + 本地 ../pymsg)

启动服务

./_run.sh     # venv 中执行 python3 dbv2.py

服务监听 tcp://127.0.0.1:8005。生产环境由 core(Nidus)按 init.txt 统一拉起,守护靠 systemd。

注意:execute 路由依赖表已存在(通过反射获取结构)。建表见 create_sqlite3.sql(Users / Notes;Logs 表由服务启动时自建)。

协议与 API

传输协议(pymsg)

每条消息为一个完整 JSON,帧格式:[4 字节大端长度][UTF-8 JSON]。 请求中的 _route 字段指定路由,目前仅有一个路由:execute。

execute 路由 — 通用 CRUD

{
  "_route": "execute",
  "action": "select",          // select | insert | update | delete
  "table": "Users",
  "columns": ["id", "name"],   // select 时可选(缺省全列)
  "values": {"name": "Bob"},   // insert / update 时必填
  "where": {                   // 可选,条件见下表
    "age": {"ge": 18, "lt": 60},
    "name": "Alice"
  },
  "order_by": [{"column": "id", "desc": true}],
  "limit": 10,
  "offset": 0
}

where 条件支持的运算符

运算符 说明 示例
直接值 等于 "name": "Alice"
eq 等于 "age": {"eq": 30}
ne 不等于 "status": {"ne": "deleted"}
gt 大于 "score": {"gt": 80}
ge 大于等于 "score": {"ge": 60}
lt 小于 "price": {"lt": 100}
le 小于等于 "price": {"le": 200}
like 模糊匹配 "name": {"like": "%Smith%"}
in 包含于列表 "id": {"in": [1,2,3]}

多个条件自动使用 AND 连接。

响应格式

成功:

{
  "success": true,
  "data": [
    {"id": 1, "username": "Alice"}
  ]
}
  • select → 行对象数组
  • insert → [{"lastrowid": N}]
  • update / delete → [{"rowcount": N}]

失败:

{"success": false, "error": "错误描述"}

使用示例(Python + pymsg)

import json
from pymsg import MsgClient

client = MsgClient("127.0.0.1", 8005, "myservice")

# 查询
resp = client.call(json.dumps({
    "_route": "execute",
    "action": "select",
    "table": "Users",
    "where": {"id": {"gt": 0}},
    "order_by": [{"column": "id", "desc": True}],
    "limit": 5,
}))
print(json.loads(resp))

# 插入
client.call(json.dumps({
    "_route": "execute",
    "action": "insert",
    "table": "Users",
    "values": {"username": "bob", "password_hash": "..."},
}))

安全机制

  • 强制使用 SQLAlchemy 参数绑定,杜绝 SQL 注入。
  • 仅反射已有表,无法创建或删除表(保证结构安全)。
  • 所有异常自动回滚事务。
  • 仅监听 127.0.0.1,仅本机服务可访问。

内置日志能力

db 内置统一日志收集:全系统各服务(waf / gateway / static / dnfs / db / ai_review / captcha / core / smtp)的日志集中写入 db 的 Logs 表。

为什么集成在 db 而不是独立服务

独立 logs 服务会导致循环调用:db 记日志→调 logs,logs 存库→调 db,无限递归。集成在 db 内部后,db 自己记日志直接写表,彻底消除循环依赖。

日志表(启动时自建)

CREATE TABLE IF NOT EXISTS Logs (
    id        INTEGER PRIMARY KEY AUTOINCREMENT,
    ts        TEXT    NOT NULL,
    service   TEXT    NOT NULL,
    level     TEXT    NOT NULL,
    message   TEXT    NOT NULL,
    trace_id  TEXT,
    extra     TEXT
);
-- 索引: (service,ts) / (level,ts) / (ts) / (trace_id)
  • level:DEBUG / INFO / WARNING / ERROR / CRITICAL
  • trace_id:可选,请求追踪 ID
  • extra:可选,JSON 字符串,存放附加结构化字段

写入方式

外部服务:复制 example_log_client.py 为各自目录下的 log_client.py,改 service_name 即用。客户端底层通过 pymsg 调 execute 路由逐条 INSERT 到 Logs 表:

from log_client import logs

logs.info("dnfs", "节点新增成功 id=42")
logs.error("dnfs", "数据库连接失败", extra={"errno": 2002})
logs.push("gateway", "WARNING", "登录失败次数超限",
          trace_id="req-abc", extra={"ip": "1.2.3.4"})

客户端特性:

  • 纯标准库实现,异步非阻塞:内存队列 + 后台线程批量推送
  • fail-safe:db 不可用时静默丢弃,绝不影响业务
  • 缓冲批量:队列满 50 条或超时 2 秒时自动 flush

db 自身:_log_internal() 直接写库,无网络往返(防递归)。

查询日志

日志查询直接走 execute 路由 select Logs 表,或用工具脚本 _view_db.sqlite3.py:

python3 _view_db.sqlite3.py                    # 列出所有表及行数
python3 _view_db.sqlite3.py Logs -n 20         # 最近 20 条日志
python3 _view_db.sqlite3.py Logs -w "level='ERROR'"
python3 _view_db.sqlite3.py -s "SELECT service, COUNT(*) FROM Logs GROUP BY service"

自动清理

  • 服务启动时开启后台守护线程,每 1 小时清理一次
  • 默认保留 30 天日志,超期自动删除
  • 日志总条数超过 100 万时,按时间倒序保留最新 100 万条

切换数据库

只需修改环境变量 DATABASE_URL,并安装对应驱动:

  • SQLite(默认):sqlite:///database.db
  • PostgreSQL:postgresql://user:pass@localhost/dbname(需 pip install psycopg2)
  • MySQL:mysql+pymysql://user:pass@localhost/dbname(需 pip install pymysql)

SQLAlchemy 会自动生成适配各数据库的 SQL 方言,无需修改接口调用。

文件说明

文件 说明
dbv2.py 服务主程序(pymsg/TCP,唯一在跑的实现)
example_log_client.py 日志客户端 SDK 模板,各服务复制为 log_client.py 使用
create_sqlite3.sql Users / Notes / Logs 建表脚本
_view_db.sqlite3.py 非交互式查看 database.db 的工具脚本
_set.sh / _run.sh 环境安装 / 启动脚本