add support for pprof server (#5527)

Signed-off-by: igor <igor@igor.io>
This commit is contained in:
Igor
2022-06-14 07:06:55 +02:00
committed by GitHub
parent 763a8bc8fe
commit 2999289e68
4 changed files with 21 additions and 0 deletions

View File

@@ -9,9 +9,11 @@ import (
"log"
"net"
"net/http"
"net/http/pprof"
"os"
"path/filepath"
"github.com/gorilla/mux"
"github.com/k3s-io/k3s/pkg/daemons/config"
"github.com/k3s-io/k3s/pkg/etcd"
"github.com/k3s-io/k3s/pkg/version"
@@ -94,6 +96,17 @@ func (c *Cluster) initClusterAndHTTPS(ctx context.Context) error {
return err
}
if c.config.EnablePProf {
mux := mux.NewRouter()
mux.HandleFunc("/debug/pprof/cmdline", pprof.Cmdline)
mux.HandleFunc("/debug/pprof/profile", pprof.Profile)
mux.HandleFunc("/debug/pprof/symbol", pprof.Symbol)
mux.HandleFunc("/debug/pprof/trace", pprof.Trace)
mux.PathPrefix("/debug/pprof/").HandlerFunc(pprof.Index)
mux.NotFoundHandler = handler
handler = mux
}
// Create a HTTP server with the registered request handlers, using logrus for logging
server := http.Server{
Handler: handler,