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/handler/cryptionhandler.go
View File

@@ -83,6 +83,12 @@ func newCryptionResponseWriter(w http.ResponseWriter) *cryptionResponseWriter {
}
}
func (w *cryptionResponseWriter) Flush() {
if flusher, ok := w.ResponseWriter.(http.Flusher); ok {
flusher.Flush()
}
}
func (w *cryptionResponseWriter) Header() http.Header {
return w.ResponseWriter.Header()
}
@@ -95,12 +101,6 @@ func (w *cryptionResponseWriter) WriteHeader(statusCode int) {
w.ResponseWriter.WriteHeader(statusCode)
}
func (w *cryptionResponseWriter) Flush() {
if flusher, ok := w.ResponseWriter.(http.Flusher); ok {
flusher.Flush()
}
}
func (w *cryptionResponseWriter) flush(key []byte) {
if w.buf.Len() == 0 {
return