[client] Add pprof build tag (#2964)

* Add pprof build tag

* Change env handling
This commit is contained in:
Zoltan Papp 2024-12-01 19:22:52 +01:00 committed by GitHub
parent e4a5fb3e91
commit ecb44ff306
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

33
client/cmd/pprof.go Normal file
View File

@ -0,0 +1,33 @@
//go:build pprof
// +build pprof
package cmd
import (
"net/http"
_ "net/http/pprof"
"os"
log "github.com/sirupsen/logrus"
)
func init() {
addr := pprofAddr()
go pprof(addr)
}
func pprofAddr() string {
listenAddr := os.Getenv("NB_PPROF_ADDR")
if listenAddr == "" {
return "localhost:6969"
}
return listenAddr
}
func pprof(listenAddr string) {
log.Infof("listening pprof on: %s\n", listenAddr)
if err := http.ListenAndServe(listenAddr, nil); err != nil {
log.Fatalf("Failed to start pprof: %v", err)
}
}