GORM预加载用户关联工具时如何获取中间表的quality字段

问题根因 #

• 现有表结构中quality是用户与工具的关联级属性，存储在中间表user_tools中，并非工具本身的固有属性，tools表本身不存在该字段。
• 原模型中ToolsInBackpack使用GORM默认的裸多对多关联，默认实现只会拉取关联双方主表的字段，不会读取中间表的自定义额外字段，因此映射不到quality的有效值。
• 原MainTool关联仅通过users.main_tool_id直接关联tools表，没有关联查询user_tools表的记录，自然无法获取当前用户持有该主工具对应的quality值。

实现步骤 #

1. 修正模型定义 #

先定义对应中间表的结构体，明确关联关系，规避GORM默认多对多跳过中间表字段的问题：

// 对应user_tools 中间表
type UserTool struct {
    ID      int64  `gorm:"primaryKey"`
    UserID  int64  `gorm:"uniqueIndex:idx_user_tool"`
    ToolID  int64  `gorm:"uniqueIndex:idx_user_tool"`
    Quality string `gorm:"type:varchar(10)"`
    // 关联预载用，关联工具主表
    Tool    Tool   `gorm:"foreignKey:ToolID"`
}

type User struct {
    Id         int64  `json:"-"`
    Name       string `json:"name"`
    MainToolId int64  `json:"-"`

    // 内部关联字段，不直接输出到JSON
    MainTool  Tool       `json:"-" gorm:"foreignKey:MainToolId"`
    UserTools []UserTool `json:"-" gorm:"foreignKey:UserID"`

    // 对外输出字段，对齐原有期望的JSON结构
    MainToolOutput      MainToolVO       `json:"main_tool"`
    BackpackToolsOutput []BackpackToolVO `json:"tools_in_backpack"`
}

// 主工具返回结构
type MainToolVO struct {
    Name    string `json:"name"`
    Quality string `json:"quality"`
}

// 背包工具返回结构
type BackpackToolVO struct {
    Name    string `json:"name"`
    Quality string `json:"quality"`
}

type Tool struct {
    Id   int64  `json:"-"`
    Name string `json:"name"`
}

注意：原Tool结构体上的Quality字段需要移除，因为该属性不属于工具主表，否则GORM查询时会因为找不到对应字段触发映射错误。

2. 修正查询逻辑 #

使用预加载拉取所有关联数据后，组装为期望的返回结构，替换原有的查询代码：

var users []User
// 预载主工具的基础信息
// 预载用户的所有关联工具记录（含中间表的quality），同时预载每条关联记录对应的工具主表信息
err := DB.
    Preload("MainTool").
    Preload("UserTools.Tool").
    Find(&users).Error
if err != nil {
    // 自行补充错误处理逻辑
}

// 遍历组装输出数据
for i := range users {
    u := &users[i]

    // 组装主工具返回值：匹配用户关联工具中与主工具ID一致的记录，获取对应quality
    var mainToolQuality string
    for _, ut := range u.UserTools {
        if ut.ToolID == u.MainToolId {
            mainToolQuality = ut.Quality
            break
        }
    }
    u.MainToolOutput = MainToolVO{
        Name:    u.MainTool.Name,
        Quality: mainToolQuality,
    }

    // 组装背包工具返回值：遍历所有用户关联工具，拼接工具名称和对应quality
    for _, ut := range u.UserTools {
        u.BackpackToolsOutput = append(u.BackpackToolsOutput, BackpackToolVO{
            Name:    ut.Tool.Name,
            Quality: ut.Quality,
        })
    }
}

效果说明 #

执行上述代码后，返回的JSON结构会与期望完全一致：主工具和背包工具的quality值都会正确从user_tools表读取，不会出现空值问题。

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