A simple wrapper for http gzipped response

gzip_handler_wrapper.go

type (
	GzipResponseWriter struct {
		io.Writer
		http.ResponseWriter
	}
)

func (g GzipResponseWriter) Write(b []byte) (int, error) {
	return g.Writer.Write(b)
}

func NewGzipHandler(f func(w http.ResponseWriter, req *http.Request)) http.HandlerFunc {
	return http.HandlerFunc(func(w http.ResponseWriter, req *http.Request) {

		encoding := req.Header.Get("Accept-Encoding")

		if strings.Contains(encoding, "gzip") {

			w.Header().Set("Content-Encoding", "gzip")
			gz := gzip.NewWriter(w)
			defer gz.Close()

			f(GzipResponseWriter{gz, w}, req)

		} else {
			f(w, req)
		}

	})
}