Add Admin Telemetry Statement (pingcap#908) (pingcap#909)

Signed-off-by: Breezewish <[email protected]>
wjhuang2016 · Jun 23, 2020 · 3a18f1e · 3a18f1e
1 parent 789c193
commit 3a18f1e
Show file tree

Hide file tree

Showing 5 changed files with 7,811 additions and 7,749 deletions.
diff --git a/ast/misc.go b/ast/misc.go
@@ -1619,6 +1619,8 @@ const (
 	AdminCaptureBindings
 	AdminEvolveBindings
 	AdminReloadBindings
+	AdminShowTelemetry
+	AdminResetTelemetryID
 )
 
 // HandleRange represents a range where handle value >= Begin and < End.
@@ -1827,6 +1829,10 @@ func (n *AdminStmt) Restore(ctx *format.RestoreCtx) error {
 		ctx.WriteKeyWord("EVOLVE BINDINGS")
 	case AdminReloadBindings:
 		ctx.WriteKeyWord("RELOAD BINDINGS")
+	case AdminShowTelemetry:
+		ctx.WriteKeyWord("SHOW TELEMETRY")
+	case AdminResetTelemetryID:
+		ctx.WriteKeyWord("RESET TELEMETRY_ID")
 	default:
 		return errors.New("Unsupported AdminStmt type")
 	}

diff --git a/misc.go b/misc.go
@@ -535,6 +535,7 @@ var tokenMap = map[string]int{
 	"REPLICA":                  replica,
 	"REPLICATION":              replication,
 	"REQUIRE":                  require,
+	"RESET":                    reset,
 	"RESPECT":                  respect,
 	"RESTORE":                  restore,
 	"RESTORES":                 restores,
@@ -637,6 +638,8 @@ var tokenMap = map[string]int{
 	"TABLE":                    tableKwd,
 	"TABLES":                   tables,
 	"TABLESPACE":               tablespace,
+	"TELEMETRY":                telemetry,
+	"TELEMETRY_ID":             telemetryID,
 	"TEMPORARY":                temporary,
 	"TEMPTABLE":                temptable,
 	"TERMINATED":               terminated,