diff --git a/response_writer.go b/response_writer.go index 0def5d0..2a21e57 100644 --- a/response_writer.go +++ b/response_writer.go @@ -12,6 +12,9 @@ type ( ResponseWriter interface { http.ResponseWriter http.Hijacker + http.Flusher + http.CloseNotifier + Status() int Written() bool WriteHeaderNow() @@ -67,3 +70,16 @@ func (w *responseWriter) Hijack() (net.Conn, *bufio.ReadWriter, error) { } return hijacker.Hijack() } + +// Implements the http.CloseNotify interface +func (w *responseWriter) CloseNotify() <-chan bool { + return w.ResponseWriter.(http.CloseNotifier).CloseNotify() +} + +// Implements the http.Flush interface +func (w *responseWriter) Flush() { + flusher, ok := w.ResponseWriter.(http.Flusher) + if ok { + flusher.Flush() + } +}