From 11240491a7793999dc36bc21ba4ed2383ae33eeb Mon Sep 17 00:00:00 2001
From: langhuihui <178529795@qq.com>
Date: Fri, 7 Jul 2023 14:47:43 +0800
Subject: [PATCH] fix: stop publisher and subscriber when conn closed

---
 server.go | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/server.go b/server.go
index e56dcf4..28915b0 100644
--- a/server.go
+++ b/server.go
@@ -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()