6.2 KiB
database
结构化 MySQL 配置、独立 TLS/mTLS、驱动日志和生成项目接入见 官方 MySQL 适配器。已有 Open/OpenConnector 接口继续支持。
使用 GORM 的项目可显式接入 gormx 可选适配器,复用本包连接池,并将 SQL 日志接入 logx;普通 database 包不导入 GORM。
database 包提供基于标准库 database/sql 的连接池生命周期,不绑定 ORM 或驱动:
Config:驱动、DSN 与连接池参数;嵌入项目配置,通过文件显式引用变量。Open/OpenConnector:通过 DSN 或标准 Connector 创建连接池、设置连接上限并在返回前执行PingContext。Health:用于就绪检查,等待连接和 Ping 均受PingTimeout限制。InTx:统一事务提交和回滚;工作函数返回错误或 panic 时回滚。Close:由嵌入的*sql.DB提供,应用退出时显式调用。
框架不导入 MySQL、PostgreSQL 等具体驱动。应用负责注册驱动:
import (
"context"
"git.yuntuo.hk/k2safe/go-omni/config"
_ "github.com/go-sql-driver/mysql"
"git.yuntuo.hk/k2safe/go-omni/database"
)
cfg, err := config.LoadProject("etc/database.yaml", database.DefaultConfig())
if err != nil {
return err
}
db, err := database.Open(context.Background(), cfg)
if err != nil {
return err
}
defer db.Close()
对应文件(框架不规定变量名):
driver: mysql
dsn: "${DB_DSN}"
maxOpenConns: "${DB_MAX_OPEN:-10}"
Open 使用的 Driver、DSN 必填。建库、迁移、表结构和业务 Repository 由项目负责。v0.22 删除 ConfigFromEnv/ApplyEnv;不再隐式扫描数据库变量。
MaxIdleConns=0 会禁用空闲连接。初始化 Ping 失败会关闭已创建的连接池。
go-omni new --database mysql 会生成项目自己的 MySQL Repository、ServiceContext、
数据库就绪探针和 schema.sql。核心包不需要知道业务表结构。
真实数据库验收(需要本地 Docker):
python3 scripts/verifyDatabase.py --version v0.2.0
脚本仅操作本次创建的独立容器与数据库,验证 HTTP CRUD、服务重启持久化、事务提交与
错误/panic/取消回滚、连接池关闭、缺少表结构时启动失败,以及停库后的 readiness 503 和恢复后的 200。
结束时清理容器、匿名数据卷和临时项目。常规 go test ./... 不依赖 Docker。
项目配置
配置结构作为项目 Config 的字段,Default() 初始化连接池,文件按需覆盖。DSN 可使用 dsn: "${DB_DSN}";database.Config 的 JSON 序列化不输出 DSN 原值。不要打印原始配置对象。当前 API 脚手架使用结构化 MySQL 适配器,由 svc 统一初始化与关闭。
Connector 接入(v0.9.1-preview)
项目已有 database/sql/driver.Connector 时使用 OpenConnector。这是标准驱动接口,核心包不增加具体驱动依赖。项目可以在同一进程创建多个使用不同证书、日志、拨号策略的连接池。
下面是 MySQL 的完整组装函数;连接地址、凭据、TLS 和驱动 Logger 都从调用方传入,不在框架中约定:
package connection
import (
"context"
"crypto/tls"
"github.com/go-sql-driver/mysql"
"git.yuntuo.hk/k2safe/go-omni/database"
)
func Open(ctx context.Context, address, user, password, name string,
tlsConfig *tls.Config, logger mysql.Logger,
pool database.PoolConfig) (*database.DB, error) {
cfg := mysql.NewConfig()
cfg.Net, cfg.Addr = "tcp", address
cfg.User, cfg.Passwd, cfg.DBName = user, password, name
cfg.TLS = tlsConfig // 项目设置 RootCAs、ServerName、MinVersion 等
if logger != nil {
cfg.Logger = logger // 项目现有的 Print(...any) 日志适配器
}
connector, err := mysql.NewConnector(cfg)
if err != nil {
return nil, err
}
return database.OpenConnector(ctx, connector, pool)
}
调用时以 database.DefaultPoolConfig() 为起点,按项目需要调整。已有 database.Config 可调用 .PoolConfig() 提取池设置;Connector 不需要填写 DSN。PoolConfig 支持通过 config.DecodeFile 加载项目的 JSON/YAML 嵌套配置,环境引用由项目在文件里显式声明。
| PoolConfig 字段 | 默认值 / 含义 |
|---|---|
MaxOpenConns |
10;0 表示不限制 |
MaxIdleConns |
5;0 禁止保留空闲连接 |
ConnMaxLifetime |
30m;0 不因连接年龄回收 |
ConnMaxIdleTime |
5m;0 不因空闲时长回收 |
PingTimeout |
5s;必须为正,约束启动 Ping 和 Health 等待 |
返回的 *database.DB 嵌入 *sql.DB,可使用 QueryContext、ExecContext、Stats,以及框架的 Health、InTx。需要标准数据库指针的 ORM/迁移器可取 db.DB;不要重复建立池。项目负责驱动支持的 context 取消与读写超时设置,框架不能强制中断不遵守 context 的自定义驱动。
生命周期: nil/带类型的 nil Connector、非法池参数、启动前已取消的 context 会在创建池前返回,Connector 仍归调用方。进入 sql.OpenDB 后由池持有;启动 Ping 失败自动关闭,成功后由调用方负责 db.Close,通常注册 app.OnShutdown(db.Close)。若 Connector 实现 io.Closer,遵循标准 database/sql 规则随池关闭;有资源的 Connector 不应跨多个池共享。资源初始化过程中、注册退出回调之前若业务校验失败,项目也须关闭池。
db.PoolConfig() 返回池配置快照。Connector 入口的 db.Config() 中 Driver / DSN 为空,不能用它重建 Connector。TLS、Logger、驱动版本检查、字符集、时区以及迁移规则保持由项目负责;不写入框架配置快照。
验收脚本使用独立本地 MySQL 8.4 容器(1 CPU / 768 MiB),自动生成测试 CA/证书与随机凭据,检查 TLS 信任/主机名、两池配置隔离、驱动日志、事务和 HTTP 超时,结束后清理本次资源:
python3 scripts/verifyConnector.py --local --report performance/connectorLocal.json
# 发布后通过全新 module cache 验证远程依赖,无本地 replace:
python3 scripts/verifyConnector.py --version v0.9.1-preview --report performance/connectorRemote.json