mediamtx/internal/pprof/pprof.go

53 lines
726 B
Go

package pprof
import (
"context"
"net"
"net/http"
_ "net/http/pprof"
)
const (
address = ":9999"
)
type Parent interface {
Log(string, ...interface{})
}
type Pprof struct {
listener net.Listener
server *http.Server
}
func New(parent Parent) (*Pprof, error) {
listener, err := net.Listen("tcp", address)
if err != nil {
return nil, err
}
pp := &Pprof{
listener: listener,
}
pp.server = &http.Server{
Handler: http.DefaultServeMux,
}
parent.Log("[pprof] opened on " + address)
go pp.run()
return pp, nil
}
func (pp *Pprof) Close() {
pp.server.Shutdown(context.Background())
}
func (pp *Pprof) run() {
err := pp.server.Serve(pp.listener)
if err != http.ErrServerClosed {
panic(err)
}
}