Merge pull request #27 from ZSC714725/http-fmp4-3

update
q191201771 · Aug 10, 2023 · 4d72913 · 4d72913
2 parents d92ea58 + b101a2c
commit 4d72913
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/fmp4/http-fmp4/session.go b/fmp4/http-fmp4/session.go
@@ -213,14 +213,14 @@ 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)
 		if _, err = conn.Read(readBuf); err != nil {
 			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
 		}
 	}