mirror of
https://github.com/ceph/ceph-csi.git
synced 2024-12-22 13:00:19 +00:00
774e8e4042
Add support for golang profiling. Standard tools like go tool pprof and curl work. example: $ go tool pprof http://localhost:8080/debug/pprof/profile $ go tool pprof http://localhost:8080/debug/pprof/heap $ curl http://localhost:8080/debug/pprof/heap?debug=1 https://golang.org/pkg/net/http/pprof/ contains more details about the pprof interface. Fixes: #1699 Signed-off-by: Yati Padia <ypadia@redhat.com>
48 lines
1.2 KiB
Go
48 lines
1.2 KiB
Go
package util
|
|
|
|
import (
|
|
"net"
|
|
"net/http"
|
|
"net/http/pprof"
|
|
"net/url"
|
|
runtime_pprof "runtime/pprof"
|
|
"strconv"
|
|
|
|
"github.com/prometheus/client_golang/prometheus/promhttp"
|
|
)
|
|
|
|
// ValidateURL validates the url.
|
|
func ValidateURL(c *Config) error {
|
|
_, err := url.Parse(c.MetricsPath)
|
|
return err
|
|
}
|
|
|
|
// StartMetricsServer starts http server.
|
|
func StartMetricsServer(c *Config) {
|
|
addr := net.JoinHostPort(c.MetricsIP, strconv.Itoa(c.MetricsPort))
|
|
http.Handle(c.MetricsPath, promhttp.Handler())
|
|
err := http.ListenAndServe(addr, nil)
|
|
if err != nil {
|
|
FatalLogMsg("failed to listen on address %v: %s", addr, err)
|
|
}
|
|
}
|
|
func addPath(name string, handler http.Handler) {
|
|
http.Handle(name, handler)
|
|
DebugLogMsg("DEBUG: registered profiling handler on /debug/pprof/%s\n", name)
|
|
}
|
|
|
|
// EnableProfiling enables golang profiling.
|
|
func EnableProfiling() {
|
|
for _, profile := range runtime_pprof.Profiles() {
|
|
name := profile.Name()
|
|
handler := pprof.Handler(name)
|
|
addPath(name, handler)
|
|
}
|
|
|
|
// static profiles as listed in net/http/pprof/pprof.go:init()
|
|
addPath("cmdline", http.HandlerFunc(pprof.Cmdline))
|
|
addPath("profile", http.HandlerFunc(pprof.Profile))
|
|
addPath("symbol", http.HandlerFunc(pprof.Symbol))
|
|
addPath("trace", http.HandlerFunc(pprof.Trace))
|
|
}
|