本文介绍在 revel + gorm 架构下，通过初始化多个数据库连接、利用自定义 filter 动态路由请求到对应分片库，并借助 gorm 内置连接池实现高效、安全的数据库分片（sharding）方案。
本文介绍在 revel + gorm 架构下，通过初始化多个数据库连接、利用自定义 filter 动态路由请求到对应分片库，并借助 gorm 内置连接池实现高效、安全的数据库分片（sharding）方案。
在 Revel 应用中集成数据库分片，核心在于连接预置、上下文传递与模型适配三步。GORM 本身不直接支持运行时切换全局 DB 实例，因此需避免修改 gorm.DB 全局单例，转而采用“按请求绑定连接”的方式——即在请求生命周期内将选定的 *gorm.DB 实例注入上下文，并由模型层统一消费。
1. 启动时预建多连接池（推荐使用 GORM v2+）
GORM v2 自动管理连接池，无需手动封装 sql.DB。在 app/init.go 的 Init() 函数中完成初始化：
import (
"github.com/revel/revel"
"gorm.io/gorm"
"gorm.io/driver/mysql" // 或 postgresql/sqlite
)
var (
DBMaster *gorm.DB
DBShard1 *gorm.DB
DBShard2 *gorm.DB
)
func init() {
revel.InterceptFunc(dbFilter, revel.BEFORE)
revel.OnAppStart(func() {
var err error
// 主库（如用户元数据）
DBMaster, err = gorm.Open(mysql.Open("user:pass@tcp(10.0.0.1:3306)/main?parseTime=true"), &gorm.Config{})
if err != nil {
panic("failed to connect master DB: " + err.Error())
}
// 分片库 1（如 tenant_id % 3 == 0）
DBShard1, err = gorm.Open(mysql.Open("user:pass@tcp(10.0.0.2:3306)/shard1?parseTime=true"), &gorm.Config{})
if err != nil {
panic("failed to connect shard1 DB: " + err.Error())
}
// 分片库 2（如 tenant_id % 3 == 1）
DBShard2, err = gorm.Open(mysql.Open("user:pass@tcp(10.0.0.3:3306)/shard2?parseTime=true"), &gorm.Config{})
if err != nil {
panic("failed to connect shard2 DB: " + err.Error())
}
// 可选：为各连接配置连接池（GORM v2 默认已启用，此处仅作显式调优）
sqlDB, _ := DBMaster.DB()
sqlDB.SetMaxOpenConns(100)
sqlDB.SetMaxIdleConns(20)
})
}
✅ 注意：GORM v2 中 *gorm.DB 是线程安全的，可安全复用；每个实例自带独立连接池，无需额外封装。
2. 在 Security Filter 中解析租户/分片标识
假设你在 app/filter/security.go 的自定义 filter 中已从 JWT 或 Header 提取 tenant_id 或 shard_key，并存入 c.Params 或 c.Session：
// 示例：从 Authorization header 解析 tenant_id
func securityFilter(c *revel.Controller, fc []revel.Filter) {
auth := c.Request.Header.Get("Authorization")
if tenantID, ok := parseTenantFromJWT(auth); ok {
c.Params.Set("tenant_id", tenantID) // 或 c.Session["tenant_id"] = tenantID
}
revel.RunFilterChain(c, fc)
}
3. 在 DbFilter 中动态选择并绑定 DB 实例
在 app/filter/db.go 中编写 dbFilter，根据 tenant_id 路由至对应 DB，并将其注入 Controller 上下文（推荐使用 c.RenderArgs 或自定义字段）：
func dbFilter(c *revel.Controller, fc []revel.Filter) {
tenantID := c.Params.Get("tenant_id")
var db *gorm.DB
switch tenantID % 3 {
case 0:
db = DBMaster
case 1:
db = DBShard1
case 2:
db = DBShard2
default:
db = DBMaster // fallback
}
// 方案一：挂载到 Controller（需在 model 中显式访问）
c.RenderArgs["db"] = db
// 方案二（更推荐）：扩展 Controller 结构体，添加 DB 字段（见下文）
revel.RunFilterChain(c, fc)
}
4. 模型层统一使用当前请求绑定的 DB
为保持模型简洁性，建议在 app/models/base.go 中定义通用基类：
type Model struct {
DB *gorm.DB `revel:"-"`
}
func (m *Model) WithDB(db *gorm.DB) *Model {
m.DB = db
return m
}
// 示例：User 模型
type User struct {
ID uint `gorm:"primaryKey"`
Name string `gorm:"size:100"`
TenantID uint `gorm:"index"`
}
func (u *User) Create(db *gorm.DB) error {
return db.Create(u).Error
}
func (u *User) FindByID(db *gorm.DB, id uint) error {
return db.First(u, id).Error
}
控制器中调用示例：
func (c App) Index() revel.Result {
db, ok := c.RenderArgs["db"].(*gorm.DB)
if !ok {
return c.RenderError("no database bound")
}
user := &models.User{TenantID: 123}
if err := user.Create(db); err != nil {
return c.RenderError(err.Error())
}
return c.RenderJSON(user)
}
⚠️ 关键注意事项
- *禁止共享 `gorm.DB实例跨请求写状态**：GORM 实例本身无状态，但若在其中调用Session()、Table()` 等链式方法生成新实例，需确保不缓存这些临时对象。
- 事务需严格限定在单个 DB 内：跨分片事务不可行，业务层须保证操作原子性约束落在同一分片。
- Filter 执行顺序至关重要：确保 securityFilter 在 dbFilter 之前执行（通过 revel.InterceptFunc(..., revel.BEFORE) 注册顺序控制）。
- 连接健康检查：生产环境建议在 OnAppStart 后增加 db.Exec("SELECT 1").Error 健康探针。
通过以上设计，你可在 Revel 中优雅实现多租户/分片场景下的数据库动态路由，兼顾性能、可维护性与 GORM 最佳实践。
