mirror of
https://github.com/shtorm-7/sing-box-extended.git
synced 2026-06-15 15:52:02 +03:00
94 lines
2.3 KiB
Go
94 lines
2.3 KiB
Go
package profiler
|
|
|
|
import (
|
|
"context"
|
|
"net/http"
|
|
"net/http/pprof"
|
|
runtimePprof "runtime/pprof"
|
|
"strings"
|
|
|
|
"github.com/sagernet/sing-box/adapter"
|
|
boxService "github.com/sagernet/sing-box/adapter/service"
|
|
"github.com/sagernet/sing-box/common/listener"
|
|
C "github.com/sagernet/sing-box/constant"
|
|
"github.com/sagernet/sing-box/log"
|
|
"github.com/sagernet/sing-box/option"
|
|
E "github.com/sagernet/sing/common/exceptions"
|
|
N "github.com/sagernet/sing/common/network"
|
|
|
|
"github.com/go-chi/chi/v5"
|
|
)
|
|
|
|
func RegisterService(registry *boxService.Registry) {
|
|
boxService.Register[option.ProfilerServiceOptions](registry, C.TypeProfiler, NewService)
|
|
}
|
|
|
|
type Service struct {
|
|
boxService.Adapter
|
|
logger log.ContextLogger
|
|
listener *listener.Listener
|
|
server *http.Server
|
|
}
|
|
|
|
func NewService(ctx context.Context, logger log.ContextLogger, tag string, options option.ProfilerServiceOptions) (adapter.Service, error) {
|
|
return &Service{
|
|
Adapter: boxService.NewAdapter(C.TypeProfiler, tag),
|
|
logger: logger,
|
|
listener: listener.New(listener.Options{
|
|
Context: ctx,
|
|
Logger: logger,
|
|
Network: []string{N.NetworkTCP},
|
|
Listen: options.ListenOptions,
|
|
}),
|
|
}, nil
|
|
}
|
|
|
|
func (s *Service) Start(stage adapter.StartStage) error {
|
|
if stage != adapter.StartStateStart {
|
|
return nil
|
|
}
|
|
r := chi.NewMux()
|
|
r.Route("/debug/pprof", func(r chi.Router) {
|
|
r.HandleFunc("/", func(w http.ResponseWriter, req *http.Request) {
|
|
if !strings.HasSuffix(req.URL.Path, "/") {
|
|
http.Redirect(w, req, req.URL.Path+"/", http.StatusMovedPermanently)
|
|
return
|
|
}
|
|
pprof.Index(w, req)
|
|
})
|
|
r.HandleFunc("/cmdline", pprof.Cmdline)
|
|
r.HandleFunc("/profile", pprof.Profile)
|
|
r.HandleFunc("/symbol", pprof.Symbol)
|
|
r.HandleFunc("/trace", pprof.Trace)
|
|
for _, p := range runtimePprof.Profiles() {
|
|
name := p.Name()
|
|
r.Handle("/"+name, pprof.Handler(name))
|
|
}
|
|
r.HandleFunc("/*", pprof.Index)
|
|
})
|
|
|
|
s.server = &http.Server{
|
|
Handler: r,
|
|
}
|
|
tcpListener, err := s.listener.ListenTCP()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
s.logger.Info("profiler listening at ", tcpListener.Addr())
|
|
go func() {
|
|
err := s.server.Serve(tcpListener)
|
|
if err != nil && !E.IsClosed(err) {
|
|
s.logger.Error(E.Cause(err, "serve profiler"))
|
|
}
|
|
}()
|
|
return nil
|
|
}
|
|
|
|
func (s *Service) Close() error {
|
|
if s.server != nil {
|
|
_ = s.server.Close()
|
|
s.server = nil
|
|
}
|
|
return s.listener.Close()
|
|
}
|