serv/caddy
1
0
Fork 0
mirror of https://github.com/caddyserver/caddy.git synced 2025-01-09 20:38:49 +03:00
Code Issues Projects Releases Packages Wiki Activity Actions

Merge pull request #737 from tw4452852/closeNotifier

Browse source 
http.CloseNotifier implementation for http.ResponseWriter wrapper
This commit is contained in:
Matt Holt 2016-04-11 00:18:06 -06:00
commit c4e2cf96e7
2 changed files with 18 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
middleware/gzip/gzip.go
View file

@ -141,3 +141,12 @@ func (w *gzipResponseWriter) Hijack() (net.Conn, *bufio.ReadWriter, error) {
} }
return nil, nil, fmt.Errorf("not a Hijacker") return nil, nil, fmt.Errorf("not a Hijacker")
} }
// CloseNotify implements http.CloseNotifier.
// It just inherits the underlying ResponseWriter's CloseNotify method.
func (w *gzipResponseWriter) CloseNotify() <-chan bool {
if cn, ok := w.ResponseWriter.(http.CloseNotifier); ok {
return cn.CloseNotify()
}
panic("not a CloseNotifier")
}

9
middleware/recorder.go
View file

@ -87,3 +87,12 @@ func (r *ResponseRecorder) Flush() {
panic("not a Flusher") // should be recovered at the beginning of middleware stack panic("not a Flusher") // should be recovered at the beginning of middleware stack
} }
} }
// CloseNotify implements http.CloseNotifier.
// It just inherits the underlying ResponseWriter's CloseNotify method.
func (r *ResponseRecorder) CloseNotify() <-chan bool {
if cn, ok := r.ResponseWriter.(http.CloseNotifier); ok {
return cn.CloseNotify()
}
panic("not a CloseNotifier")
}