mirror of
https://github.com/Mrs4s/go-cqhttp.git
synced 2025-05-04 19:17:37 +08:00
26a7a1f0b6
* feat(server): add RegisterCustom * feat(config): seprate config & server * fix: make lint happy * fix: make lint happy * fix: ParseEnv nil pointer error * typo(config): generateConfig hint * fix(config): panic on range overflow
76 lines
2.0 KiB
Go
76 lines
2.0 KiB
Go
// Package pprof provide pprof server of go-cqhttp
|
|
package pprof
|
|
|
|
import (
|
|
"fmt"
|
|
"net/http"
|
|
"net/http/pprof"
|
|
"os"
|
|
"time"
|
|
|
|
log "github.com/sirupsen/logrus"
|
|
"gopkg.in/yaml.v3"
|
|
|
|
"github.com/Mrs4s/go-cqhttp/coolq"
|
|
"github.com/Mrs4s/go-cqhttp/modules/config"
|
|
"github.com/Mrs4s/go-cqhttp/modules/servers"
|
|
)
|
|
|
|
const pprofDefault = ` # pprof 性能分析服务器, 一般情况下不需要启用.
|
|
# 如果遇到性能问题请上传报告给开发者处理
|
|
# 注意: pprof服务不支持中间件、不支持鉴权. 请不要开放到公网
|
|
- pprof:
|
|
# pprof服务器监听地址
|
|
host: 127.0.0.1
|
|
# pprof服务器监听端口
|
|
port: 7700
|
|
`
|
|
|
|
// pprofServer pprof性能分析服务器相关配置
|
|
type pprofServer struct {
|
|
Disabled bool `yaml:"disabled"`
|
|
Host string `yaml:"host"`
|
|
Port int `yaml:"port"`
|
|
}
|
|
|
|
func init() {
|
|
config.AddServer(&config.Server{
|
|
Brief: "pprof 性能分析服务器",
|
|
Default: pprofDefault,
|
|
})
|
|
}
|
|
|
|
// runPprof 启动 pprof 性能分析服务器
|
|
func runPprof(_ *coolq.CQBot, node yaml.Node) {
|
|
var conf pprofServer
|
|
switch err := node.Decode(&conf); {
|
|
case err != nil:
|
|
log.Warn("读取pprof配置失败 :", err)
|
|
fallthrough
|
|
case conf.Disabled:
|
|
return
|
|
}
|
|
|
|
addr := fmt.Sprintf("%s:%d", conf.Host, conf.Port)
|
|
mux := http.NewServeMux()
|
|
mux.HandleFunc("/debug/pprof/", pprof.Index)
|
|
mux.HandleFunc("/debug/pprof/cmdline", pprof.Cmdline)
|
|
mux.HandleFunc("/debug/pprof/profile", pprof.Profile)
|
|
mux.HandleFunc("/debug/pprof/symbol", pprof.Symbol)
|
|
mux.HandleFunc("/debug/pprof/trace", pprof.Trace)
|
|
server := http.Server{Addr: addr, Handler: mux}
|
|
log.Infof("pprof debug 服务器已启动: %v/debug/pprof", addr)
|
|
log.Warnf("警告: pprof 服务不支持鉴权, 请不要运行在公网.")
|
|
if err := server.ListenAndServe(); err != nil && err != http.ErrServerClosed {
|
|
log.Error(err)
|
|
log.Infof("pprof 服务启动失败, 请检查端口是否被占用.")
|
|
log.Warnf("将在五秒后退出.")
|
|
time.Sleep(time.Second * 5)
|
|
os.Exit(1)
|
|
}
|
|
}
|
|
|
|
func init() {
|
|
servers.Register("pprof", runPprof)
|
|
}
|