parser: fix some parse error for parenthese subqueries (#1255)

pingcap · Aug 16, 2021 · b4bbfae · b4bbfae
1 parent f8bd71b
commit b4bbfae
Show file tree

Hide file tree

Showing 4 changed files with 7,314 additions and 7,044 deletions.
diff --git a/ast/dml.go b/ast/dml.go
@@ -1544,6 +1544,7 @@ func (s *SetOprType) String() string {
 type SetOprStmt struct {
 	dmlNode
 
+	IsInBraces bool
 	SelectList *SetOprSelectList
 	OrderBy    *OrderByClause
 	Limit      *Limit
@@ -1563,6 +1564,12 @@ func (n *SetOprStmt) Restore(ctx *format.RestoreCtx) error {
 			return errors.Annotate(err, "An error occurred while restore UnionStmt.With")
 		}
 	}
+	if n.IsInBraces {
+		ctx.WritePlain("(")
+		defer func() {
+			ctx.WritePlain(")")
+		}()
+	}
 
 	if err := n.SelectList.Restore(ctx); err != nil {
 		return errors.Annotate(err, "An error occurred while restore SetOprStmt.SelectList")