From 6822ad950f92e2c8e8da37fd1ed63c38f2ede630 Mon Sep 17 00:00:00 2001 From: "m.nabokikh" Date: Tue, 12 Apr 2022 12:12:37 +0400 Subject: [PATCH] feat: enable profiling endpoints Signed-off-by: m.nabokikh --- cmd/dex/config.go | 2 ++ cmd/dex/serve.go | 13 +++++++++++++ examples/config-dev.yaml | 1 + 3 files changed, 16 insertions(+) diff --git a/cmd/dex/config.go b/cmd/dex/config.go index 37167bb0..7bb7fbb7 100644 --- a/cmd/dex/config.go +++ b/cmd/dex/config.go @@ -151,6 +151,8 @@ type Web struct { // Telemetry is the config format for telemetry including the HTTP server config. type Telemetry struct { HTTP string `json:"http"` + // EnableProfiling makes profiling endpoints available via web interface host:port/debug/pprof/ + EnableProfiling bool `json:"enableProfiling"` } // GRPC is the config for the gRPC API. diff --git a/cmd/dex/serve.go b/cmd/dex/serve.go index 1916f834..c8fb95eb 100644 --- a/cmd/dex/serve.go +++ b/cmd/dex/serve.go @@ -8,6 +8,7 @@ import ( "fmt" "net" "net/http" + "net/http/pprof" "os" "runtime" "strings" @@ -368,6 +369,10 @@ func runServe(options serveOptions) error { return fmt.Errorf("listening (%s) on %s: %v", name, c.Telemetry.HTTP, err) } + if c.Telemetry.EnableProfiling { + pprofHandler(telemetryRouter) + } + server := &http.Server{ Handler: telemetryRouter, } @@ -550,3 +555,11 @@ func applyConfigOverrides(options serveOptions, config *Config) { config.Frontend.Dir = os.Getenv("DEX_FRONTEND_DIR") } } + +func pprofHandler(router *http.ServeMux) { + router.HandleFunc("/debug/pprof/", pprof.Index) + router.HandleFunc("/debug/pprof/cmdline", pprof.Cmdline) + router.HandleFunc("/debug/pprof/profile", pprof.Profile) + router.HandleFunc("/debug/pprof/symbol", pprof.Symbol) + router.HandleFunc("/debug/pprof/trace", pprof.Trace) +} diff --git a/examples/config-dev.yaml b/examples/config-dev.yaml index 6cae823c..bf11570a 100644 --- a/examples/config-dev.yaml +++ b/examples/config-dev.yaml @@ -63,6 +63,7 @@ web: # Configuration for telemetry telemetry: http: 0.0.0.0:5558 + # enableProfiling: true # Uncomment this block to enable the gRPC API. This values MUST be different # from the HTTP endpoints.