Skip to content

Instantly share code, notes, and snippets.

@the42
Created March 2, 2012 07:34
Show Gist options
  • Save the42/1956518 to your computer and use it in GitHub Desktop.
Save the42/1956518 to your computer and use it in GitHub Desktop.
GZip encoding for GO V1 using custom responsewriter
package main
import (
"compress/gzip"
"io"
"net/http"
"strings"
)
type gzipResponseWriter struct {
io.Writer
http.ResponseWriter
}
func (w gzipResponseWriter) Write(b []byte) (int, error) {
return w.Writer.Write(b)
}
func makeGzipHandler(fn http.HandlerFunc) http.HandlerFunc {
return func(w http.ResponseWriter, r *http.Request) {
if !strings.Contains(r.Header.Get("Accept-Encoding"), "gzip") {
fn(w, r)
return
}
w.Header().Set("Content-Encoding", "gzip")
gz := gzip.NewWriter(w)
defer gz.Close()
gzr := gzipResponseWriter{Writer: gz, ResponseWriter: w}
fn(gzr, r)
}
}
func handler(w http.ResponseWriter, r *http.Request) {
w.Header().Set("Content-Type", "text/plain")
w.Write([]byte("This is a test."))
}
func main() {
http.ListenAndServe(":1113", makeGzipHandler(handler))
}
@optimality
Copy link

Also, the error from gz.Close() isn't being checked, which leads to mysterious failures like golang/go#14975 (comment). If you use this, make sure to check that error.

@erikdubbelboer
Copy link

A added a new commit to my fork that fixes some issues: https://gist.github.com/erikdubbelboer/7df2b2b9f34f9f839a84

The issue from @optimality about not handling gz.Close errors still stands as everyone should handle this in their own way since http.HandlerFunc can't return errors.

@hrieke
Copy link

hrieke commented Jun 29, 2018

License?

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment