gorm代码生成工具如何生成两个模型的双向关系？

Question

比如两个表，user和post:
user hasMany post
post belongTo user

gen工具里面关系配置函数gen.FieldRelate，第三个table参数是个对象。
要生成双向关系变量会互相依赖，只能生成一方的。
怎么解？

Answer 1

package main

import (
    "gorm.io/driver/mysql"
    "gorm.io/gen"
    "gorm.io/gorm"
)

func main() {
    db, _ := gorm.Open(mysql.Open("user:pass@tcp(127.0.0.1:3306)/dbname?charset=utf8mb4&parseTime=True&loc=Local"))
    
    g := gen.NewGenerator(gen.Config{
        OutPath: "./dal/query",
        Mode: gen.WithDefaultQuery,
    })
    
    g.UseDB(db)
    
    user := g.GenerateModelAs("users", "User")
    post := g.GenerateModelAs("posts", "Post")
    
    // 先应用基本模型
    g.ApplyBasic(user, post)
    
    // 然后添加关系
    userWithPosts := g.GenerateModel("users", 
        gen.FieldRelate(field.HasMany, "Posts", post, &field.RelateConfig{
            RelateField: "UserID",
        }),
    )
    
    postWithUser := g.GenerateModel("posts", 
        gen.FieldRelate(field.BelongsTo, "User", user, &field.RelateConfig{
            RelateField: "UserID",
        }),
    )
    
    // 应用带关系的模型
    g.ApplyBasic(userWithPosts, postWithUser)
    
    // 执行生成
    g.Execute()
}

Answer 2

解决方案

1. 使用字符串类型名称代替结构体引用
   在gen.FieldRelate配置中，将关联模型类型用字符串表示（如"Post"、"User"），避免直接引用未生成的结构体。
2. 正确配置外键约束
   在RelateConfig中显式指定foreignKey，确保关联关系的GORM标签正确生成。

// 生成User模型时配置HasMany关系
user := g.GenerateModel("users",
    gen.FieldRelate(model.HasMany, "Posts", "Post", 
        &field.RelateConfig{
            GORMTag: field.GormTag{"foreignKey": []string{"UserID"}},
        }),
)

// 生成Post模型时配置BelongsTo关系
post := g.GenerateModel("posts",
    gen.FieldRelate(model.BelongsTo, "User", "User", 
        &field.RelateConfig{
            GORMTag: field.GormTag{"foreignKey": []string{"UserID"}},
        }),
)

// 将模型加入生成器
g.ApplyBasic(user, post)

关键点说明

• 字符串类型名称：用"Post"和"User"替代结构体类型，解决编译时的循环依赖。
• 外键一致性：两侧均通过foreignKey:UserID指明外键，确保GORM能正确识别关联。
• 生成顺序无关：由于使用字符串引用，无需担心模型生成的先后顺序。

生成结果示例

// User模型
type User struct {
    ID    int64
    Posts []Post `gorm:"foreignKey:UserID"` // HasMany
}

// Post模型
type Post struct {
    ID     int64
    UserID int64
    User   User `gorm:"foreignKey:UserID"` // BelongsTo
}