70 lines
1.7 KiB
Go
70 lines
1.7 KiB
Go
package mixin
|
|
|
|
import (
|
|
"context"
|
|
"time"
|
|
|
|
"entgo.io/ent"
|
|
"entgo.io/ent/schema/field"
|
|
"entgo.io/ent/schema/mixin"
|
|
)
|
|
|
|
// TimeFormat ISO 8601 时间格式
|
|
const TimeFormat = time.RFC3339
|
|
|
|
// NowString 返回当前时间的 ISO 8601 格式字符串
|
|
func NowString() string {
|
|
return time.Now().Format(TimeFormat)
|
|
}
|
|
|
|
// skipAutoUpdateKey context key for skipping auto update
|
|
type skipAutoUpdateKey struct{}
|
|
|
|
// SkipAutoUpdate 返回跳过自动更新 updated_at 的 context
|
|
func SkipAutoUpdate(ctx context.Context) context.Context {
|
|
return context.WithValue(ctx, skipAutoUpdateKey{}, true)
|
|
}
|
|
|
|
// TimeMixin 时间字段混入
|
|
// created_at: 创建时间
|
|
// updated_at: 更新时间（自动更新）
|
|
type TimeMixin struct {
|
|
mixin.Schema
|
|
}
|
|
|
|
// Fields of the TimeMixin.
|
|
func (TimeMixin) Fields() []ent.Field {
|
|
return []ent.Field{
|
|
field.String("created_at").
|
|
DefaultFunc(NowString).
|
|
Immutable().
|
|
StructTag(`json:"created_at"`).
|
|
Comment("creation time"),
|
|
field.String("updated_at").
|
|
DefaultFunc(NowString).
|
|
StructTag(`json:"updated_at"`).
|
|
Comment("update time"),
|
|
}
|
|
}
|
|
|
|
// Hooks of the TimeMixin.
|
|
func (TimeMixin) Hooks() []ent.Hook {
|
|
return []ent.Hook{
|
|
func(next ent.Mutator) ent.Mutator {
|
|
return ent.MutateFunc(func(ctx context.Context, m ent.Mutation) (ent.Value, error) {
|
|
// 跳过自动更新（用于同步导入场景）
|
|
if ctx.Value(skipAutoUpdateKey{}) != nil {
|
|
return next.Mutate(ctx, m)
|
|
}
|
|
// 只在更新操作时设置 updated_at
|
|
if m.Op().Is(ent.OpUpdate | ent.OpUpdateOne) {
|
|
if setter, ok := m.(interface{ SetUpdatedAt(string) }); ok {
|
|
setter.SetUpdatedAt(NowString())
|
|
}
|
|
}
|
|
return next.Mutate(ctx, m)
|
|
})
|
|
},
|
|
}
|
|
}
|