gzip.go

package gow

import "compress/gzip"

const (
	BestCompression    = gzip.BestCompression
	BestSpeed          = gzip.BestSpeed
	DefaultCompression = gzip.DefaultCompression
	NoCompression      = gzip.NoCompression
)

// Gzip middleware
//	1~9 Compress level
func Gzip(level int, options ...Option) HandlerFunc {
	return newGzipHandler(level, options...).Handle
}

type gzipWriter struct {
	ResponseWriter
	writer *gzip.Writer
}

// WriteString response string
func (g *gzipWriter) WriteString(s string) (int, error) {
	g.Header().Del("Content-Length")
	return g.writer.Write([]byte(s))
}

// Write response []byte
func (g *gzipWriter) Write(data []byte) (int, error) {
	g.Header().Del("Content-Length")
	return g.writer.Write(data)
}

// WriteHeader write gzip header
func (g *gzipWriter) WriteHeader(code int) {
	g.Header().Del("Content-Length")
	g.ResponseWriter.WriteHeader(code)
}