mirror of
https://github.com/aler9/rtsp-simple-server
synced 2025-10-17 21:23:46 +08:00
move static sources into dedicated package (#2616)
This commit is contained in:
Alessandro Ros
262
internal/core/static_source_handler.go
Normal file
262
internal/core/static_source_handler.go
Normal file
@@ -0,0 +1,262 @@
|
||||
package core
|
||||
|
||||
import (
|
||||
"context"
|
||||
"fmt"
|
||||
"strings"
|
||||
"time"
|
||||
|
||||
"github.com/bluenviron/mediamtx/internal/conf"
|
||||
"github.com/bluenviron/mediamtx/internal/defs"
|
||||
"github.com/bluenviron/mediamtx/internal/logger"
|
||||
hlssource "github.com/bluenviron/mediamtx/internal/staticsources/hls"
|
||||
rpicamerasource "github.com/bluenviron/mediamtx/internal/staticsources/rpicamera"
|
||||
rtmpsource "github.com/bluenviron/mediamtx/internal/staticsources/rtmp"
|
||||
rtspsource "github.com/bluenviron/mediamtx/internal/staticsources/rtsp"
|
||||
srtsource "github.com/bluenviron/mediamtx/internal/staticsources/srt"
|
||||
udpsource "github.com/bluenviron/mediamtx/internal/staticsources/udp"
|
||||
webrtcsource "github.com/bluenviron/mediamtx/internal/staticsources/webrtc"
|
||||
)
|
||||
|
||||
const (
|
||||
staticSourceHandlerRetryPause = 5 * time.Second
|
||||
)
|
||||
|
||||
type staticSourceHandlerParent interface {
|
||||
logger.Writer
|
||||
staticSourceHandlerSetReady(context.Context, defs.PathSourceStaticSetReadyReq)
|
||||
staticSourceHandlerSetNotReady(context.Context, defs.PathSourceStaticSetNotReadyReq)
|
||||
}
|
||||
|
||||
// staticSourceHandler is a static source handler.
|
||||
type staticSourceHandler struct {
|
||||
conf *conf.Path
|
||||
parent staticSourceHandlerParent
|
||||
|
||||
ctx context.Context
|
||||
ctxCancel func()
|
||||
instance defs.StaticSource
|
||||
running bool
|
||||
|
||||
// in
|
||||
chReloadConf chan *conf.Path
|
||||
chInstanceSetReady chan defs.PathSourceStaticSetReadyReq
|
||||
chInstanceSetNotReady chan defs.PathSourceStaticSetNotReadyReq
|
||||
|
||||
// out
|
||||
done chan struct{}
|
||||
}
|
||||
|
||||
func newStaticSourceHandler(
|
||||
cnf *conf.Path,
|
||||
readTimeout conf.StringDuration,
|
||||
writeTimeout conf.StringDuration,
|
||||
writeQueueSize int,
|
||||
parent staticSourceHandlerParent,
|
||||
) *staticSourceHandler {
|
||||
s := &staticSourceHandler{
|
||||
conf: cnf,
|
||||
parent: parent,
|
||||
chReloadConf: make(chan *conf.Path),
|
||||
chInstanceSetReady: make(chan defs.PathSourceStaticSetReadyReq),
|
||||
chInstanceSetNotReady: make(chan defs.PathSourceStaticSetNotReadyReq),
|
||||
}
|
||||
|
||||
switch {
|
||||
case strings.HasPrefix(cnf.Source, "rtsp://") ||
|
||||
strings.HasPrefix(cnf.Source, "rtsps://"):
|
||||
s.instance = &rtspsource.Source{
|
||||
ReadTimeout: readTimeout,
|
||||
WriteTimeout: writeTimeout,
|
||||
WriteQueueSize: writeQueueSize,
|
||||
Parent: s,
|
||||
}
|
||||
|
||||
case strings.HasPrefix(cnf.Source, "rtmp://") ||
|
||||
strings.HasPrefix(cnf.Source, "rtmps://"):
|
||||
s.instance = &rtmpsource.Source{
|
||||
ReadTimeout: readTimeout,
|
||||
WriteTimeout: writeTimeout,
|
||||
Parent: s,
|
||||
}
|
||||
|
||||
case strings.HasPrefix(cnf.Source, "http://") ||
|
||||
strings.HasPrefix(cnf.Source, "https://"):
|
||||
s.instance = &hlssource.Source{
|
||||
Parent: s,
|
||||
}
|
||||
|
||||
case strings.HasPrefix(cnf.Source, "udp://"):
|
||||
s.instance = &udpsource.Source{
|
||||
ReadTimeout: readTimeout,
|
||||
Parent: s,
|
||||
}
|
||||
|
||||
case strings.HasPrefix(cnf.Source, "srt://"):
|
||||
s.instance = &srtsource.Source{
|
||||
ReadTimeout: readTimeout,
|
||||
Parent: s,
|
||||
}
|
||||
|
||||
case strings.HasPrefix(cnf.Source, "whep://") ||
|
||||
strings.HasPrefix(cnf.Source, "wheps://"):
|
||||
s.instance = &webrtcsource.Source{
|
||||
ReadTimeout: readTimeout,
|
||||
Parent: s,
|
||||
}
|
||||
|
||||
case cnf.Source == "rpiCamera":
|
||||
s.instance = &rpicamerasource.Source{
|
||||
Parent: s,
|
||||
}
|
||||
}
|
||||
|
||||
return s
|
||||
}
|
||||
|
||||
func (s *staticSourceHandler) close(reason string) {
|
||||
s.stop(reason)
|
||||
}
|
||||
|
||||
func (s *staticSourceHandler) start(onDemand bool) {
|
||||
if s.running {
|
||||
panic("should not happen")
|
||||
}
|
||||
|
||||
s.running = true
|
||||
s.instance.Log(logger.Info, "started%s",
|
||||
func() string {
|
||||
if onDemand {
|
||||
return " on demand"
|
||||
}
|
||||
return ""
|
||||
}())
|
||||
|
||||
s.ctx, s.ctxCancel = context.WithCancel(context.Background())
|
||||
s.done = make(chan struct{})
|
||||
|
||||
go s.run()
|
||||
}
|
||||
|
||||
func (s *staticSourceHandler) stop(reason string) {
|
||||
if !s.running {
|
||||
panic("should not happen")
|
||||
}
|
||||
|
||||
s.running = false
|
||||
s.instance.Log(logger.Info, "stopped: %s", reason)
|
||||
|
||||
s.ctxCancel()
|
||||
|
||||
// we must wait since s.ctx is not thread safe
|
||||
<-s.done
|
||||
}
|
||||
|
||||
func (s *staticSourceHandler) Log(level logger.Level, format string, args ...interface{}) {
|
||||
s.parent.Log(level, format, args...)
|
||||
}
|
||||
|
||||
func (s *staticSourceHandler) run() {
|
||||
defer close(s.done)
|
||||
|
||||
var runCtx context.Context
|
||||
var runCtxCancel func()
|
||||
runErr := make(chan error)
|
||||
runReloadConf := make(chan *conf.Path)
|
||||
|
||||
recreate := func() {
|
||||
runCtx, runCtxCancel = context.WithCancel(context.Background())
|
||||
go func() {
|
||||
runErr <- s.instance.Run(defs.StaticSourceRunParams{
|
||||
Context: runCtx,
|
||||
Conf: s.conf,
|
||||
ReloadConf: runReloadConf,
|
||||
})
|
||||
}()
|
||||
}
|
||||
|
||||
recreate()
|
||||
|
||||
recreating := false
|
||||
recreateTimer := newEmptyTimer()
|
||||
|
||||
for {
|
||||
select {
|
||||
case err := <-runErr:
|
||||
runCtxCancel()
|
||||
s.instance.Log(logger.Error, err.Error())
|
||||
recreating = true
|
||||
recreateTimer = time.NewTimer(staticSourceHandlerRetryPause)
|
||||
|
||||
case req := <-s.chInstanceSetReady:
|
||||
s.parent.staticSourceHandlerSetReady(s.ctx, req)
|
||||
|
||||
case req := <-s.chInstanceSetNotReady:
|
||||
s.parent.staticSourceHandlerSetNotReady(s.ctx, req)
|
||||
|
||||
case newConf := <-s.chReloadConf:
|
||||
s.conf = newConf
|
||||
if !recreating {
|
||||
cReloadConf := runReloadConf
|
||||
cInnerCtx := runCtx
|
||||
go func() {
|
||||
select {
|
||||
case cReloadConf <- newConf:
|
||||
case <-cInnerCtx.Done():
|
||||
}
|
||||
}()
|
||||
}
|
||||
|
||||
case <-recreateTimer.C:
|
||||
recreate()
|
||||
recreating = false
|
||||
|
||||
case <-s.ctx.Done():
|
||||
if !recreating {
|
||||
runCtxCancel()
|
||||
<-runErr
|
||||
}
|
||||
return
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func (s *staticSourceHandler) reloadConf(newConf *conf.Path) {
|
||||
select {
|
||||
case s.chReloadConf <- newConf:
|
||||
case <-s.ctx.Done():
|
||||
}
|
||||
}
|
||||
|
||||
// APISourceDescribe instanceements source.
|
||||
func (s *staticSourceHandler) APISourceDescribe() defs.APIPathSourceOrReader {
|
||||
return s.instance.APISourceDescribe()
|
||||
}
|
||||
|
||||
// setReady is called by a staticSource.
|
||||
func (s *staticSourceHandler) SetReady(req defs.PathSourceStaticSetReadyReq) defs.PathSourceStaticSetReadyRes {
|
||||
req.Res = make(chan defs.PathSourceStaticSetReadyRes)
|
||||
select {
|
||||
case s.chInstanceSetReady <- req:
|
||||
res := <-req.Res
|
||||
|
||||
if res.Err == nil {
|
||||
s.instance.Log(logger.Info, "ready: %s", mediaInfo(req.Desc.Medias))
|
||||
}
|
||||
|
||||
return res
|
||||
|
||||
case <-s.ctx.Done():
|
||||
return defs.PathSourceStaticSetReadyRes{Err: fmt.Errorf("terminated")}
|
||||
}
|
||||
}
|
||||
|
||||
// setNotReady is called by a staticSource.
|
||||
func (s *staticSourceHandler) SetNotReady(req defs.PathSourceStaticSetNotReadyReq) {
|
||||
req.Res = make(chan struct{})
|
||||
select {
|
||||
case s.chInstanceSetNotReady <- req:
|
||||
<-req.Res
|
||||
case <-s.ctx.Done():
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user