fix bug

wubin1989 · Dec 17, 2023 · 2cadfbe · 2cadfbe
1 parent 77801e6
commit 2cadfbe
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 3 deletions.
diff --git a/toolkit/caches/caches.go b/toolkit/caches/caches.go
@@ -186,8 +186,25 @@ func getTablesPostgres(db *gorm.DB) []string {
 	sql := db.Statement.SQL.String()
 	w := &walk.AstWalker{
 		Fn: func(ctx interface{}, node interface{}) (stop bool) {
-			if tableName, ok := node.(*tree.TableName); ok {
-				tableNames = append(tableNames, tableName.Table())
+			//log.Printf("%T", node)
+			switch expr := node.(type) {
+			case *tree.TableName:
+				tableNames = append(tableNames, expr.Table())
+			case *tree.Insert:
+				fmtCtx := tree.NewFmtCtx(tree.FmtSimple)
+				expr.Table.Format(fmtCtx)
+				tableName := fmtCtx.String()
+				tableNames = append(tableNames, tableName)
+			case *tree.Update:
+				fmtCtx := tree.NewFmtCtx(tree.FmtSimple)
+				expr.Table.Format(fmtCtx)
+				tableName := fmtCtx.String()
+				tableNames = append(tableNames, tableName)
+			case *tree.Delete:
+				fmtCtx := tree.NewFmtCtx(tree.FmtSimple)
+				expr.Table.Format(fmtCtx)
+				tableName := fmtCtx.String()
+				tableNames = append(tableNames, tableName)
 			}
 			return false
 		},

diff --git a/version/version.go b/version/version.go
@@ -1,3 +1,3 @@
 package version
 
-const Release = "v2.2.0"
+const Release = "v2.2.1"