close statusChan on ProxyStream instead

sonroyaalmerol · Dec 14, 2024 · 4562ff2 · 4562ff2
1 parent 2bd916d
commit 4562ff2
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/handlers/stream_handler.go b/handlers/stream_handler.go
@@ -73,7 +73,6 @@ func StreamHandler(w http.ResponseWriter, r *http.Request, cm *store.Concurrency
 			}
 
 			exitStatus := make(chan int)
-			defer close(exitStatus)
 
 			utils.SafeLogf("Proxying %s to %s\n", r.RemoteAddr, selectedUrl)
 			go stream.ProxyStream(ctx, selectedIndex, resp, r, w, exitStatus)

diff --git a/proxy/proxy_stream.go b/proxy/proxy_stream.go
@@ -14,6 +14,8 @@ import (
 )
 
 func (instance *StreamInstance) ProxyStream(ctx context.Context, m3uIndex int, resp *http.Response, r *http.Request, w http.ResponseWriter, statusChan chan int) {
+	defer close(statusChan)
+
 	bufferMbInt, err := strconv.Atoi(os.Getenv("BUFFER_MB"))
 	if err != nil || bufferMbInt < 0 {
 		bufferMbInt = 0