Implementing Flush method to httplog

This commit is contained in:
jhadvig
2014-10-21 01:41:23 +02:00
parent 69c7228006
commit f759bfdf71
2 changed files with 13 additions and 1 deletions

View File

@@ -161,6 +161,16 @@ func (rl *respLogger) Write(b []byte) (int, error) {
return rl.w.Write(b)
}
// Flush implements http.Flusher even if the underlying http.Writer doesn't implement it.
// Flush is used for streaming purposes and allows to flush buffered data to the client.
func (rl *respLogger) Flush() {
if flusher, ok := rl.w.(http.Flusher); ok {
flusher.Flush()
} else {
glog.V(2).Infof("Unable to convert %v into http.Flusher", rl.w)
}
}
// WriteHeader implements http.ResponseWriter.
func (rl *respLogger) WriteHeader(status int) {
rl.status = status