status-go/profiling/profiler.go

45 lines
1.0 KiB
Go
Raw Permalink Normal View History

2018-03-23 13:58:40 +00:00
package profiling
import (
"fmt"
"net/http"
hpprof "net/http/pprof"
2023-01-13 17:12:46 +00:00
"time"
2018-03-23 13:58:40 +00:00
"github.com/ethereum/go-ethereum/log"
"github.com/status-im/status-go/common"
2018-03-23 13:58:40 +00:00
)
// Profiler runs and controls a HTTP pprof interface.
type Profiler struct {
server *http.Server
}
// NewProfiler creates an instance of the profiler with
// the given port.
func NewProfiler(port int) *Profiler {
mux := http.NewServeMux()
mux.HandleFunc("/debug/pprof/", hpprof.Index)
mux.HandleFunc("/debug/pprof/cmdline", hpprof.Cmdline)
mux.HandleFunc("/debug/pprof/profile", hpprof.Profile)
mux.HandleFunc("/debug/pprof/symbol", hpprof.Symbol)
mux.HandleFunc("/debug/pprof/trace", hpprof.Trace)
p := Profiler{
server: &http.Server{
2023-01-13 17:12:46 +00:00
Addr: fmt.Sprintf(":%d", port),
ReadHeaderTimeout: 5 * time.Second,
Handler: mux,
2018-03-23 13:58:40 +00:00
},
}
return &p
}
// Go starts the HTTP pprof in the background.
func (p *Profiler) Go() {
go func() {
defer common.LogOnPanic()
2018-03-23 13:58:40 +00:00
log.Info("debug server stopped", "err", p.server.ListenAndServe())
}()
log.Info("debug server started")
}