在 Gorm 的 AfterFind() 钩子中获取智能 Select struct 的值

发布于10月13日

我有一个模型:

 type User struct {
      ID     uint
      Name   string
      Age    int
      Gender string
      // hundreds of fields
 }

使用此挂钩:

func (m *User) AfterFind(tx *gorm.DB) (err error) {
  // I don't know how to get Gender value of APIUser struct from here
  return
}

和这个聪明的 Select struct :

type APIUser struct {
  ID   uint
  Gender string
}

然后，我运行以下查询:

DB.Model(&User{}).Find(&APIUser{}, id)

在AfterFind()钩子中，我想在将从数据库检索到的数据发送到客户端之前对其进行操作，但我无法在该钩子中获得值.如何做到这一点呢？

推荐答案

AfterFind应同时设置为APIUser和User

以下是示例代码

type User struct {
    Id       uint64
    Avatar   string
    Nickname string
    Password string
}

func (user *User) AfterFind(*gorm.DB) error {
    return nil
}

type UserSimple struct {
    Id     uint64
    Avatar string
}

func (v *UserSimple) AfterFind(*gorm.DB) error {
    v.Avatar = "prefix/" + v.Avatar
    return nil
}

us := &UserSimple{}
db.Model(&User{}).Where("id = ?", 123).Find(us)

输出