kercylan
41 lines
848 B
Go
41 lines
848 B
Go
package compress
|
|
|
|
import (
|
|
"bytes"
|
|
"compress/gzip"
|
|
"io"
|
|
)
|
|
|
|
// GZipCompress 对数据进行GZip压缩，返回bytes.Buffer和错误信息
|
|
func GZipCompress(data []byte) (bytes.Buffer, error) {
|
|
var buf bytes.Buffer
|
|
gzipWriter := gzip.NewWriter(&buf)
|
|
_, err := gzipWriter.Write(data)
|
|
if err != nil {
|
|
return buf, err
|
|
}
|
|
if err := gzipWriter.Close(); err != nil {
|
|
return buf, err
|
|
}
|
|
return buf, nil
|
|
}
|
|
|
|
// GZipUnCompress 对已进行GZip压缩的数据进行解压缩，返回字节数组及错误信息
|
|
func GZipUnCompress(dataByte []byte) ([]byte, error) {
|
|
data := *bytes.NewBuffer(dataByte)
|
|
gzipReader, err := gzip.NewReader(&data)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
result, err := io.ReadAll(gzipReader)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
if err := gzipReader.Close(); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return result, nil
|
|
}
|