如何在Gorm的AfterFind()钩子中获取智能查询结构体的值

首先明确：你当前的User.AfterFind钩子在执行DB.Model(&User{}).Find(&APIUser{}, id)时并不会被触发。因为Gorm的钩子是绑定在具体结构体实例上的，这条语句里Gorm直接把数据库查询结果映射到APIUser实例，不会创建User实例，自然不会触发User的钩子。

根据你的需求，有两种可行的实现方式：

方式一：给APIUser添加AfterFind钩子 #

既然最终数据要返回APIUser，直接给它绑定钩子，就能直接操作其字段：

type APIUser struct {
  ID     uint
  Gender string
}

func (m *APIUser) AfterFind(tx *gorm.DB) (err error) {
    // 在这里直接处理Gender字段，比如做格式转换
    switch m.Gender {
    case "M":
        m.Gender = "男性"
    case "F":
        m.Gender = "女性"
    default:
        m.Gender = "未知"
    }
    return
}

执行原查询语句时，Gorm会自动触发APIUser.AfterFind，处理完成后再把APIUser返回给客户端即可。

方式二：先查询User再转换为APIUser #

如果必须在User.AfterFind中处理数据，可以先查询完整的User实例（触发钩子处理字段），再手动映射到APIUser：

// 先查询User，触发AfterFind钩子处理数据
var user User
if err := DB.First(&user, id).Error; err != nil {
    // 处理查询错误
    return err
}

// 将处理后的User字段映射到APIUser
apiUser := APIUser{
    ID:     user.ID,
    Gender: user.Gender, // 这里的Gender已经是User.AfterFind处理后的值
}

// 返回apiUser给客户端

这种方式的缺点是会查询User的所有字段（即使你只需要ID和Gender），如果User字段很多，可能会有性能损耗。

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