如何在golang-revel中实现多数据库的迁移操作

针对你的场景，要实现多数据库迁移，核心是根据路由中的db参数切换对应的GORM数据库连接，再执行迁移逻辑。以下是具体实现步骤和代码修改：

1. 配置多数据库连接 #

首先在项目初始化时，根据不同数据库标识（对应路由里的:db值）初始化并存储对应的GORM连接实例。可以在项目的init.go或专门的数据库配置文件中添加：

package app

import (
  "gorm.io/driver/mysql"
  "gorm.io/gorm"
  "github.com/revel/revel"
)

// 存储多数据库连接，key与路由中的db标识一一对应
var dbConnections = make(map[string]*gorm.DB)

func init() {
  // 示例：初始化两个数据库连接，可根据实际需求扩展
  // 从配置文件读取数据库DSN
  db1DSN, _ := revel.Config.String("db.db1.dsn")
  db1, err := gorm.Open(mysql.Open(db1DSN), &gorm.Config{})
  if err != nil {
    revel.ERROR.Fatal("Failed to connect db1: ", err)
  }
  dbConnections["db1"] = db1

  db2DSN, _ := revel.Config.String("db.db2.dsn")
  db2, err := gorm.Open(mysql.Open(db2DSN), &gorm.Config{})
  if err != nil {
    revel.ERROR.Fatal("Failed to connect db2: ", err)
  }
  dbConnections["db2"] = db2
}

2. 修改迁移入口方法（app.go） #

从路由中提取db参数，匹配对应的数据库连接，再传入迁移逻辑：

func (c App) Migrate() revel.Result {
  dbID := c.Params.Get("db")
  db, exists := dbConnections[dbID]
  if !exists {
    return c.RenderError(fmt.Errorf("数据库 %s 不存在", dbID))
  }

  messages := c.Migration(db, nil)
  return c.Render(messages)
}

3. 调整BaseController的Migration方法（base.go） #

把原来依赖c.Txn的单库逻辑，改为接收传入的多库GORM实例，并替换所有数据库操作的连接：

func (c *BaseController) Migration(db *gorm.DB, tables []interface{}) []string {
  var (
    md       = &m.Model{}
    mh       = &MigrationHelpers{}
    mo       = &m.Module{}
    pr       = &m.Privilege{}
    messages = []string{}
  )

  messages = append(messages, "* Running Auto Migration for database: "+db.Name())

  // 执行表迁移
  if tables == nil {
    for idx, table := range md.Tables() {
      tableIdx := fmt.Sprintf("%d", idx)
      messages = append(messages, "* Migrating Table "+tableIdx)
      if err := db.AutoMigrate(table).Error; err != nil {
        messages = append(messages, err.Error())
      }
    }
  } else {
    db.Set("gorm:table_options", "ENGINE=InnoDB").AutoMigrate(tables)
  }
  
  messages = append(messages, "* SCRIPTS AutoMigration Done!")

  // 注意：需同步修改Module、Privilege的List方法，使其接收传入的db实例
  mh.traverseModule(mo.List(db), c)
  messages = append(messages, "* New MODULES Detection Done!")
  mh.traversePrivilege(pr.List(db), c)
  messages = append(messages, "* New PRIVILEGES Detection Done!")

  for _, message := range messages {
    fmt.Println(message)
  }

  return messages
}

4. 调整关联数据查询方法 #

原来的mo.List(c.Controller)和pr.List(c.Controller)需要修改为接收*gorm.DB参数，确保使用当前迁移的数据库连接：

// 示例：Module.List方法修改
func (m *Module) List(db *gorm.DB) []Module {
  var modules []Module
  db.Find(&modules)
  return modules
}

// 示例：Privilege.List方法修改
func (p *Privilege) List(db *gorm.DB) []Privilege {
  var privileges []Privilege
  db.Find(&privileges)
  return privileges
}

关键注意事项 #

• 路由中的:db参数值必须与dbConnections中的key完全匹配，才能正确找到对应数据库；
• 所有涉及数据库操作的方法都要改为传入指定的DB实例，避免复用默认连接导致操作错误；
• 可根据需求添加数据库连接池配置、健康检查等优化逻辑。

内容的提问来源于stack exchange，提问作者Michael Halim