yjp 1 жил өмнө
parent
commit
7f1d5ff4e3
1 өөрчлөгдсөн 7 нэмэгдсэн , 9 устгасан
  1. 7 9
      sql_parser.go

+ 7 - 9
sql_parser.go

@@ -109,10 +109,10 @@ func insertWalk(sql string) (*insertClause, error) {
 	w := &walk.AstWalker{
 		Fn: func(ctx interface{}, node interface{}) (stop bool) {
 			realNode := node.(*tree.Insert)
+
 			// 获取table
-			tableFmtCtx := tree.NewFmtCtx(tree.FmtSimple)
-			realNode.Table.Format(tableFmtCtx)
-			clause.table = tableFmtCtx.String()
+			tableName := realNode.Table.(*tree.TableName)
+			clause.table = tableName.Table()
 
 			// 获取table row
 			clause.tableRow = make(map[string]clauseTableRowValue)
@@ -161,9 +161,8 @@ func deleteWalk(sql string) (*deleteClause, error) {
 			realNode := node.(*tree.Delete)
 
 			// 获取table
-			tableFmtCtx := tree.NewFmtCtx(tree.FmtSimple)
-			realNode.Table.Format(tableFmtCtx)
-			clause.table = tableFmtCtx.String()
+			tableName := realNode.Table.(*tree.AliasedTableExpr)
+			clause.table = tableName.String()
 
 			// 获取where
 			clause.where = realNode.Where.Expr.String()
@@ -199,9 +198,8 @@ func updateWalk(sql string) (*updateClause, error) {
 			realNode := node.(*tree.Update)
 
 			// 获取table
-			tableFmtCtx := tree.NewFmtCtx(tree.FmtSimple)
-			realNode.Table.Format(tableFmtCtx)
-			clause.table = tableFmtCtx.String()
+			tableName := realNode.Table.(*tree.AliasedTableExpr)
+			clause.table = tableName.String()
 
 			// 获取where
 			clause.where = realNode.Where.Expr.String()