neo-go/pkg/network/metrics/pprof.go

36 lines
855 B
Go
Raw Normal View History

package metrics
import (
"net/http"
"net/http/pprof"
2019-12-30 07:43:05 +00:00
"go.uber.org/zap"
)
2019-12-17 11:51:07 +00:00
// PprofService https://golang.org/pkg/net/http/pprof/.
type PprofService Service
// NewPprofService creates a new service for gathering pprof metrics.
2019-12-30 07:43:05 +00:00
func NewPprofService(cfg Config, log *zap.Logger) *Service {
if log == nil {
return nil
}
handler := http.NewServeMux()
2019-12-17 11:49:28 +00:00
handler.HandleFunc("/debug/pprof/", pprof.Index)
handler.HandleFunc("/debug/pprof/cmdline", pprof.Cmdline)
handler.HandleFunc("/debug/pprof/profile", pprof.Profile)
handler.HandleFunc("/debug/pprof/symbol", pprof.Symbol)
handler.HandleFunc("/debug/pprof/trace", pprof.Trace)
return &Service{
2019-12-30 07:43:05 +00:00
Server: &http.Server{
Addr: cfg.Address + ":" + cfg.Port,
Handler: handler,
2019-12-30 07:43:05 +00:00
},
config: cfg,
serviceType: "Pprof",
log: log.With(zap.String("service", "Pprof")),
}
}