gzip: Fix Content-Length header for proxies requests (closes #38)

1 files changed, 9 insertions, 0 deletions

diff --git a/middleware/gzip/gzip.go b/middleware/gzip/gzip.go
index 694118321..8bbc1d146 100644
--- a/middleware/gzip/gzip.go
+++ b/middleware/gzip/gzip.go
@@ -64,6 +64,15 @@ type gzipResponseWriter struct {
 	http.ResponseWriter
 }
 
+// WriteHeader wraps the underlying WriteHeader method to prevent
+// problems with conflicting headers from proxied backends. For
+// example, a backend system that calculates Content-Length would
+// be wrong because it doesn't know it's being gzipped.
+func (w gzipResponseWriter) WriteHeader(code int) {
+	w.Header().Del("Content-Length")
+	w.ResponseWriter.WriteHeader(code)
+}
+
 // Write wraps the underlying Write method to do compression.
 func (w gzipResponseWriter) Write(b []byte) (int, error) {
 	if w.Header().Get("Content-Type") == "" {