Add pprof endpoints to vizier go services

Signed-off-by: Dom Del Nano <ddelnano@gmail.com>

Author: ddelnano

src/vizier/services/cloud_connector/cloud_connector_server.go

@@ -23,6 +23,7 @@ import (
 	"errors"
 	"fmt"
 	"net/http"
+	_ "net/http/pprof"
 	"time"
 
 	"github.com/gofrs/uuid"
@@ -174,6 +175,8 @@ func main() {
 	defer svr.Stop()
 
 	mux := http.NewServeMux()
+	// This handles all the pprof endpoints.
+	mux.Handle("/debug/", http.DefaultServeMux)
 	// Set up healthz endpoint.
 	healthz.RegisterDefaultChecks(mux)
 	// Set up readyz endpoint.

src/vizier/services/metadata/metadata_server.go

@@ -23,6 +23,7 @@ import (
 	"crypto/tls"
 	"fmt"
 	"net/http"
+	_ "net/http/pprof"
 	"os"
 	"path/filepath"
 	"strings"
@@ -283,6 +284,8 @@ func main() {
 		log.WithError(err).Fatal("Failed to create api environment")
 	}
 	mux := http.NewServeMux()
+	// This handles all the pprof endpoints.
+	mux.Handle("/debug/", http.DefaultServeMux)
 	healthz.RegisterDefaultChecks(mux)
 	metrics.MustRegisterMetricsHandlerNoDefaultMetrics(mux)
 

src/vizier/services/query_broker/query_broker_server.go

@@ -24,6 +24,7 @@ import (
 	"fmt"
 	"net"
 	"net/http"
+	_ "net/http/pprof"
 	"time"
 
 	"github.com/cenkalti/backoff/v4"
@@ -106,6 +107,8 @@ func main() {
 		log.WithError(err).Fatal("Failed to create api environment.")
 	}
 	mux := http.NewServeMux()
+	// This handles all the pprof endpoints.
+	mux.Handle("/debug/", http.DefaultServeMux)
 	healthz.RegisterDefaultChecks(mux)
 	metrics.MustRegisterMetricsHandlerNoDefaultMetrics(mux)