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

lyonzhi · Feb 21, 2019 · c604768 · c604768
1 parent 9bb3b52
commit c604768
Show file tree

Hide file tree

Showing 4 changed files with 5,641 additions and 5,589 deletions.
diff --git a/ast/dml.go b/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")
 		}

diff --git a/misc.go b/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,