From c124ccff0e6604b76832ddbe0531e8c8c49967da Mon Sep 17 00:00:00 2001
From: Son Roy Almerol <github@snry.me>
Date: Sat, 24 Aug 2024 17:01:54 -0400
Subject: [PATCH] clear buffer and flush out writer on return

---
 stream_handler.go | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/stream_handler.go b/stream_handler.go
index 6387630..5cd2fc3 100644
--- 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 {