1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
|
package main
import (
"expvar"
"net"
"net/http"
"net/http/pprof"
"runtime"
"time"
"github.com/moby/buildkit/util/bklog"
"github.com/prometheus/client_golang/prometheus/promhttp"
"golang.org/x/net/trace"
)
func setupDebugHandlers(addr string) error {
m := http.NewServeMux()
m.Handle("/debug/vars", expvar.Handler())
m.Handle("/debug/pprof/", http.HandlerFunc(pprof.Index))
m.Handle("/debug/pprof/cmdline", http.HandlerFunc(pprof.Cmdline))
m.Handle("/debug/pprof/profile", http.HandlerFunc(pprof.Profile))
m.Handle("/debug/pprof/symbol", http.HandlerFunc(pprof.Symbol))
m.Handle("/debug/pprof/trace", http.HandlerFunc(pprof.Trace))
m.Handle("/debug/requests", http.HandlerFunc(trace.Traces))
m.Handle("/debug/events", http.HandlerFunc(trace.Events))
m.Handle("/debug/gc", http.HandlerFunc(func(rw http.ResponseWriter, req *http.Request) {
runtime.GC()
bklog.G(req.Context()).Debugf("triggered GC from debug endpoint")
}))
m.Handle("/metrics", promhttp.Handler())
// setting debugaddr is opt-in. permission is defined by listener address
trace.AuthRequest = func(_ *http.Request) (bool, bool) {
return true, true
}
l, err := net.Listen("tcp", addr)
if err != nil {
return err
}
server := &http.Server{
Addr: l.Addr().String(),
Handler: m,
ReadHeaderTimeout: time.Minute,
}
bklog.L.Debugf("debug handlers listening at %s", addr)
go func() {
if err := server.Serve(l); err != nil {
bklog.L.Errorf("failed to serve debug handlers: %v", err)
}
}()
return nil
}
|
