ensure concurrency updated even without instance

sonroyaalmerol · Dec 15, 2024 · 876534a · 876534a
1 parent 99d9528
commit 876534a
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/proxy/proxy_stream.go b/proxy/proxy_stream.go
@@ -73,10 +73,13 @@ func (instance *StreamInstance) ProxyStream(ctx context.Context, m3uIndex int, r
 		return
 	}
 
-	instance.Cm.UpdateConcurrency(m3uIndex, true)
-	defer instance.Cm.UpdateConcurrency(m3uIndex, false)
+	cm := instance.Cm
+
+	cm.UpdateConcurrency(m3uIndex, true)
 
 	defer func() {
+		cm.UpdateConcurrency(m3uIndex, false)
+
 		buffer = nil
 		if flusher, ok := w.(http.Flusher); ok {
 			flusher.Flush()