add cancellable context for registerForStats call

laurfb · Oct 5, 2021 · 1cae98d · 1cae98d
1 parent e8b97b5
commit 1cae98d
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/srt/server.go b/srt/server.go
@@ -196,7 +196,9 @@ func (s *ServerImpl) Handle(ctx context.Context, sock *srtgo.SrtSocket, addr *ne
 		streamid: &streamid,
 	}
 
-	s.registerForStats(ctx, conn)
+	subctx, cancel := context.WithCancel(ctx)
+	defer cancel()
+	s.registerForStats(subctx, conn)
 
 	switch streamid.Mode() {
 	case stream.ModePlay: