xmengnet/mihomo
1
1
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

chore: add pprof api, when log-level is debug

Browse source
This commit is contained in:
Skyxim 2023-02-23 23:30:53 +08:00
parent d5d62a4ffd
commit a1d008e6f0
2 changed files with 16 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
hub/hub.go
View file

@ -4,6 +4,7 @@ import (
"github.com/Dreamacro/clash/config" "github.com/Dreamacro/clash/config"
"github.com/Dreamacro/clash/hub/executor" "github.com/Dreamacro/clash/hub/executor"
"github.com/Dreamacro/clash/hub/route" "github.com/Dreamacro/clash/hub/route"
"github.com/Dreamacro/clash/log"
) )
type Option func(*config.Config) type Option func(*config.Config)
@ -43,7 +44,7 @@ func Parse(options ...Option) error {
if cfg.General.ExternalController != "" { if cfg.General.ExternalController != "" {
go route.Start(cfg.General.ExternalController, cfg.General.ExternalControllerTLS, go route.Start(cfg.General.ExternalController, cfg.General.ExternalControllerTLS,
cfg.General.Secret, cfg.TLS.Certificate, cfg.TLS.PrivateKey) cfg.General.Secret, cfg.TLS.Certificate, cfg.TLS.PrivateKey,cfg.General.LogLevel==log.DEBUG)
} }
executor.ApplyConfig(cfg, true) executor.ApplyConfig(cfg, true)

16
hub/route/server.go
View file

@ -5,6 +5,7 @@ import (
"crypto/tls" "crypto/tls"
"encoding/json" "encoding/json"
"net/http" "net/http"
"runtime/debug"
"strings" "strings"
"time" "time"
@ -13,8 +14,8 @@ import (
C "github.com/Dreamacro/clash/constant" C "github.com/Dreamacro/clash/constant"
"github.com/Dreamacro/clash/log" "github.com/Dreamacro/clash/log"
"github.com/Dreamacro/clash/tunnel/statistic" "github.com/Dreamacro/clash/tunnel/statistic"
"github.com/go-chi/chi/v5" "github.com/go-chi/chi/v5"
"github.com/go-chi/chi/v5/middleware"
"github.com/go-chi/cors" "github.com/go-chi/cors"
"github.com/go-chi/render" "github.com/go-chi/render"
"github.com/gorilla/websocket" "github.com/gorilla/websocket"
@ -43,7 +44,7 @@ func SetUIPath(path string) {
} }
func Start(addr string, tlsAddr string, secret string, func Start(addr string, tlsAddr string, secret string,
certificat, privateKey string) { certificat, privateKey string, isDebug bool) {
if serverAddr != "" { if serverAddr != "" {
return return
} }
@ -59,6 +60,17 @@ func Start(addr string, tlsAddr string, secret string,
MaxAge: 300, MaxAge: 300,
}) })
r.Use(corsM.Handler) r.Use(corsM.Handler)
if isDebug {
r.Mount("/debug", func() http.Handler {
r := chi.NewRouter()
r.Put("/gc", func(w http.ResponseWriter, r *http.Request) {
debug.FreeOSMemory()
})
handler := middleware.Profiler
r.Mount("/", handler())
return r
}())
}
r.Group(func(r chi.Router) { r.Group(func(r chi.Router) {
r.Use(authentication) r.Use(authentication)
r.Get("/", hello) r.Get("/", hello)