如何用Gorm Gen生成带关联的ORM并实现Go项目数据库测试

一、自动生成带关联的Gorm结构体（适配15张表场景） #

Gorm Gen默认不会自动识别数据库外键生成关联字段，针对多表场景，推荐以下两种方案：

方案1：扩展Gorm Gen自动识别关联 #

通过读取MySQL元数据（information_schema）提取关联关系，动态构建Gen的生成配置，避免手动逐个定义。

步骤：

1. 编写辅助逻辑提取关联：
   查询information_schema.KEY_COLUMN_USAGE获取外键信息，识别多对多中间表（仅含两个外键且为主键的表）、一对多关系。
2. 动态生成Gen配置：

   package main
   
   import (
     "gorm.io/driver/mysql"
     "gorm.io/gorm"
     "gorm.io/gen"
     "gorm.io/gen/field"
   )
   
   func main() {
     dsn := "user:pass@tcp(127.0.0.1:3306)/your_db?charset=utf8mb4&parseTime=True&loc=Local"
     db, _ := gorm.Open(mysql.Open(dsn), &gorm.Config{})
   
     g := gen.NewGenerator(gen.Config{
       OutPath: "./dal/model",
       Mode:    gen.WithDefaultQuery, // 关闭冗余查询接口，只生成结构体
     })
     g.UseDB(db)
   
     // 示例：手动配置多对多关联（可通过循环替换为自动处理所有表）
     g.GenerateModel("user",
       gen.FieldRelate(field.HasMany, "Accounts", "account", &field.RelateConfig{
         JoinTable: "user_account",
       }),
       gen.FieldTag("ID", `json:"id" gorm:"primary_key"`),
     )
   
     g.GenerateModel("account",
       gen.FieldRelate(field.HasMany, "Users", "user", &field.RelateConfig{
         JoinTable: "user_account",
       }),
       gen.FieldTag("ID", `json:"id" gorm:"primary_key"`),
       gen.FieldTag("Name", `json:"name"`),
     )
   
     // 生成剩余表的基础结构体（带Gorm/JSON标签）
     g.GenerateAllTable()
     g.Execute()
   }
   

   若需自动处理15张表，可编写逻辑遍历所有外键记录，批量生成FieldRelate配置。

方案2：脚本补全关联字段 #

先用Gorm Gen生成基础结构体，再用Go/Python脚本读取数据库元数据，自动给结构体文件追加关联字段和标签。例如：

• 识别user_account中间表后，在User结构体中添加：

  Accounts []Account `json:"accounts" gorm:"many2many:user_account"`
  

• 脚本可通过正则匹配结构体位置，自动插入字段，避免手动重复操作。

二、Docker Compose MySQL的功能测试流程实现 #

1. 测试环境配置 #

创建docker-compose.test.yml定义测试用MySQL：

version: '3.8'
services:
  test-mysql:
    image: mysql:8.0
    environment:
      MYSQL_ROOT_PASSWORD: testpass
      MYSQL_DATABASE: test_db
    ports:
      - "3307:3306"
    healthcheck:
      test: ["CMD", "mysqladmin", "ping", "-h", "localhost"]
      interval: 5s
      timeout: 5s
      retries: 3

2. 测试生命周期管理 #

在TestMain中统一处理测试前准备、测试执行、测试后清理：

package test

import (
  "os"
  "testing"
  "gorm.io/driver/mysql"
  "gorm.io/gorm"
  "your-project/dal/model"
)

var testDB *gorm.DB

func TestMain(m *testing.M) {
  // 连接测试MySQL容器
  dsn := "root:testpass@tcp(127.0.0.1:3307)/test_db?charset=utf8mb4&parseTime=True&loc=Local"
  var err error
  testDB, err = gorm.Open(mysql.Open(dsn), &gorm.Config{})
  if err != nil {
    panic("连接测试数据库失败")
  }

  // 测试前：创建所有表
  err = testDB.AutoMigrate(
    &model.User{}, &model.Account{},
    // 追加其他13个表结构体
  )
  if err != nil {
    panic("表迁移失败")
  }

  // 运行所有测试用例
  exitCode := m.Run()

  // 测试后：清理数据（可选：删除表或截断）
  err = testDB.Migrator().DropTable(
    &model.User{}, &model.Account{},
    // 追加其他表
  )
  if err != nil {
    panic("清理测试表失败")
  }

  os.Exit(exitCode)
}

3. 高效数据清理：事务回滚 #

若删除表耗时过长，可在单个测试用例中使用事务回滚，自动清理测试数据：

func TestUserAccountPreload(t *testing.T) {
  tx := testDB.Begin()
  defer tx.Rollback() // 测试结束自动回滚

  // 插入测试数据
  user := model.User{}
  tx.Create(&user)
  account := model.Account{Name: "测试账户"}
  tx.Create(&account)
  tx.Model(&user).Association("Accounts").Append(&account)

  // 验证预加载
  var fetchedUser model.User
  tx.Preload("Accounts").First(&fetchedUser, user.ID)
  if len(fetchedUser.Accounts) != 1 {
    t.Fatalf("预期关联1个账户，实际得到%d个", len(fetchedUser.Accounts))
  }
}

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