From 3ca827d8b1f4823284a96be84bfb1fe26a02f172 Mon Sep 17 00:00:00 2001
From: godcong <jumbycc@163.com>
Date: Fri, 29 Nov 2024 19:39:24 +0800
Subject: [PATCH] refactor(ent): split created/updated/deleted schema and
 improve indexing

- Split CreatedSchema into CreateSchema and UpdateSchema
- Rename UpdatedSchema to UpdateSchema
- Rename DeletedSchema to DeleteSchema
- Update CreateUpdateSchema to include fields and indexes from both CreateSchema and UpdateSchema
- Add Immutable() constraint to UniqueString method
---
 dbo/ent/mixin/mixin.go  | 47 +++++++++++++++++++++++++++++------------
 dbo/ent/mixin/schema.go |  3 ++-
 2 files changed, 36 insertions(+), 14 deletions(-)

diff --git a/dbo/ent/mixin/mixin.go b/dbo/ent/mixin/mixin.go
index 3dc418e..3a5ae17 100644
--- a/dbo/ent/mixin/mixin.go
+++ b/dbo/ent/mixin/mixin.go
@@ -52,13 +52,34 @@ func (Audit) Indexes() []ent.Index {
 	}
 }
 
-// CreatedSchema schema to include control and time fields.
-type CreatedSchema struct {
+// CreateUpdateSchema schema to include control and time fields.
+type CreateUpdateSchema struct {
 	mixin.Schema
 }
 
 // Fields of the mixin.
-func (CreatedSchema) Fields() []ent.Field {
+func (CreateUpdateSchema) Fields() []ent.Field {
+	return append(
+		CreateSchema{}.Fields(),
+		UpdateSchema{}.Fields()...,
+	)
+}
+
+// Indexes of the mixin.
+func (CreateUpdateSchema) Indexes() []ent.Index {
+	return append(
+		CreateSchema{}.Indexes(),
+		UpdateSchema{}.Indexes()...,
+	)
+}
+
+// CreateSchema schema to include control and time fields.
+type CreateSchema struct {
+	mixin.Schema
+}
+
+// Fields of the mixin.
+func (CreateSchema) Fields() []ent.Field {
 	return []ent.Field{
 		field.Time("created_time").
 			Default(time.Now).
@@ -67,19 +88,19 @@ func (CreatedSchema) Fields() []ent.Field {
 }
 
 // Indexes of the mixin.
-func (CreatedSchema) Indexes() []ent.Index {
+func (CreateSchema) Indexes() []ent.Index {
 	return []ent.Index{
 		index.Fields("created_time"),
 	}
 }
 
-// UpdatedSchema schema to include control and time fields.
-type UpdatedSchema struct {
+// UpdateSchema schema to include control and time fields.
+type UpdateSchema struct {
 	mixin.Schema
 }
 
 // Fields of the mixin.
-func (UpdatedSchema) Fields() []ent.Field {
+func (UpdateSchema) Fields() []ent.Field {
 	return []ent.Field{
 		field.Time("update_time").
 			Default(time.Now).
@@ -88,19 +109,19 @@ func (UpdatedSchema) Fields() []ent.Field {
 }
 
 // Indexes of the mixin.
-func (UpdatedSchema) Indexes() []ent.Index {
+func (UpdateSchema) Indexes() []ent.Index {
 	return []ent.Index{
 		index.Fields("update_time"),
 	}
 }
 
-// DeletedSchema schema to include control and time fields.
-type DeletedSchema struct {
+// DeleteSchema schema to include control and time fields.
+type DeleteSchema struct {
 	mixin.Schema
 }
 
 // Fields of the Model.
-func (DeletedSchema) Fields() []ent.Field {
+func (DeleteSchema) Fields() []ent.Field {
 	return []ent.Field{
 		field.Time("delete_time").
 			Optional().
@@ -109,11 +130,11 @@ func (DeletedSchema) Fields() []ent.Field {
 }
 
 // Indexes of the mixin.
-func (DeletedSchema) Indexes() []ent.Index {
+func (DeleteSchema) Indexes() []ent.Index {
 	return []ent.Index{
 		index.Fields("delete_time"),
 	}
 }
 
 // SoftDeleteSchema schema to include control and time fields.
-type SoftDeleteSchema = DeletedSchema
+type SoftDeleteSchema = DeleteSchema
diff --git a/dbo/ent/mixin/schema.go b/dbo/ent/mixin/schema.go
index fa55fab..3716d3b 100644
--- a/dbo/ent/mixin/schema.go
+++ b/dbo/ent/mixin/schema.go
@@ -35,7 +35,8 @@ func PrimaryID(name string) ent.Field {
 	// Create a unique string field with the given name and maximum length.
 	return field.String(name).
 		MaxLen(36).
-		Unique()
+		Unique().
+		Immutable()
 }
 
 // Time returns a time field with a default value of ZeroTime and a custom schema type for MySQL.