mediamtx/internal/pprof/pprof.go

61 lines
948 B
Go

package pprof
import (
"context"
"net"
"net/http"
// start pprof
_ "net/http/pprof"
"github.com/aler9/rtsp-simple-server/internal/logger"
)
// Parent is implemented by program.
type Parent interface {
Log(logger.Level, string, ...interface{})
}
// PPROF is a performance metrics exporter.
type PPROF struct {
listener net.Listener
server *http.Server
}
// New allocates a PPROF.
func New(
address string,
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(logger.Info, "[pprof] opened on "+address)
go pp.run()
return pp, nil
}
// Close closes a PPROF.
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)
}
}