update

pingcap · Feb 20, 2024 · c4d9dbf · c4d9dbf
1 parent 246e3ce
commit c4d9dbf
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/pkg/parser/parser.go b/pkg/parser/parser.go
@@ -19989,7 +19989,7 @@ yynewstate:
 			var setOprList []ast.Node
 			switch x := yyS[yypt-1].expr.(*ast.SubqueryExpr).Query.(type) {
 			case *ast.SelectStmt:
-				setOprList = []ast.Node{&ast.SetOprSelectList{Selects: []ast.Node{x}}}
+				setOprList = []ast.Node{&ast.SetOprSelectList{Selects: []ast.Node{x}, With: x.With}}
 			case *ast.SetOprStmt:
 				setOprList = []ast.Node{&ast.SetOprSelectList{Selects: x.SelectList.Selects, With: x.With, Limit: x.Limit, OrderBy: x.OrderBy}}
 			}
@@ -20002,7 +20002,7 @@ yynewstate:
 			var setOprList []ast.Node
 			switch x := yyS[yypt-1].expr.(*ast.SubqueryExpr).Query.(type) {
 			case *ast.SelectStmt:
-				setOprList = []ast.Node{&ast.SetOprSelectList{Selects: []ast.Node{x}}}
+				setOprList = []ast.Node{&ast.SetOprSelectList{Selects: []ast.Node{x}, With: x.With}}
 			case *ast.SetOprStmt:
 				setOprList = []ast.Node{&ast.SetOprSelectList{Selects: x.SelectList.Selects, With: x.With, Limit: x.Limit, OrderBy: x.OrderBy}}
 			}
@@ -20015,7 +20015,7 @@ yynewstate:
 			var setOprList []ast.Node
 			switch x := yyS[yypt-2].expr.(*ast.SubqueryExpr).Query.(type) {
 			case *ast.SelectStmt:
-				setOprList = []ast.Node{&ast.SetOprSelectList{Selects: []ast.Node{x}}}
+				setOprList = []ast.Node{&ast.SetOprSelectList{Selects: []ast.Node{x}, With: x.With}}
 			case *ast.SetOprStmt:
 				setOprList = []ast.Node{&ast.SetOprSelectList{Selects: x.SelectList.Selects, With: x.With, Limit: x.Limit, OrderBy: x.OrderBy}}
 			}

diff --git a/pkg/parser/parser.y b/pkg/parser/parser.y
@@ -10368,7 +10368,7 @@ SetOprStmtWithLimitOrderBy:
 		var setOprList []ast.Node
 		switch x := $1.(*ast.SubqueryExpr).Query.(type) {
 		case *ast.SelectStmt:
-			setOprList = []ast.Node{&ast.SetOprSelectList{Selects: []ast.Node{x}}}
+			setOprList = []ast.Node{&ast.SetOprSelectList{Selects: []ast.Node{x}, With: x.With}}
 		case *ast.SetOprStmt:
 			setOprList = []ast.Node{&ast.SetOprSelectList{Selects: x.SelectList.Selects, With: x.With, Limit: x.Limit, OrderBy: x.OrderBy}}
 		}
@@ -10381,7 +10381,7 @@ SetOprStmtWithLimitOrderBy:
 		var setOprList []ast.Node
 		switch x := $1.(*ast.SubqueryExpr).Query.(type) {
 		case *ast.SelectStmt:
-			setOprList = []ast.Node{&ast.SetOprSelectList{Selects: []ast.Node{x}}}
+			setOprList = []ast.Node{&ast.SetOprSelectList{Selects: []ast.Node{x}, With: x.With}}
 		case *ast.SetOprStmt:
 			setOprList = []ast.Node{&ast.SetOprSelectList{Selects: x.SelectList.Selects, With: x.With, Limit: x.Limit, OrderBy: x.OrderBy}}
 		}
@@ -10420,7 +10420,7 @@ SetOprClause:
 		var setOprList []ast.Node
 		switch x := $1.(*ast.SubqueryExpr).Query.(type) {
 		case *ast.SelectStmt:
-			setOprList = []ast.Node{&ast.SetOprSelectList{Selects: []ast.Node{x}}}
+			setOprList = []ast.Node{&ast.SetOprSelectList{Selects: []ast.Node{x}, With: x.With}}
 		case *ast.SetOprStmt:
 			setOprList = []ast.Node{&ast.SetOprSelectList{Selects: x.SelectList.Selects, With: x.With, Limit: x.Limit, OrderBy: x.OrderBy}}
 		}