frostfs-node/pkg/util/profiler/pprof.go

package profiler

import (
	"context"
	"expvar"
	"net/http"
	"net/http/pprof"

	"github.com/spf13/viper"
	"go.uber.org/zap"
)

// Profiler is an interface of profiler.
type Profiler interface {
	Start(ctx context.Context)
	Stop()
}

const profilerKey = "pprof"

// NewProfiler is a profiler's constructor.
func NewProfiler(l *zap.Logger, v *viper.Viper) Profiler {
	if !v.GetBool(profilerKey + ".enabled") {
		l.Debug("pprof server disabled")
		return nil
	}

	mux := http.NewServeMux()

	mux.Handle("/debug/vars", expvar.Handler())

	mux.HandleFunc("/debug/pprof/", pprof.Index)
	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)

	return newHTTPServer(httpParams{
		Key:     profilerKey,
		Viper:   v,
		Logger:  l,
		Handler: mux,
	})
}