package main import ( "net/http/pprof" rtp "runtime/pprof" "github.com/fasthttp/router" "github.com/valyala/fasthttp" "github.com/valyala/fasthttp/fasthttpadaptor" ) func attachProfiler(r *router.Router) { r.GET("/debug/pprof/", pprofHandler()) r.GET("/debug/pprof/:name", pprofHandler()) } 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) } }