Commit 80ef5d76 authored by Matthew Holt's avatar Matthew Holt

Moved gzip middleware into its own package

Trying a different format where the middleware is a type that satisfies http.Handler
parent affd4708
package middleware package gzip
import ( import (
"compress/gzip" "compress/gzip"
"io" "io"
"net/http" "net/http"
"strings" "strings"
"github.com/mholt/caddy/middleware"
) )
func Gzip(p parser) Middleware { // New creates a new gzip middleware instance.
func New(c middleware.Controller) (middleware.Middleware, error) {
return func(next http.HandlerFunc) http.HandlerFunc { return func(next http.HandlerFunc) http.HandlerFunc {
return func(w http.ResponseWriter, r *http.Request) { gz := Gzip{next: next}
if !strings.Contains(r.Header.Get("Accept-Encoding"), "gzip") { return gz.ServeHTTP
next(w, r) }, nil
return }
}
w.Header().Set("Content-Encoding", "gzip") // Gzip is a http.Handler middleware type which gzips HTTP responses.
gzipWriter := gzip.NewWriter(w) type Gzip struct {
defer gzipWriter.Close() next http.HandlerFunc
gz := gzipResponseWriter{Writer: gzipWriter, ResponseWriter: w} // TODO: Compression level, other settings
next(gz, r) }
}
// ServeHTTP serves a gzipped response if the client supports it.
func (g *Gzip) ServeHTTP(w http.ResponseWriter, r *http.Request) {
if !strings.Contains(r.Header.Get("Accept-Encoding"), "gzip") {
g.next(w, r)
return
} }
w.Header().Set("Content-Encoding", "gzip")
gzipWriter := gzip.NewWriter(w)
defer gzipWriter.Close()
gz := gzipResponseWriter{Writer: gzipWriter, ResponseWriter: w}
g.next(gz, r)
} }
// gzipResponeWriter wraps the underlying Write method // gzipResponeWriter wraps the underlying Write method
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment