Merge pull request #3047 from actiontech/fix_generate_rollback_panic

Improves rollback SQL generation

Author: winfredLIN

sqle/driver/mysql/rollback.go

@@ -10,6 +10,7 @@ import (
 	"github.com/actiontech/sqle/sqle/errors"
 
 	"github.com/pingcap/parser/ast"
+	"github.com/pingcap/parser/format"
 	_model "github.com/pingcap/parser/model"
 )
 
@@ -415,7 +416,7 @@ func (i *MysqlDriverImpl) generateInsertRollbackSql(stmt *ast.InsertStmt) (strin
 			for n, name := range columnsName {
 				_, isPk := pkColumnsName[name]
 				if isPk {
-					where = append(where, fmt.Sprintf("%s = '%s'", name, util.ExprFormat(value[n])))
+					where = append(where, fmt.Sprintf("%s = '%s'", name, restore(value[n])))
 				}
 			}
 			if len(where) != len(pkColumnsName) {
@@ -437,7 +438,7 @@ func (i *MysqlDriverImpl) generateInsertRollbackSql(stmt *ast.InsertStmt) (strin
 			name := setExpr.Column.Name.String()
 			_, isPk := pkColumnsName[name]
 			if isPk {
-				where = append(where, fmt.Sprintf("%s = '%s'", name, util.ExprFormat(setExpr.Expr)))
+				where = append(where, fmt.Sprintf("%s = '%s'", name, restore(setExpr.Expr)))
 			}
 		}
 		if len(where) != len(pkColumnsName) {
@@ -449,6 +450,18 @@ func (i *MysqlDriverImpl) generateInsertRollbackSql(stmt *ast.InsertStmt) (strin
 	return rollbackSql, "", nil
 }
 
+// 还原抽象语法树节点至SQL
+func restore(node ast.Node) (sql string) {
+	var buf strings.Builder
+	rc := format.NewRestoreCtx(format.DefaultRestoreFlags, &buf)
+
+	if err := node.Restore(rc); err != nil {
+		return
+	}
+	sql = buf.String()
+	return
+}
+
 // generateDeleteRollbackSql generate insert SQL for delete.
 func (i *MysqlDriverImpl) generateDeleteRollbackSql(stmt *ast.DeleteStmt) (string, string, error) {
 	// not support multi-table syntax
@@ -603,7 +616,7 @@ func (i *MysqlDriverImpl) generateUpdateRollbackSql(stmt *ast.UpdateStmt) (strin
 					colChanged = true
 					if isPk {
 						isPkChanged = true
-						pkValue = util.ExprFormat(l.Expr)
+						pkValue = restore(l.Expr)
 					}
 				}
 			}
@@ -682,12 +695,12 @@ func (i *MysqlDriverImpl) generateGetRecordsSql(expr string, tableName *ast.Tabl
 		recordSql = fmt.Sprintf("%s AS %s", recordSql, tableAlias)
 	}
 	if where != nil {
-		recordSql = fmt.Sprintf("%s WHERE %s", recordSql, util.ExprFormat(where))
+		recordSql = fmt.Sprintf("%s WHERE %s", recordSql, restore(where))
 	}
 	if order != nil {
 		recordSql = fmt.Sprintf("%s ORDER BY", recordSql)
 		for _, item := range order.Items {
-			recordSql = fmt.Sprintf("%s %s", recordSql, util.ExprFormat(item.Expr))
+			recordSql = fmt.Sprintf("%s %s", recordSql, restore(item.Expr))
 			if item.Desc {
 				recordSql = fmt.Sprintf("%s DESC", recordSql)
 			}