09dc25d336
* Prepare readme for release * Update vendored packages * New version of dns dependency handles rejecting UPDATEs under the hood * Go 1.11 required
45 lines
1.4 KiB
Go
45 lines
1.4 KiB
Go
package pprofhandler
|
|
|
|
import (
|
|
"net/http/pprof"
|
|
rtp "runtime/pprof"
|
|
"strings"
|
|
|
|
"github.com/valyala/fasthttp"
|
|
"github.com/valyala/fasthttp/fasthttpadaptor"
|
|
)
|
|
|
|
var (
|
|
cmdline = fasthttpadaptor.NewFastHTTPHandlerFunc(pprof.Cmdline)
|
|
profile = fasthttpadaptor.NewFastHTTPHandlerFunc(pprof.Profile)
|
|
symbol = fasthttpadaptor.NewFastHTTPHandlerFunc(pprof.Symbol)
|
|
trace = fasthttpadaptor.NewFastHTTPHandlerFunc(pprof.Trace)
|
|
index = fasthttpadaptor.NewFastHTTPHandlerFunc(pprof.Index)
|
|
)
|
|
|
|
// PprofHandler serves server runtime profiling data in the format expected by the pprof visualization tool.
|
|
//
|
|
// See https://golang.org/pkg/net/http/pprof/ for details.
|
|
func PprofHandler(ctx *fasthttp.RequestCtx) {
|
|
ctx.Response.Header.Set("Content-Type", "text/html")
|
|
if strings.HasPrefix(string(ctx.Path()), "/debug/pprof/cmdline") {
|
|
cmdline(ctx)
|
|
} else if strings.HasPrefix(string(ctx.Path()), "/debug/pprof/profile") {
|
|
profile(ctx)
|
|
} else if strings.HasPrefix(string(ctx.Path()), "/debug/pprof/symbol") {
|
|
symbol(ctx)
|
|
} else if strings.HasPrefix(string(ctx.Path()), "/debug/pprof/trace") {
|
|
trace(ctx)
|
|
} else {
|
|
for _, v := range rtp.Profiles() {
|
|
ppName := v.Name()
|
|
if strings.HasPrefix(string(ctx.Path()), "/debug/pprof/"+ppName) {
|
|
namedHandler := fasthttpadaptor.NewFastHTTPHandlerFunc(pprof.Handler(ppName).ServeHTTP)
|
|
namedHandler(ctx)
|
|
return
|
|
}
|
|
}
|
|
index(ctx)
|
|
}
|
|
}
|