在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在插入或更新数据时用PhoneNumber列的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