支持注册pprof相关路由

This commit is contained in:
2025-02-21 15:16:43 +08:00
parent bc94242d12
commit 4909e75928
5 changed files with 23 additions and 3 deletions

View File

@ -12,6 +12,7 @@ import (
apiDoc "git.zhangdeman.cn/gateway/api-doc"
"git.zhangdeman.cn/zhangdeman/consts"
"git.zhangdeman.cn/zhangdeman/gin/middleware"
"github.com/gin-contrib/pprof"
"net/http"
"strings"
@ -80,6 +81,10 @@ func NewServer(port int, optionList ...SetServerOptionFunc) *server {
// 启用全局中间件
r.Use(option.globalMiddlewareList...)
}
// 启用pprof, 注册相关路由
if option.enablePprof {
pprof.Register(r)
}
return &server{
router: r,
uiInstance: apiDoc.NewSwaggerUI(option.serverInfo, option.serverList, option.swaggerUiTheme),