[parser] Parser: Support SQL_BIG_RESULT, SQL_SMALL_RESULT, SQL_BUFFER…

…_RESULT (pingcap#304) * Support SQL_BIG_RESULT, SQL_SMALL_RESULT, SQL_BUFFER_RESULT * update test case * code formatting * word wrap
xhebox · May 29, 2019 · 19c511c · 19c511c
1 parent 6528e24
commit 19c511c
Show file tree

Hide file tree

Showing 6 changed files with 6,442 additions and 6,308 deletions.
diff --git a/parser/ast/dml.go b/parser/ast/dml.go
@@ -777,6 +777,18 @@ func (n *SelectStmt) Restore(ctx *RestoreCtx) error {
 		ctx.WritePlain(" ")
 	}
 
+	if n.SelectStmtOpts.SQLSmallResult {
+		ctx.WriteKeyWord("SQL_SMALL_RESULT ")
+	}
+
+	if n.SelectStmtOpts.SQLBigResult {
+		ctx.WriteKeyWord("SQL_BIG_RESULT ")
+	}
+
+	if n.SelectStmtOpts.SQLBufferResult {
+		ctx.WriteKeyWord("SQL_BUFFER_RESULT ")
+	}
+
 	if !n.SelectStmtOpts.SQLCache {
 		ctx.WriteKeyWord("SQL_NO_CACHE ")
 	}

diff --git a/parser/ast/misc.go b/parser/ast/misc.go
@@ -1974,12 +1974,15 @@ func (i Ident) String() string {
 
 // SelectStmtOpts wrap around select hints and switches
 type SelectStmtOpts struct {
-	Distinct      bool
-	SQLCache      bool
-	CalcFoundRows bool
-	StraightJoin  bool
-	Priority      mysql.PriorityEnum
-	TableHints    []*TableOptimizerHint
+	Distinct        bool
+	SQLBigResult    bool
+	SQLBufferResult bool
+	SQLCache        bool
+	SQLSmallResult  bool
+	CalcFoundRows   bool
+	StraightJoin    bool
+	Priority        mysql.PriorityEnum
+	TableHints      []*TableOptimizerHint
 }
 
 // TableOptimizerHint is Table level optimizer hint

diff --git a/parser/misc.go b/parser/misc.go
@@ -469,9 +469,12 @@ var tokenMap = map[string]int{
 	"SOME":                     some,
 	"SPLIT":                    split,
 	"SQL":                      sql,
+	"SQL_BIG_RESULT":           sqlBigResult,
+	"SQL_BUFFER_RESULT":        sqlBufferResult,
 	"SQL_CACHE":                sqlCache,
 	"SQL_CALC_FOUND_ROWS":      sqlCalcFoundRows,
 	"SQL_NO_CACHE":             sqlNoCache,
+	"SQL_SMALL_RESULT":         sqlSmallResult,
 	"SOURCE":                   source,
 	"SSL":                      ssl,
 	"START":                    start,