make responseWriterWithSize satisfy Flusher()

TykTechnologies · Nov 21, 2024 · 2d2c049 · 2d2c049
1 parent 545b959
commit 2d2c049
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/trace/handler.go b/trace/handler.go
@@ -21,6 +21,10 @@ func (rw *responseWriterWithSize) Write(p []byte) (int, error) {
 	return n, err
 }
 
+func (rw *responseWriterWithSize) Flush() {
+	rw.ResponseWriter.(http.Flusher).Flush()
+}
+
 // NewHTTPHandler wraps the provided http.Handler with one that starts a span
 // and injects the span context into the outbound request headers.
 // You need to initialize the TracerProvider first since it utilizes the underlying