本文介绍在 revel 框架中基于 gorm 实现多数据库连接池初始化、运行时动态切换连接（如按请求上下文选择分片库），并安全共享至模型层的完整实践方案。
本文介绍在 revel 框架中基于 gorm 实现多数据库连接池初始化、运行时动态切换连接（如按请求上下文选择分片库），并安全共享至模型层的完整实践方案。
在 Revel + GORM 架构中实现数据库分片（Sharding），关键在于连接池预初始化、请求级连接路由与模型层连接注入三者的协同。GORM 内置连接池管理能力，无需手动封装 sql.DB，但需合理组织连接实例生命周期与上下文传递机制。
一、应用启动时预初始化多个 DB 连接池
在 app/init.go 的 init() 或 AppInit() 函数中，一次性创建并配置多个 GORM DB 实例（对应不同分片）。建议使用 gorm.Config 显式启用连接池参数，并设置唯一标识便于后续识别：
import (
"github.com/revel/revel"
"gorm.io/gorm"
"gorm.io/driver/mysql" // 或 postgresql/sqlite
)
var (
DBMain *gorm.DB
DBShard1 *gorm.DB
DBShard2 *gorm.DB
)
func init() {
revel.InterceptFunc(appDBFilter, revel.BEFORE)
// 初始化主库（如用户元数据）
db1, err := gorm.Open(mysql.Open("user:pass@tcp(10.0.0.1:3306)/main?parseTime=true"), &gorm.Config{
PrepareStmt: true,
})
if err != nil {
panic("failed to connect to main DB: " + err.Error())
}
DBMain = db1
// 初始化分片库（如按 tenant_id 或 region 分片）
db2, err := gorm.Open(mysql.Open("user:pass@tcp(10.0.0.2:3306)/shard_1?parseTime=true"), &gorm.Config{
PrepareStmt: true,
})
if err != nil {
panic("failed to connect to shard-1 DB: " + err.Error())
}
DBShard1 = db2
db3, err := gorm.Open(mysql.Open("user:pass@tcp(10.0.0.3:3306)/shard_2?parseTime=true"), &gorm.Config{
PrepareStmt: true,
})
if err != nil {
panic("failed to connect to shard-2 DB: " + err.Error())
}
DBShard2 = db3
}
✅ 注意：GORM 默认已启用连接池（&sql.DB{} 级别），可通过 DB.Exec("SELECT 1").RowsAffected 验证连通性；生产环境务必配置 SetMaxOpenConns() 和 SetMaxIdleConns()（例如 DBMain.DB().SetMaxOpenConns(100)）。
二、通过自定义 Filter 动态绑定当前请求的 DB 实例
在 app/filters.go 中定义 dbFilter，利用 Revel 的 c.Params.Set() 将选中的 DB 实例注入请求上下文。该 Filter 应置于 SECURITY_FILTER 之后（确保 tenant_id、region 等分片键已解析）：
func dbFilter(c *revel.Controller, fc []revel.Filter) {
// 假设 security filter 已将分片标识存入 c.Params
shardKey := c.Params.Get("shard_key") // 如 "shard-1", "shard-2", "main"
var db *gorm.DB
switch shardKey {
case "shard-1":
db = DBShard1
case "shard-2":
db = DBShard2
default:
db = DBMain
}
// 注入到 Controller 上下文，供 Controller 及 Model 使用
c.Params.Set("db", db)
revel.INFO.Printf("DB filter: bound %s to request", shardKey)
fc[0](c, fc[1:])
}
三、在 Controller 与 Model 中安全使用当前 DB 实例
Controller 中可直接获取绑定的 DB 实例：
func (c App) Index() revel.Result {
db := c.Params.Get("db").(*gorm.DB)
var users []User
if err := db.Find(&users).Error; err != nil {
return c.RenderError(err)
}
return c.RenderJSON(users)
}
更推荐的方式是封装通用 Model 方法，避免重复取值：
type User struct {
ID uint `gorm:"primaryKey"`
Name string `gorm:"not null"`
}
func (u *User) Save(db *gorm.DB) error {
return db.Save(u).Error
}
// 在 Controller 中调用：
// db := c.Params.Get("db").(*gorm.DB)
// user := &User{Name: "Alice"}
// user.Save(db)
四、重要注意事项与最佳实践
- ? 线程安全：GORM 实例本身是并发安全的，但 *gorm.DB 是只读配置对象，每次 .Model() 或 .Where() 返回新会话，无需额外同步。
- ? Filter 执行顺序：确保 dbFilter 在 SECURITY_FILTER 之后、ACTION_FILTER 之前注册，否则分片键可能未就绪。
- ? 禁止全局 DB 切换：切勿在运行时调用 gorm.Open() 创建新连接——应严格复用预初始化的连接池。
- ? 事务隔离：跨分片事务需由业务层协调（GORM 不支持分布式事务），建议设计为单分片内强一致性 + 最终一致性补偿。
- ?️ 扩展性增强：可将分片路由逻辑抽象为 ShardRouter 接口，支持按 tenant_id % N、哈希或配置映射等多种策略。
通过以上结构化设计，你可在 Revel 应用中稳健支撑水平分库场景，兼顾性能、可维护性与可观测性。
