[parser] parser: add support for show pump/drainer status (#217)

parser/ast/dml.go

@@ -1755,6 +1755,8 @@ const (
 	ShowPrivileges
 	ShowErrors
 	ShowBindings
+	ShowPumpStatus
+	ShowDrainerStatus
 )
 
 // ShowStmt is a statement to provide information about databases, tables, columns and so on.
@@ -1937,6 +1939,10 @@ func (n *ShowStmt) Restore(ctx *RestoreCtx) error {
 				ctx.WriteKeyWord("SESSION ")
 			}
 			ctx.WriteKeyWord("BINDINGS")
+		case ShowPumpStatus:
+			ctx.WriteKeyWord("PUMP STATUS")
+		case ShowDrainerStatus:
+			ctx.WriteKeyWord("DRAINER STATUS")
 		default:
 			return errors.New("Unknown ShowStmt type")
 		}

parser/misc.go

@@ -234,6 +234,7 @@ var tokenMap = map[string]int{
 	"DIV":                      div,
 	"DO":                       do,
 	"DOUBLE":                   doubleType,
+	"DRAINER":                  drainer,
 	"DROP":                     drop,
 	"DUAL":                     dual,
 	"DUPLICATE":                duplicate,
@@ -392,6 +393,7 @@ var tokenMap = map[string]int{
 	"PROCESS":                  process,
 	"PROCESSLIST":              processlist,
 	"PROFILES":                 profiles,
+	"PUMP":                     pump,
 	"QUARTER":                  quarter,
 	"QUERY":                    query,
 	"QUERIES":                  queries,

parser/parser.y

@@ -477,8 +477,10 @@ import (
 	buckets		"BUCKETS"
 	cancel		"CANCEL"
 	ddl		"DDL"
+	drainer         "DRAINER"
 	jobs		"JOBS"
 	job		    "JOB"
+	pump            "PUMP"
 	stats		"STATS"
 	statsMeta       "STATS_META"
 	statsHistograms "STATS_HISTOGRAMS"
@@ -3045,7 +3047,7 @@ UnReservedKeyword:
 
 
 TiDBKeyword:
-"ADMIN" | "BUCKETS" | "CANCEL" | "DDL" | "JOBS" | "JOB" | "STATS" | "STATS_META" | "STATS_HISTOGRAMS" | "STATS_BUCKETS" | "STATS_HEALTHY" | "TIDB" | "TIDB_HJ" | "TIDB_SMJ" | "TIDB_INLJ" | "RESTORE"
+"ADMIN" | "BUCKETS" | "CANCEL" | "DDL" | "DRAINER" | "JOBS" | "JOB" | "PUMP" | "STATS" | "STATS_META" | "STATS_HISTOGRAMS" | "STATS_BUCKETS" | "STATS_HEALTHY" | "TIDB" | "TIDB_HJ" | "TIDB_SMJ" | "TIDB_INLJ" | "RESTORE"
 
 NotKeywordToken:
  "ADDDATE" | "BIT_AND" | "BIT_OR" | "BIT_XOR" | "CAST" | "COPY" | "COUNT" | "CURTIME" | "DATE_ADD" | "DATE_SUB" | "EXTRACT" | "GET_FORMAT" | "GROUP_CONCAT"
@@ -6282,6 +6284,18 @@ ShowTargetFilterable:
 			Tp: ast.ShowProcedureStatus,
 		}
 	}
+|	"PUMP" "STATUS"
+	{
+		$$ = &ast.ShowStmt {
+			Tp: ast.ShowPumpStatus,
+		}
+	}
+|	"DRAINER" "STATUS"
+	{
+		$$ = &ast.ShowStmt {
+			Tp: ast.ShowDrainerStatus,
+		}
+	}
 |	"FUNCTION" "STATUS"
 	{
 		// This statement is similar to SHOW PROCEDURE STATUS but for stored functions.