时间戳

首页 > 解决方案 > BeforeSave 不填充数据

go - BeforeSave 不填充数据

问题描述

我有以下代码，一个非常简单的模型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)
}

标签： gogo-gorm

解决方案

问题可能是您的 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变量。

注意：此实现也适用于批处理操作和使用映射而不是结构的插入/更新。

推荐阅读