2018-09-22 05:17:57 +02:00
|
|
|
package compress
|
|
|
|
|
|
|
|
import (
|
|
|
|
"bytes"
|
|
|
|
"compress/flate"
|
|
|
|
"io"
|
|
|
|
)
|
|
|
|
|
|
|
|
// Compress returns a compressed byte slice.
|
|
|
|
func Compress(src []byte) []byte {
|
|
|
|
compressedData := new(bytes.Buffer)
|
2018-09-26 23:31:45 +02:00
|
|
|
compress(src, compressedData, 9)
|
2018-09-22 05:17:57 +02:00
|
|
|
return compressedData.Bytes()
|
|
|
|
}
|
|
|
|
|
|
|
|
// Decompress returns a decompressed byte slice.
|
|
|
|
func Decompress(src []byte) []byte {
|
|
|
|
compressedData := bytes.NewBuffer(src)
|
|
|
|
deCompressedData := new(bytes.Buffer)
|
|
|
|
decompress(compressedData, deCompressedData)
|
|
|
|
return deCompressedData.Bytes()
|
|
|
|
}
|
|
|
|
|
|
|
|
// compress uses flate to compress a byte slice to a corresponding level
|
|
|
|
func compress(src []byte, dest io.Writer, level int) {
|
|
|
|
compressor, _ := flate.NewWriter(dest, level)
|
|
|
|
compressor.Write(src)
|
|
|
|
compressor.Close()
|
|
|
|
}
|
|
|
|
|
|
|
|
// compress uses flate to decompress an io.Reader
|
|
|
|
func decompress(src io.Reader, dest io.Writer) {
|
|
|
|
decompressor := flate.NewReader(src)
|
|
|
|
io.Copy(dest, decompressor)
|
|
|
|
decompressor.Close()
|
|
|
|
}
|