zeromicro/go-zero
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

simplify code with http.Flusher type conversion (#325)

Browse Source 
* simplify code with http.Flusher type conversion

* simplify code with http.Flusher type conversion, better version
This commit is contained in:
Kevin Wan
2020-12-29 10:25:55 +08:00
committed by GitHub
parent 0bd2a0656c
commit e97e1f10db
8 changed files with 29 additions and 32 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
rest/internal/security/withcoderesponsewriter.go
View File

@@ -7,6 +7,12 @@ type WithCodeResponseWriter struct {
Code int
}
func (w *WithCodeResponseWriter) Flush() {
if flusher, ok := w.Writer.(http.Flusher); ok {
flusher.Flush()
}
}
func (w *WithCodeResponseWriter) Header() http.Header {
return w.Writer.Header()
}
@@ -19,9 +25,3 @@ func (w *WithCodeResponseWriter) WriteHeader(code int) {
w.Writer.WriteHeader(code)
w.Code = code
}
func (w *WithCodeResponseWriter) Flush() {
if flusher, ok := w.Writer.(http.Flusher); ok {
flusher.Flush()
}
}