From f910bd4fce179d0ce8d53440d0093f197d82ad34 Mon Sep 17 00:00:00 2001
From: Alex X <alexey.khit@gmail.com>
Date: Sat, 16 Sep 2023 11:09:46 +0300
Subject: [PATCH] Add auto Flush to core WriteBuffer

---
 pkg/core/writebuffer.go | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/pkg/core/writebuffer.go b/pkg/core/writebuffer.go
index edc72e9e..50ad013b 100644
--- a/pkg/core/writebuffer.go
+++ b/pkg/core/writebuffer.go
@@ -3,6 +3,7 @@ package core
 import (
 	"bytes"
 	"io"
+	"net/http"
 	"sync"
 )
 
@@ -32,6 +33,8 @@ func (w *WriteBuffer) Write(p []byte) (n int, err error) {
 	} else if n, err = w.Writer.Write(p); err != nil {
 		w.err = err
 		w.done()
+	} else if f, ok := w.Writer.(http.Flusher); ok {
+		f.Flush()
 	}
 	w.mu.Unlock()
 	return