51 lines
1.2 KiB
Go
51 lines
1.2 KiB
Go
// Copyright (C) 2019 Storj Labs, Inc.
|
|
// See LICENSE for copying information.
|
|
|
|
package process
|
|
|
|
import (
|
|
"flag"
|
|
"fmt"
|
|
"net"
|
|
"net/http"
|
|
"net/http/pprof"
|
|
|
|
"go.uber.org/zap"
|
|
monkit "gopkg.in/spacemonkeygo/monkit.v2"
|
|
"gopkg.in/spacemonkeygo/monkit.v2/present"
|
|
)
|
|
|
|
var (
|
|
debugAddr = flag.String("debug.addr", "localhost:0", "address to listen on for debug endpoints")
|
|
)
|
|
|
|
func init() {
|
|
// zero out the http.DefaultServeMux net/http/pprof so unhelpfully
|
|
// side-effected.
|
|
*http.DefaultServeMux = http.ServeMux{}
|
|
}
|
|
|
|
func initDebug(logger *zap.Logger, r *monkit.Registry) (err error) {
|
|
var mux http.ServeMux
|
|
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)
|
|
mux.Handle("/mon/", http.StripPrefix("/mon", present.HTTP(r)))
|
|
mux.HandleFunc("/health", func(w http.ResponseWriter, r *http.Request) {
|
|
_, _ = fmt.Fprintln(w, "OK")
|
|
})
|
|
ln, err := net.Listen("tcp", *debugAddr)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
go func() {
|
|
err := (&http.Server{Handler: &mux}).Serve(ln)
|
|
if err != nil {
|
|
logger.Error("debug server died", zap.Error(err))
|
|
}
|
|
}()
|
|
return nil
|
|
}
|