From 5ac14ffaab4ffd73e80c65669539f6c6edb21dcd Mon Sep 17 00:00:00 2001 From: Rogee Date: Thu, 16 Jan 2025 10:01:15 +0800 Subject: [PATCH] feat: support ignore table --- cmd/gen_model.go | 11 +++++++++++ 1 file changed, 11 insertions(+) diff --git a/cmd/gen_model.go b/cmd/gen_model.go index ed3092a..9cefea4 100644 --- a/cmd/gen_model.go +++ b/cmd/gen_model.go @@ -74,6 +74,17 @@ func commandGenModelE(cmd *cobra.Command, args []string) error { UseSchema(func(schema metadata.Schema) template.Schema { return template. DefaultSchema(schema). + UseSQLBuilder( + template.DefaultSQLBuilder().UseTable(func(table metadata.Table) template.TableSQLBuilder { + tbl := template.DefaultTableSQLBuilder(table) + + if lo.Contains(conf.Ignores, table.Name) { + tbl.Skip = true + log.Infof("Skip table %s", table.Name) + } + return tbl + }), + ). UseModel( template. DefaultModel().