From a309893ea032702d2a0d02c66d5442654d7295e4 Mon Sep 17 00:00:00 2001
From: Itay Ariel <itay@cnvrg.io>
Date: Mon, 24 Jul 2023 17:40:41 +0300
Subject: [PATCH] feat: flush writes to http response on every chunk

Signed-off-by: Itay Ariel <itay@cnvrg.io>
---
 interceptor/middleware/loggingresponsewriter.go | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/interceptor/middleware/loggingresponsewriter.go b/interceptor/middleware/loggingresponsewriter.go
index 0ebb9c6c7..9c893f1c1 100644
--- a/interceptor/middleware/loggingresponsewriter.go
+++ b/interceptor/middleware/loggingresponsewriter.go
@@ -32,6 +32,9 @@ func (lrw *loggingResponseWriter) Header() http.Header {
 
 func (lrw *loggingResponseWriter) Write(bytes []byte) (int, error) {
 	n, err := lrw.downstreamResponseWriter.Write(bytes)
+	if f, ok := lrw.downstreamResponseWriter.(http.Flusher); ok {
+		f.Flush()
+	}
 
 	lrw.bytesWritten += n