64 lines
1.4 KiB
Go
64 lines
1.4 KiB
Go
package restful
|
|
|
|
import (
|
|
"bytes"
|
|
"compress/gzip"
|
|
"compress/zlib"
|
|
"sync"
|
|
)
|
|
|
|
// GzipWriterPool is used to get reusable zippers.
|
|
// The Get() result must be type asserted to *gzip.Writer.
|
|
var GzipWriterPool = &sync.Pool{
|
|
New: func() interface{} {
|
|
return newGzipWriter()
|
|
},
|
|
}
|
|
|
|
func newGzipWriter() *gzip.Writer {
|
|
// create with an empty bytes writer; it will be replaced before using the gzipWriter
|
|
writer, err := gzip.NewWriterLevel(new(bytes.Buffer), gzip.BestSpeed)
|
|
if err != nil {
|
|
panic(err.Error())
|
|
}
|
|
return writer
|
|
}
|
|
|
|
// GzipReaderPool is used to get reusable zippers.
|
|
// The Get() result must be type asserted to *gzip.Reader.
|
|
var GzipReaderPool = &sync.Pool{
|
|
New: func() interface{} {
|
|
return newGzipReader()
|
|
},
|
|
}
|
|
|
|
func newGzipReader() *gzip.Reader {
|
|
// create with an empty reader (but with GZIP header); it will be replaced before using the gzipReader
|
|
w := GzipWriterPool.Get().(*gzip.Writer)
|
|
b := new(bytes.Buffer)
|
|
w.Reset(b)
|
|
w.Flush()
|
|
w.Close()
|
|
reader, err := gzip.NewReader(bytes.NewReader(b.Bytes()))
|
|
if err != nil {
|
|
panic(err.Error())
|
|
}
|
|
return reader
|
|
}
|
|
|
|
// ZlibWriterPool is used to get reusable zippers.
|
|
// The Get() result must be type asserted to *zlib.Writer.
|
|
var ZlibWriterPool = &sync.Pool{
|
|
New: func() interface{} {
|
|
return newZlibWriter()
|
|
},
|
|
}
|
|
|
|
func newZlibWriter() *zlib.Writer {
|
|
writer, err := zlib.NewWriterLevel(new(bytes.Buffer), gzip.BestSpeed)
|
|
if err != nil {
|
|
panic(err.Error())
|
|
}
|
|
return writer
|
|
}
|