diff --git a/pkg/middleware/logger.go b/pkg/middleware/logger.go index 1477997..75aabf2 100644 --- a/pkg/middleware/logger.go +++ b/pkg/middleware/logger.go @@ -124,6 +124,12 @@ func (lrw *loggingResponseWriter) WriteHeader(code int) { lrw.ResponseWriter.WriteHeader(code) } +func (lrw *loggingResponseWriter) Flush() { + if f, ok := lrw.ResponseWriter.(http.Flusher); ok { + f.Flush() + } +} + func newLoggingResponseWriter(w http.ResponseWriter) *loggingResponseWriter { // WriteHeader(int) is not called if our response implicitly returns 200 OK, so // we default to that status code.