🚧 Refactor basic services

internal/models/schema/mixin/time.go

@@ -0,0 +1,57 @@
+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)
+}
+
+// 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) {
+				// 只在更新操作时设置 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)
+			})
+		},
+	}
+}