implement fallback (#104)

conf/pathconf.go

@@ -24,6 +24,7 @@ type PathConf struct {
 	SourceOnDemandStartTimeout time.Duration            `yaml:"sourceOnDemandStartTimeout"`
 	SourceOnDemandCloseAfter   time.Duration            `yaml:"sourceOnDemandCloseAfter"`
 	SourceRedirect             string                   `yaml:"sourceRedirect"`
+	Fallback                   string                   `yaml:"fallback"`
 	RunOnInit                  string                   `yaml:"runOnInit"`
 	RunOnInitRestart           bool                     `yaml:"runOnInitRestart"`
 	RunOnDemand                string                   `yaml:"runOnDemand"`
@@ -150,6 +151,17 @@ func (pconf *PathConf) fillAndCheck(name string) error {
 		pconf.SourceOnDemandCloseAfter = 10 * time.Second
 	}
 
+	if pconf.Fallback != "" {
+		u, err := url.Parse(pconf.Fallback)
+		if err != nil {
+			return fmt.Errorf("'%s' is not a valid rtsp url", pconf.Fallback)
+		}
+
+		if u.Scheme != "rtsp" {
+			return fmt.Errorf("'%s' is not a valid rtsp url", pconf.Fallback)
+		}
+	}
+
 	if pconf.PublishUser != "" {
 		if !reUserPass.MatchString(pconf.PublishUser) {
 			return fmt.Errorf("publish username contains unsupported characters (supported are %s)", userPassSupportedChars)

main_test.go

@@ -689,6 +689,44 @@ func TestRedirect(t *testing.T) {
 	require.Equal(t, 0, code)
 }
 
+func TestFallback(t *testing.T) {
+	p1, err := testProgram("paths:\n" +
+		"  path1:\n" +
+		"    fallback: rtsp://" + ownDockerIp + ":8554/path2\n" +
+		"  path2:\n")
+	require.NoError(t, err)
+	defer p1.close()
+
+	time.Sleep(1 * time.Second)
+
+	cnt1, err := newContainer("ffmpeg", "source", []string{
+		"-re",
+		"-stream_loop", "-1",
+		"-i", "/emptyvideo.ts",
+		"-c", "copy",
+		"-f", "rtsp",
+		"-rtsp_transport", "udp",
+		"rtsp://" + ownDockerIp + ":8554/path2",
+	})
+	require.NoError(t, err)
+	defer cnt1.close()
+
+	time.Sleep(1 * time.Second)
+
+	cnt2, err := newContainer("ffmpeg", "dest", []string{
+		"-rtsp_transport", "udp",
+		"-i", "rtsp://" + ownDockerIp + ":8554/path1",
+		"-vframes", "1",
+		"-f", "image2",
+		"-y", "/dev/null",
+	})
+	require.NoError(t, err)
+	defer cnt2.close()
+
+	code := cnt2.wait()
+	require.Equal(t, 0, code)
+}
+
 func TestRunOnDemand(t *testing.T) {
 	p1, err := testProgram("paths:\n" +
 		"  all:\n" +

path/path.go

@@ -612,6 +612,13 @@ func (pa *Path) onClientDescribe(c *client.Client) {
 		return
 
 	case sourceStateNotReady:
+		if pa.conf.Fallback != "" {
+			pa.addClient(c, clientStatePreRemove)
+			pa.removeClient(c)
+			c.OnPathDescribeData(nil, pa.conf.Fallback, nil)
+			return
+		}
+
 		pa.addClient(c, clientStatePreRemove)
 		pa.removeClient(c)
 		c.OnPathDescribeData(nil, "", fmt.Errorf("no one is publishing to path '%s'", pa.name))

rtsp-simple-server.yml

@@ -66,6 +66,9 @@ paths:
     # redirected to.
     sourceRedirect:
 
+    # fallback url to redirect clients to when nobody is publishing to this path
+    fallback:
+
     # username required to publish.
     publishUser:
     # password required to publish.