feat: support softdelete in table functions

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

View File

@@ -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