mirrors/lnd
1
0
Fork 0
mirror of https://github.com/lightningnetwork/lnd.git synced 2025-03-13 11:09:23 +01:00
Code Issues Projects Releases Wiki Activity

Merge pull request #7777 from yyforyongyu/fix-pprof-redirect

Browse source 
lnd: fix pprof server redirects
This commit is contained in:
Oliver Gugger 2023-06-22 09:20:31 +02:00 • committed by GitHub
commit 13e198fbde
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 22 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

32
lnd.go
View file

@ -11,9 +11,9 @@ import (
"io/ioutil"
"net"
"net/http"
_ "net/http/pprof" // nolint:gosec // used to set up profiling HTTP handlers.
"net/http/pprof"
"os"
"runtime/pprof"
runtimePprof "runtime/pprof"
"strings"
"sync"
"time"
@ -187,16 +187,26 @@ func Main(cfg *Config, lisCfg ListenerCfg, implCfg *ImplementationCfg,
// Enable http profiling server if requested.
if cfg.Profile != "" {
// Create the http handler.
profileRedirect := http.RedirectHandler(
"/debug/pprof", http.StatusSeeOther,
)
http.Handle("/", profileRedirect)
pprofMux := http.NewServeMux()
pprofMux.HandleFunc("/debug/pprof/", pprof.Index)
pprofMux.HandleFunc("/debug/pprof/cmdline", pprof.Cmdline)
pprofMux.HandleFunc("/debug/pprof/profile", pprof.Profile)
pprofMux.HandleFunc("/debug/pprof/symbol", pprof.Symbol)
pprofMux.HandleFunc("/debug/pprof/trace", pprof.Trace)
// Redirect all requests to the pprof handler, thus visiting
// `127.0.0.1:6060` will be redirected to
// `127.0.0.1:6060/debug/pprof`.
pprofMux.Handle("/", http.RedirectHandler(
"/debug/pprof/", http.StatusSeeOther,
))
ltndLog.Infof("Pprof listening on %v", cfg.Profile)
// Create the pprof server.
pprofServer := &http.Server{
Addr: cfg.Profile,
Handler: profileRedirect,
Handler: pprofMux,
ReadHeaderTimeout: 5 * time.Second,
}
@ -225,9 +235,11 @@ func Main(cfg *Config, lisCfg ListenerCfg, implCfg *ImplementationCfg,
if err != nil {
return mkErr("unable to create CPU profile: %v", err)
}
pprof.StartCPUProfile(f)
defer f.Close()
defer pprof.StopCPUProfile()
_ = runtimePprof.StartCPUProfile(f)
defer func() {
_ = f.Close()
}()
defer runtimePprof.StopCPUProfile()
}
// Run configuration dependent DB pre-initialization. Note that this