From b101a2c79c5de9202d506edf6d93fac8f9ea6e33 Mon Sep 17 00:00:00 2001
From: ZSC714725 <zang199193@163.com>
Date: Thu, 10 Aug 2023 22:20:22 +0800
Subject: [PATCH] update

---
 fmp4/http-fmp4/session.go | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/fmp4/http-fmp4/session.go b/fmp4/http-fmp4/session.go
index 8a7cb75..2c71927 100644
--- a/fmp4/http-fmp4/session.go
+++ b/fmp4/http-fmp4/session.go
@@ -213,7 +213,6 @@ func (session *HttpFmp4Session) handleSession(c *gin.Context) {
 		return
 	}
 	session.hooks.AddConsumer(session.subscriberId, session)
-	defer session.hooks.RemoveConsumer(session.subscriberId)
 	connCloseErr := make(chan error, 1)
 	go func() {
 		readBuf := make([]byte, 1024)
@@ -221,6 +220,7 @@ func (session *HttpFmp4Session) handleSession(c *gin.Context) {
 			connCloseErr <- err
 		}
 	}()
+
 	session.muxer.WriteInitSegment(conn)
 	for {
 		writeFragment := func(data []byte) error {
@@ -272,6 +272,7 @@ func (session *HttpFmp4Session) handleSession(c *gin.Context) {
 			}
 		case <-connCloseErr:
 			session.OnStop()
+			return
 		}
 	}