添加callback,迁移文件
This commit is contained in:
@@ -1,17 +1,9 @@
|
||||
package model
|
||||
|
||||
import "gorm.io/gorm"
|
||||
|
||||
// atlas:ignore
|
||||
//
|
||||
// BaseModel 基础模型,包含通用字段
|
||||
type BaseModel struct {
|
||||
Creator uint `gorm:"not null" json:"creator"`
|
||||
Updater uint `gorm:"not null" json:"updater"`
|
||||
}
|
||||
|
||||
func (b *BaseModel) BeforeCreate(tx *gorm.DB) error {
|
||||
if userID, ok := tx.Statement.Context.Value("current_user_id").(uint); ok {
|
||||
b.Creator = userID
|
||||
b.Updater = userID
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user