forked from External/mediamtx
64 lines
1.2 KiB
Go
64 lines
1.2 KiB
Go
package core
|
|
|
|
import (
|
|
"context"
|
|
"log"
|
|
"net"
|
|
"net/http"
|
|
"time"
|
|
|
|
// start pprof
|
|
_ "net/http/pprof"
|
|
|
|
"github.com/bluenviron/mediamtx/internal/conf"
|
|
"github.com/bluenviron/mediamtx/internal/logger"
|
|
)
|
|
|
|
type pprofParent interface {
|
|
logger.Writer
|
|
}
|
|
|
|
type pprof struct {
|
|
parent pprofParent
|
|
|
|
ln net.Listener
|
|
httpServer *http.Server
|
|
}
|
|
|
|
func newPPROF(
|
|
address string,
|
|
readTimeout conf.StringDuration,
|
|
parent pprofParent,
|
|
) (*pprof, error) {
|
|
ln, err := net.Listen(restrictNetwork("tcp", address))
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
pp := &pprof{
|
|
parent: parent,
|
|
ln: ln,
|
|
}
|
|
|
|
pp.httpServer = &http.Server{
|
|
Handler: http.DefaultServeMux,
|
|
ReadHeaderTimeout: time.Duration(readTimeout),
|
|
ErrorLog: log.New(&nilWriter{}, "", 0),
|
|
}
|
|
|
|
pp.Log(logger.Info, "listener opened on "+address)
|
|
|
|
go pp.httpServer.Serve(pp.ln)
|
|
|
|
return pp, nil
|
|
}
|
|
|
|
func (pp *pprof) close() {
|
|
pp.Log(logger.Info, "listener is closing")
|
|
pp.httpServer.Shutdown(context.Background())
|
|
pp.ln.Close() // in case Shutdown() is called before Serve()
|
|
}
|
|
|
|
func (pp *pprof) Log(level logger.Level, format string, args ...interface{}) {
|
|
pp.parent.Log(level, "[pprof] "+format, args...)
|
|
}
|