rogee/atomctl
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: support softdelete in table functions

Browse Source
This commit is contained in:
yanghao05
2025-05-23 22:24:53 +08:00
parent ab36ea0e5d
commit e83332ea6a
2 changed files with 16 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
pkg/ast/model/generage.go
View File

@@ -29,6 +29,7 @@ type TableModelParam struct {
PkgName string PkgName string
CamelTable string // user CamelTable string // user
PascalTable string // User PascalTable string // User
SoftDelete bool
} }
func Generate(tables []string, transformer Transformer) error { func Generate(tables []string, transformer Transformer) error {
@@ -65,6 +66,7 @@ func Generate(tables []string, transformer Transformer) error {
return err return err
} }
modelContent := make(map[string]string)
for _, file := range files { for _, file := range files {
// get filename without ext // get filename without ext
name := strings.TrimSuffix(file.Name(), filepath.Ext(file.Name())) 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 { if err := os.Rename(from, to); err != nil {
return err return err
} }
// read file content
content, err := os.ReadFile(to)
if err != nil {
return err
}
modelContent[name] = string(content)
} }
// remove database/schemas/public/model // remove database/schemas/public/model
@@ -99,6 +108,10 @@ func Generate(tables []string, transformer Transformer) error {
PascalTable: lo.PascalCase(table), PascalTable: lo.PascalCase(table),
PkgName: gomod.GetModuleName(), PkgName: gomod.GetModuleName(),
} }
if strings.Contains(table, "DeletedAt") {
tableInfo.SoftDelete = true
}
items = append(items, tableInfo) items = append(items, tableInfo)
// tableFuncsFile // tableFuncsFile

3
pkg/ast/model/table_funcs.go.tpl
View File

@@ -44,6 +44,8 @@ func (m *{{.PascalTable}}) BatchCreate(ctx context.Context, models []*{{.PascalT
return nil return nil
} }
// if SoftDelete
{{- if .SoftDelete }}
func (m *{{.PascalTable}}) Delete(ctx context.Context) error { 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))) 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") m.log().WithField("func", "BatchSoftDelete").Infof("{{.PascalTable}} items soft deleted successfully")
return nil return nil
} }
{{- end}}
func (m *{{.PascalTable}}) ForceDelete(ctx context.Context) error { func (m *{{.PascalTable}}) ForceDelete(ctx context.Context) error {