fix: stop publisher and subscriber when conn closed

This commit is contained in:
langhuihui
2023-07-07 14:47:43 +08:00
parent e1c1777455
commit 11240491a7

View File

@@ -37,6 +37,14 @@ func (config *RTMPConfig) ServeTCP(conn net.Conn) {
defer conn.Close()
senders := make(map[uint32]*RTMPSubscriber)
receivers := make(map[uint32]*RTMPReceiver)
defer func() {
for _, sender := range senders {
sender.Stop()
}
for _, receiver := range receivers {
receiver.Stop()
}
}()
nc := NewNetConnection(conn)
ctx, cancel := context.WithCancel(engine.Engine)
defer cancel()