124 lines
3.4 KiB
Go
124 lines
3.4 KiB
Go
package ginpprof
|
|
|
|
import (
|
|
"net/http/pprof"
|
|
"strings"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
)
|
|
|
|
// Wrap adds several routes from package `net/http/pprof` to *gin.Engine object
|
|
func Wrap(router *gin.Engine) {
|
|
WrapGroup(&router.RouterGroup)
|
|
}
|
|
|
|
// Wrapper make sure we are backward compatible
|
|
var Wrapper = Wrap
|
|
|
|
// WrapGroup adds several routes from package `net/http/pprof` to *gin.RouterGroup object
|
|
func WrapGroup(router *gin.RouterGroup) {
|
|
routers := []struct {
|
|
Method string
|
|
Path string
|
|
Handler gin.HandlerFunc
|
|
}{
|
|
{"GET", "/debug/pprof/", IndexHandler()},
|
|
{"GET", "/debug/pprof/heap", HeapHandler()},
|
|
{"GET", "/debug/pprof/goroutine", GoroutineHandler()},
|
|
{"GET", "/debug/pprof/block", BlockHandler()},
|
|
{"GET", "/debug/pprof/threadcreate", ThreadCreateHandler()},
|
|
{"GET", "/debug/pprof/cmdline", CmdlineHandler()},
|
|
{"GET", "/debug/pprof/profile", ProfileHandler()},
|
|
{"GET", "/debug/pprof/symbol", SymbolHandler()},
|
|
{"POST", "/debug/pprof/symbol", SymbolHandler()},
|
|
{"GET", "/debug/pprof/trace", TraceHandler()},
|
|
{"GET", "/debug/pprof/mutex", MutexHandler()},
|
|
}
|
|
|
|
basePath := strings.TrimSuffix(router.BasePath(), "/")
|
|
var prefix string
|
|
|
|
switch {
|
|
case basePath == "":
|
|
prefix = ""
|
|
case strings.HasSuffix(basePath, "/debug"):
|
|
prefix = "/debug"
|
|
case strings.HasSuffix(basePath, "/debug/pprof"):
|
|
prefix = "/debug/pprof"
|
|
}
|
|
|
|
for _, r := range routers {
|
|
router.Handle(r.Method, strings.TrimPrefix(r.Path, prefix), r.Handler)
|
|
}
|
|
}
|
|
|
|
// IndexHandler will pass the call from /debug/pprof to pprof
|
|
func IndexHandler() gin.HandlerFunc {
|
|
return func(ctx *gin.Context) {
|
|
pprof.Index(ctx.Writer, ctx.Request)
|
|
}
|
|
}
|
|
|
|
// HeapHandler will pass the call from /debug/pprof/heap to pprof
|
|
func HeapHandler() gin.HandlerFunc {
|
|
return func(ctx *gin.Context) {
|
|
pprof.Handler("heap").ServeHTTP(ctx.Writer, ctx.Request)
|
|
}
|
|
}
|
|
|
|
// GoroutineHandler will pass the call from /debug/pprof/goroutine to pprof
|
|
func GoroutineHandler() gin.HandlerFunc {
|
|
return func(ctx *gin.Context) {
|
|
pprof.Handler("goroutine").ServeHTTP(ctx.Writer, ctx.Request)
|
|
}
|
|
}
|
|
|
|
// BlockHandler will pass the call from /debug/pprof/block to pprof
|
|
func BlockHandler() gin.HandlerFunc {
|
|
return func(ctx *gin.Context) {
|
|
pprof.Handler("block").ServeHTTP(ctx.Writer, ctx.Request)
|
|
}
|
|
}
|
|
|
|
// ThreadCreateHandler will pass the call from /debug/pprof/threadcreate to pprof
|
|
func ThreadCreateHandler() gin.HandlerFunc {
|
|
return func(ctx *gin.Context) {
|
|
pprof.Handler("threadcreate").ServeHTTP(ctx.Writer, ctx.Request)
|
|
}
|
|
}
|
|
|
|
// CmdlineHandler will pass the call from /debug/pprof/cmdline to pprof
|
|
func CmdlineHandler() gin.HandlerFunc {
|
|
return func(ctx *gin.Context) {
|
|
pprof.Cmdline(ctx.Writer, ctx.Request)
|
|
}
|
|
}
|
|
|
|
// ProfileHandler will pass the call from /debug/pprof/profile to pprof
|
|
func ProfileHandler() gin.HandlerFunc {
|
|
return func(ctx *gin.Context) {
|
|
pprof.Profile(ctx.Writer, ctx.Request)
|
|
}
|
|
}
|
|
|
|
// SymbolHandler will pass the call from /debug/pprof/symbol to pprof
|
|
func SymbolHandler() gin.HandlerFunc {
|
|
return func(ctx *gin.Context) {
|
|
pprof.Symbol(ctx.Writer, ctx.Request)
|
|
}
|
|
}
|
|
|
|
// TraceHandler will pass the call from /debug/pprof/trace to pprof
|
|
func TraceHandler() gin.HandlerFunc {
|
|
return func(ctx *gin.Context) {
|
|
pprof.Trace(ctx.Writer, ctx.Request)
|
|
}
|
|
}
|
|
|
|
// MutexHandler will pass the call from /debug/pprof/mutex to pprof
|
|
func MutexHandler() gin.HandlerFunc {
|
|
return func(ctx *gin.Context) {
|
|
pprof.Handler("mutex").ServeHTTP(ctx.Writer, ctx.Request)
|
|
}
|
|
}
|