From e83332ea6a97838d06e65bd07248955c43f639ba Mon Sep 17 00:00:00 2001
From: yanghao05 <yanghao05@qianxin.com>
Date: Fri, 23 May 2025 22:24:53 +0800
Subject: [PATCH] feat: support softdelete in table functions

---
 pkg/ast/model/generage.go        | 13 +++++++++++++
 pkg/ast/model/table_funcs.go.tpl |  3 +++
 2 files changed, 16 insertions(+)

diff --git a/pkg/ast/model/generage.go b/pkg/ast/model/generage.go
index 763440c..f084c15 100644
--- a/pkg/ast/model/generage.go
+++ b/pkg/ast/model/generage.go
@@ -29,6 +29,7 @@ type TableModelParam struct {
 	PkgName     string
 	CamelTable  string // user
 	PascalTable string // User
+	SoftDelete  bool
 }
 
 func Generate(tables []string, transformer Transformer) error {
@@ -65,6 +66,7 @@ func Generate(tables []string, transformer Transformer) error {
 		return err
 	}
 
+	modelContent := make(map[string]string)
 	for _, file := range files {
 		// get filename without ext
 		name := strings.TrimSuffix(file.Name(), filepath.Ext(file.Name()))
@@ -75,6 +77,13 @@ func Generate(tables []string, transformer Transformer) error {
 		if err := os.Rename(from, to); err != nil {
 			return err
 		}
+
+		// read file content
+		content, err := os.ReadFile(to)
+		if err != nil {
+			return err
+		}
+		modelContent[name] = string(content)
 	}
 
 	// remove database/schemas/public/model
@@ -99,6 +108,10 @@ func Generate(tables []string, transformer Transformer) error {
 			PascalTable: lo.PascalCase(table),
 			PkgName:     gomod.GetModuleName(),
 		}
+		if strings.Contains(table, "DeletedAt") {
+			tableInfo.SoftDelete = true
+		}
+
 		items = append(items, tableInfo)
 
 		// tableFuncsFile
diff --git a/pkg/ast/model/table_funcs.go.tpl b/pkg/ast/model/table_funcs.go.tpl
index b544fac..ad19d2d 100644
--- a/pkg/ast/model/table_funcs.go.tpl
+++ b/pkg/ast/model/table_funcs.go.tpl
@@ -44,6 +44,8 @@ func (m *{{.PascalTable}}) BatchCreate(ctx context.Context, models []*{{.PascalT
 	return nil
 }
 
+// if SoftDelete
+{{- if .SoftDelete }}
 
 func (m *{{.PascalTable}}) Delete(ctx context.Context) error {
 	stmt := table.{{.PascalTable}}.UPDATE().SET(table.{{.PascalTable}}.DeletedAt.Set(TimestampzT(time.Now()))).WHERE(table.{{.PascalTable}}.ID.EQ(Int(m.ID)))
@@ -75,6 +77,7 @@ func (m *{{.PascalTable}}) BatchDelete(ctx context.Context, ids []int64) error {
 	m.log().WithField("func", "BatchSoftDelete").Infof("{{.PascalTable}} items soft deleted successfully")
 	return nil
 }
+{{- end}}
 
 
 func (m *{{.PascalTable}}) ForceDelete(ctx context.Context) error {