LogalDeveloper/owncast
LogalDeveloper/owncast
0
Fork 0
Code

Break rtmp read loop after disconnect

Browse Source
This commit is contained in:
Gabe Kangas 2020-10-07 19:59:55 -07:00
parent 06f8a01d5e
commit bcf4cb09e7
1 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
core/rtmp/rtmp.go
View File

@ -14,12 +14,12 @@ import (
"github.com/nareix/joy5/format/flv/flvio" "github.com/nareix/joy5/format/flv/flvio"
log "github.com/sirupsen/logrus" log "github.com/sirupsen/logrus"
"github.com/nareix/joy5/format/rtmp"
"github.com/owncast/owncast/config" "github.com/owncast/owncast/config"
"github.com/owncast/owncast/core" "github.com/owncast/owncast/core"
"github.com/owncast/owncast/core/ffmpeg" "github.com/owncast/owncast/core/ffmpeg"
"github.com/owncast/owncast/models" "github.com/owncast/owncast/models"
"github.com/owncast/owncast/utils" "github.com/owncast/owncast/utils"
"github.com/nareix/joy5/format/rtmp"
) )
var ( var (
@ -131,10 +131,14 @@ func HandleConn(c *rtmp.Conn, nc net.Conn) {
w := flv.NewMuxer(f) w := flv.NewMuxer(f)
for { for {
if !_isConnected {
break
}
pkt, err := c.ReadPacket() pkt, err := c.ReadPacket()
if err == io.EOF { if err == io.EOF {
handleDisconnect(nc) handleDisconnect(nc)
return break
} }
if err := w.WritePacket(pkt); err != nil { if err := w.WritePacket(pkt); err != nil {