server: add http api to get some info of sub-optimal query (#1… (#10925)

pingcap · Jun 24, 2019 · a480762 · a480762
1 parent 23cf1ad
commit a480762
Show file tree

Hide file tree

Showing 3 changed files with 366 additions and 28 deletions.
diff --git a/server/http_status.go b/server/http_status.go
@@ -49,6 +49,22 @@ func (s *Server) startStatusHTTP() {
 	go s.startHTTPServer()
 }
 
+func serveError(w http.ResponseWriter, status int, txt string) {
+	w.Header().Set("Content-Type", "text/plain; charset=utf-8")
+	w.Header().Set("X-Go-Pprof", "1")
+	w.Header().Del("Content-Disposition")
+	w.WriteHeader(status)
+	_, err := fmt.Fprintln(w, txt)
+	terror.Log(err)
+}
+
+func sleepWithCtx(ctx context.Context, d time.Duration) {
+	select {
+	case <-time.After(d):
+	case <-ctx.Done():
+	}
+}
+
 func (s *Server) startHTTPServer() {
 	router := mux.NewRouter()
 
@@ -123,26 +139,6 @@ func (s *Server) startHTTPServer() {
 	serverMux.HandleFunc("/debug/pprof/symbol", pprof.Symbol)
 	serverMux.HandleFunc("/debug/pprof/trace", pprof.Trace)
 
-	serveError := func(w http.ResponseWriter, status int, txt string) {
-		w.Header().Set("Content-Type", "text/plain; charset=utf-8")
-		w.Header().Set("X-Go-Pprof", "1")
-		w.Header().Del("Content-Disposition")
-		w.WriteHeader(status)
-		_, err := fmt.Fprintln(w, txt)
-		terror.Log(err)
-	}
-
-	sleep := func(w http.ResponseWriter, d time.Duration) {
-		var clientGone <-chan bool
-		if cn, ok := w.(http.CloseNotifier); ok {
-			clientGone = cn.CloseNotify()
-		}
-		select {
-		case <-time.After(d):
-		case <-clientGone:
-		}
-	}
-
 	serverMux.HandleFunc("/debug/zip", func(w http.ResponseWriter, r *http.Request) {
 		w.Header().Set("Content-Disposition", fmt.Sprintf(`attachment; filename="tidb_debug"`+time.Now().Format("20060102150405")+".zip"))
 
@@ -191,7 +187,7 @@ func (s *Server) startHTTPServer() {
 		if sec <= 0 || err != nil {
 			sec = 10
 		}
-		sleep(w, time.Duration(sec)*time.Second)
+		sleepWithCtx(r.Context(), time.Duration(sec)*time.Second)
 		rpprof.StopCPUProfile()
 
 		// dump config
@@ -220,11 +216,13 @@ func (s *Server) startHTTPServer() {
 		err = zw.Close()
 		terror.Log(err)
 	})
+	fetcher := sqlInfoFetcher{store: tikvHandlerTool.Store}
+	serverMux.HandleFunc("/debug/sub-optimal-plan", fetcher.zipInfoForSQL)
 
 	var (
-		err            error
 		httpRouterPage bytes.Buffer
 		pathTemplate   string
+		err            error
 	)
 	httpRouterPage.WriteString("<html><head><title>TiDB Status and Metrics Report</title></head><body><h1>TiDB Status and Metrics Report</h1><table>")
 	err = router.Walk(func(route *mux.Route, router *mux.Router, ancestors []*mux.Route) error {