add fasthttp pprof handler

This commit is contained in:
Evgeniy Kulikov 2020-02-28 19:56:50 +03:00
parent c6a06ca709
commit 437133e280
No known key found for this signature in database
GPG key ID: BF6AEE0A2A699BF2

37
pprof.go Normal file
View file

@ -0,0 +1,37 @@
package main
import (
"net/http/pprof"
rtp "runtime/pprof"
"github.com/valyala/fasthttp"
"github.com/valyala/fasthttp/fasthttpadaptor"
)
func pprofHandler() fasthttp.RequestHandler {
items := rtp.Profiles()
profiles := map[string]fasthttp.RequestHandler{
"": fasthttpadaptor.NewFastHTTPHandlerFunc(pprof.Index),
"cmdline": fasthttpadaptor.NewFastHTTPHandlerFunc(pprof.Cmdline),
"profile": fasthttpadaptor.NewFastHTTPHandlerFunc(pprof.Profile),
"symbol": fasthttpadaptor.NewFastHTTPHandlerFunc(pprof.Symbol),
"trace": fasthttpadaptor.NewFastHTTPHandlerFunc(pprof.Trace),
}
for i := range items {
name := items[i].Name()
profiles[name] = fasthttpadaptor.NewFastHTTPHandler(pprof.Handler(name))
}
return func(ctx *fasthttp.RequestCtx) {
name, _ := ctx.UserValue("name").(string)
if handler, ok := profiles[name]; ok {
handler(ctx)
return
}
ctx.Error("Not found", fasthttp.StatusNotFound)
}
}