The ResponseWriters defined in rest.handler add Flush interface. (#318)

2020-12-28 21:30:24 +08:00
parent 71a2b20301
commit 0bd2a0656c
8 changed files with 88 additions and 0 deletions
--- a/rest/handler/authhandler.go
+++ b/rest/handler/authhandler.go
@@ -138,3 +138,9 @@ func (grw *guardedResponseWriter) WriteHeader(statusCode int) {
 	grw.wroteHeader = true
 	grw.writer.WriteHeader(statusCode)
 }
+
+func (grw *guardedResponseWriter) Flush() {
+	if flusher, ok := grw.writer.(http.Flusher); ok {
+		flusher.Flush()
+	}
+}