Files
go-zero/core/httphandler/gunziphandler.go
2020-07-26 17:09:05 +08:00

28 lines
493 B
Go

package httphandler
import (
"compress/gzip"
"net/http"
"strings"
"zero/core/httpx"
)
const gzipEncoding = "gzip"
func GunzipHandler(next http.Handler) http.Handler {
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
if strings.Contains(r.Header.Get(httpx.ContentEncoding), gzipEncoding) {
reader, err := gzip.NewReader(r.Body)
if err != nil {
w.WriteHeader(http.StatusBadRequest)
return
}
r.Body = reader
}
next.ServeHTTP(w, r)
})
}