From 6762bea530915e85d5c022541bcb274fb89a4048 Mon Sep 17 00:00:00 2001 From: yanghao05 Date: Fri, 23 May 2025 19:50:54 +0800 Subject: [PATCH] feat: update --- pkg/ast/model/generage.go | 11 ++++++++++- 1 file changed, 10 insertions(+), 1 deletion(-) diff --git a/pkg/ast/model/generage.go b/pkg/ast/model/generage.go index a14cfbe..ef47449 100644 --- a/pkg/ast/model/generage.go +++ b/pkg/ast/model/generage.go @@ -31,7 +31,16 @@ type TableModelParam struct { func Generate(tables []string, transformer Transformer) error { baseDir := "app/model" modelDir := "database/schemas/public/model" - // move database/schemas/public/model files to app/model + tableDir := "database/schemas/public/table" + defer func() { + os.RemoveAll("database/schemas") + }() + + os.RemoveAll("database/table") + // move tableDir to database/table + if err := os.Rename(tableDir, "database/table"); err != nil { + return err + } // remove all files in app/model with ext .gen.go files, err := os.ReadDir(baseDir)