Fix stream consumer stuck after closing bitrate worker (#9)

HYBIOT · Jun 11, 2024 · d18295b · d18295b
1 parent 00e531e
commit d18295b
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/pkg/isapi/custom.go b/pkg/isapi/custom.go
@@ -3,6 +3,6 @@ package isapi
 import "github.com/AlexxIT/go2rtc/pkg/custom"
 
 func (c *Client) startBitrateWorker() {
-	c.stopBitrateWorker = make(chan struct{})
+	c.stopBitrateWorker = make(chan struct{}, 1)
 	go custom.StartBitrateWorker(&c.send, &c.bitrate, c.stopBitrateWorker)
 }
diff --git a/pkg/mp4/custom.go b/pkg/mp4/custom.go
@@ -3,6 +3,6 @@ package mp4
 import "github.com/AlexxIT/go2rtc/pkg/custom"
 
 func (c *Consumer) startBitrateWorker() {
-	c.stopBitrateWorker = make(chan struct{})
+	c.stopBitrateWorker = make(chan struct{}, 1)
 	go custom.StartBitrateWorker(&c.SuperConsumer.Send, &c.SuperConsumer.Bitrate, c.stopBitrateWorker)
 }
diff --git a/pkg/webrtc/custom.go b/pkg/webrtc/custom.go
@@ -5,6 +5,6 @@ import (
 )
 
 func (c *Conn) startBitrateWorker() {
-	c.stopBitrateWorker = make(chan struct{})
+	c.stopBitrateWorker = make(chan struct{}, 1)
 	go custom.StartBitrateWorker(&c.send, &c.bitrate, c.stopBitrateWorker)
 }