Add timeout query param to RTSP incoming source #1118

AlexxIT · May 24, 2024 · d9d2bdf · d9d2bdf
1 parent b3e9ed2
commit d9d2bdf
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 1 deletion.
diff --git a/internal/rtsp/rtsp.go b/internal/rtsp/rtsp.go
@@ -210,6 +210,11 @@ func tcpHandler(conn *rtsp.Conn) {
 				return
 			}
 
+			query := conn.URL.Query()
+			if s := query.Get("timeout"); s != "" {
+				conn.Timeout = core.Atoi(s)
+			}
+
 			log.Debug().Str("stream", name).Msg("[rtsp] new producer")
 
 			stream.AddProducer(conn)

diff --git a/pkg/rtsp/conn.go b/pkg/rtsp/conn.go
@@ -124,7 +124,11 @@ func (c *Conn) Handle() (err error) {
 
 	case core.ModePassiveProducer:
 		// polling frames from remote RTSP Client (ex FFmpeg)
-		timeout = time.Second * 15
+		if c.Timeout == 0 {
+			timeout = time.Second * 15
+		} else {
+			timeout = time.Second * time.Duration(c.Timeout)
+		}
 
 	case core.ModePassiveConsumer:
 		// pushing frames to remote RTSP Client (ex VLC)