Add in support for the other /debug/pprof URLs into ProxyFS's & PFSAg…

…ent's embedded HTTP Servers
NVIDIA · Jul 10, 2020 · a8a435f · a8a435f
1 parent ed04151
commit a8a435f
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 13 deletions.
diff --git a/httpserver/request_handler.go b/httpserver/request_handler.go
@@ -157,8 +157,6 @@ func doGet(responseWriter http.ResponseWriter, request *http.Request) {
 		doGetOfMetrics(responseWriter, request)
 	case "/stats" == path:
 		doGetOfStats(responseWriter, request)
-	case "/debug/pprof" == path:
-		pprof.Index(responseWriter, request)
 	case "/debug/pprof/cmdline" == path:
 		pprof.Cmdline(responseWriter, request)
 	case "/debug/pprof/profile" == path:
@@ -167,6 +165,8 @@ func doGet(responseWriter http.ResponseWriter, request *http.Request) {
 		pprof.Symbol(responseWriter, request)
 	case "/debug/pprof/trace" == path:
 		pprof.Trace(responseWriter, request)
+	case strings.HasPrefix(request.URL.Path, "/debug/pprof"):
+		pprof.Index(responseWriter, request)
 	case "/open-iconic/font/css/open-iconic-bootstrap.min.css" == path:
 		responseWriter.Header().Set("Content-Type", openIconicBootstrapDotCSSContentType)
 		responseWriter.WriteHeader(http.StatusOK)

diff --git a/pfsagentd/http_server.go b/pfsagentd/http_server.go
@@ -78,24 +78,24 @@ func serveGet(responseWriter http.ResponseWriter, request *http.Request) {
 
 	path = strings.TrimRight(request.URL.Path, "/")
 
-	switch path {
-	case "/config":
+	switch {
+	case "/config" == path:
 		serveGetOfConfig(responseWriter, request)
-	case "/debug/pprof":
-		pprof.Index(responseWriter, request)
-	case "/debug/pprof/cmdline":
+	case "/debug/pprof/cmdline" == path:
 		pprof.Cmdline(responseWriter, request)
-	case "/debug/pprof/profile":
+	case "/debug/pprof/profile" == path:
 		pprof.Profile(responseWriter, request)
-	case "/debug/pprof/symbol":
+	case "/debug/pprof/symbol" == path:
 		pprof.Symbol(responseWriter, request)
-	case "/debug/pprof/trace":
+	case "/debug/pprof/trace" == path:
 		pprof.Trace(responseWriter, request)
-	case "/metrics":
+	case strings.HasPrefix(path, "/debug/pprof"):
+		pprof.Index(responseWriter, request)
+	case "/metrics" == path:
 		serveGetOfMetrics(responseWriter, request)
-	case "/stats":
+	case "/stats" == path:
 		serveGetOfStats(responseWriter, request)
-	case "/version":
+	case "/version" == path:
 		serveGetOfVersion(responseWriter, request)
 	default:
 		responseWriter.WriteHeader(http.StatusNotFound)