Merge pull request #110 from sonroyaalmerol/flush-buffer

Clear buffer and flush out writer on return
sonroyaalmerol · Aug 25, 2024 · 5d6d841 · 5d6d841
2 parents 5dbc7d8 + 02ad7c1
commit 5d6d841
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/stream_handler.go b/stream_handler.go
@@ -96,6 +96,13 @@ func proxyStream(m3uIndex int, resp *http.Response, r *http.Request, w http.Resp
 		buffer = make([]byte, bufferMbInt*1024*1024)
 	}
 
+	defer func() {
+		buffer = nil
+		if flusher, ok := w.(http.Flusher); ok {
+			flusher.Flush()
+		}
+	}()
+
 	for {
 		n, err := resp.Body.Read(buffer)
 		if err != nil {