diff --git a/database/model/base.go b/database/model/base.go new file mode 100644 index 0000000..1ac7194 --- /dev/null +++ b/database/model/base.go @@ -0,0 +1,28 @@ +package model + +import ( + "time" + + "gorm.io/gorm" +) + +// BaseModel 基础模型,包含自定义的创建时间和更新时间 +type BaseModel struct { + Id int `json:"id" gorm:"column:id;primaryKey;autoIncrement;common:主键ID"` // 主键 ID + CreateTime time.Time `gorm:"column:create_time;type:datetime;not null" json:"created_at"` // 创建时间 + UpdateTime time.Time `gorm:"column:update_time;type:datetime;not null" json:"updated_at"` // 更新时间 +} + +// BeforeCreate 创建前自动设置时间 +func (b *BaseModel) BeforeCreate(tx *gorm.DB) error { + now := time.Now() + b.CreateTime = now + b.UpdateTime = now + return nil +} + +// BeforeUpdate 更新前自动更新时间 +func (b *BaseModel) BeforeUpdate(tx *gorm.DB) error { + b.UpdateTime = time.Now() + return nil +}