[baseserver] Serve pprof on /debug/pprof

Author: easyCZ

components/common-go/baseserver/server.go

@@ -7,6 +7,7 @@ package baseserver
 import (
 	"context"
 	"fmt"
+	"github.com/gitpod-io/gitpod/common-go/pprof"
 	"github.com/prometheus/client_golang/prometheus/promhttp"
 	"github.com/sirupsen/logrus"
 	"google.golang.org/grpc"
@@ -216,6 +217,7 @@ func (s *Server) newHTTPMux() *http.ServeMux {
 	mux.HandleFunc("/ready", func(w http.ResponseWriter, r *http.Request) {
 		_, _ = w.Write([]byte(`ready`))
 	})
+	s.Logger().WithField("protocol", "http").Debug("Serving readiness handler on /ready")
 
 	// Metrics endpoint
 	metricsHandler := promhttp.Handler()
@@ -225,6 +227,10 @@ func (s *Server) newHTTPMux() *http.ServeMux {
 		)
 	}
 	mux.Handle("/metrics", metricsHandler)
+	s.Logger().WithField("protocol", "http").Debug("Serving metrics on /metrics")
+
+	mux.Handle(pprof.Path, pprof.Handler())
+	s.Logger().WithField("protocol", "http").Debug("Serving profiler on /debug/pprof")
 
 	return mux
 }

components/common-go/baseserver/server_test.go

@@ -7,6 +7,7 @@ package baseserver_test
 import (
 	"fmt"
 	"github.com/gitpod-io/gitpod/common-go/baseserver"
+	"github.com/gitpod-io/gitpod/common-go/pprof"
 	"github.com/prometheus/client_golang/prometheus"
 	"github.com/stretchr/testify/require"
 	"net/http"
@@ -58,3 +59,12 @@ func TestServer_ServesMetricsEndpointWithCustomMetricsConfig(t *testing.T) {
 	require.NoError(t, err)
 	require.Equal(t, http.StatusOK, resp.StatusCode)
 }
+
+func TestServer_ServesPprof(t *testing.T) {
+	srv := baseserver.NewForTests(t)
+	baseserver.StartServerForTests(t, srv)
+
+	resp, err := http.Get(srv.HTTPAddress() + pprof.Path)
+	require.NoError(t, err)
+	require.Equalf(t, http.StatusOK, resp.StatusCode, "must serve pprof on %s", pprof.Path)
+}