自动在 gorm 中插入哈希编码数据

Question

我有一个表示我的用户数据的表。有一个表示电话号码的字段，我想在更新或插入时自动将其哈希存储在数据库中。所以我的模型是：type Users struct {    gorm.Model    ID          uint          `gorm:"autoIncrement;unique" json:"id"`    PhoneNumber string        `json:"phone_number"`    HashID      string        `gorm:"primaryKey" json:"hash_id"`    Name        string        `gorm:"default:dear user" json:"name"`    Rank        uint          `json:"rank"`    Score       uint          `json:"score"`    Image       string        `json:"image"`    Email       string        `json:"email"`    Address     string        `json:"address"`    Birthday    string        `json:"birthday"`    Biography   string        `json:"biography"}在插入或更新数据时，如何告诉 GORM 使用“电话号码”列的 sha256 哈希代码填充 HashID 列？

Answer 1

你需要这样的东西：

package main

import (

    "crypto/sha256"

    "fmt"

    "gorm.io/driver/sqlite"

    "gorm.io/gorm"

)

type Users struct {

    gorm.Model

    Key  string `json:"phone_number"`

    Hash string `gorm:"primaryKey" json:"hash_id"`

}

func (u *Users) BeforeCreate(tx *gorm.DB) (err error) {

    h := sha256.Sum256([]byte(u.Key))

    u.Hash = fmt.Sprintf("%x", h[:])

    return nil

}

func (u *Users) BeforeSave(tx *gorm.DB) (err error) {

    h := sha256.Sum256([]byte(u.Key))

    u.Hash = fmt.Sprintf("%x", h[:])

    return nil

}

func main() {

    db, err := gorm.Open(sqlite.Open("test.db"), &gorm.Config{})

    if err != nil {

        panic("failed to connect database")

    }

    db.AutoMigrate(&Users{})

    u := Users{Key: "123"}

    db.Create(&u)

}

检查 https://gorm.io/docs/index.html