BeforeSave 不填充数据

Question

我有以下代码，一个非常简单的模型 User，我正在尝试更新现有用户的密码；然而，在 BeforeSave 中，我无法填充 u *User 数据，因此第一次创建时，密码被散列，但当我更新它时，密码变成纯文本。知道我在这里做错了什么吗？

package main

import (
    "fmt"

    "golang.org/x/crypto/bcrypt"
    "gorm.io/driver/mysql"
    "gorm.io/gorm"
)

type User struct {
    gorm.Model
    Username string `gorm:"type:varchar(40);unique" json:"username,omitempty"`
    Password string `gorm:"size:255" json:"password,omitempty"`
    NickName string `gorm:"type:varchar(32)" json:"nick_name,omitempty"`
}

func MakePassword(password string) (string, error) {
    bytes, err := bcrypt.GenerateFromPassword([]byte(password), bcrypt.DefaultCost)
    return string(bytes), err
}

// BeforeSave : hook before a user is saved
func (u *User) BeforeSave(tx *gorm.DB) (err error) {
    fmt.Println("================= user: ", u)
    if u.Password != "" {
        hash, err := MakePassword(u.Password)
        if err != nil {
            return nil
        }
        tx.Statement.SetColumn("password", hash)
    }
    return
}

func main() {
    var connectionString = fmt.Sprintf(
        "%s:%s@/%s?charset=utf8&parseTime=True&loc=Local",
        "root", "password", "myproject",
    )
    db, _ := gorm.Open(mysql.Open(connectionString), &gorm.Config{})
    db.AutoMigrate(&User{})
    db.Save(&User{
        Username: "username",
        Password: "123",
    })
    // Up to this point, everything works, password is hashed.
    // the following does not work as I expected

    userMap := make(map[string]interface{})
    userMap["id"] = 1
    userMap["password"] = "new_password"

    db.Model(&User{}).Where(&User{
        Model: gorm.Model{ID: uint(userMap["id"].(int))},
    }).Updates(userMap)
}

Answer 1

问题可能是您的 if 语句：if u.Password != ""。使用地图更新时，我几乎可以肯定会使用空模型，所以 u.Password 将是空的。

我这样做是用 BeforeCreate 和 BeforeUpdate 而不是 BeforeSave:

func (u *User) BeforeCreate(tx *gorm.DB) error {
    return u.bcryptPassword(tx)
}

func (u *User) BeforeUpdate(tx *gorm.DB) error {
    if tx.Statement.Changed("Password") {
        return u.bcryptPassword(tx)
    }

    return nil
}

func (u *User) bcryptPassword(tx *gorm.DB) error {
    var newPass string
    switch u := tx.Statement.Dest.(type) {
    case map[string]interface{}:
        newPass = u["password"].(string)
    case *User:
        newPass = u.Password
    case []*User:
        newPass = u[tx.Statement.CurDestIndex].Password
    }

    b, err := bcrypt.GenerateFromPassword([]byte(newPass), 10)
    if err != nil {
        return err
    }
    tx.Statement.SetColumn("password", b)

    return nil
}

您应该依赖给定的 gorm 交易，而不是您的 u 变量。

注意：此实现也适用于批处理操作和 inserts/updates 使用地图而不是结构。

BeforeSave 不填充数据

BeforeSave does not populate data

go

go-gorm