package global

import (
	"gorm.io/gorm"
	"time"
)

type BaseModel struct {
	ID        uint           `gorm:"primarykey" json:"ID"` // 主键ID
	CreatedAt time.Time      // 创建时间
	UpdatedAt time.Time      // 更新时间
	DeletedAt gorm.DeletedAt `gorm:"index" json:"-"` // 删除时间
}

// BeforeCreate 定义一个钩子，在创建之前执行自动插入字段
func (model BaseModel) BeforeCreate(db *gorm.DB) (err error) {
	db.Statement.SetColumn("created_at", time.Now())
	db.Statement.SetColumn("updated_at", time.Now())
	return
}

// BeforeUpdate 定义一个钩子，在更新之前执行自动更新字段
func (model BaseModel) BeforeUpdate(db *gorm.DB) (err error) {
	db.Statement.SetColumn("updated_at", time.Now())
	return
}