You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
60 lines
1.4 KiB
60 lines
1.4 KiB
// Copyright GoFrame Author(https://goframe.org). All Rights Reserved.
|
|
//
|
|
// This Source Code Form is subject to the terms of the MIT License.
|
|
// If a copy of the MIT was not distributed with this file,
|
|
// You can obtain one at https://github.com/gogf/gf.
|
|
|
|
// Package gcompress provides kinds of compression algorithms for binary/bytes data.
|
|
package gcompress
|
|
|
|
import (
|
|
"bytes"
|
|
"compress/zlib"
|
|
"io"
|
|
|
|
"github.com/gogf/gf/v2/errors/gerror"
|
|
)
|
|
|
|
// Zlib compresses `data` with zlib algorithm.
|
|
func Zlib(data []byte) ([]byte, error) {
|
|
if data == nil || len(data) < 13 {
|
|
return data, nil
|
|
}
|
|
var (
|
|
err error
|
|
in bytes.Buffer
|
|
writer = zlib.NewWriter(&in)
|
|
)
|
|
|
|
if _, err = writer.Write(data); err != nil {
|
|
err = gerror.Wrapf(err, `zlib.Writer.Write failed`)
|
|
return nil, err
|
|
}
|
|
if err = writer.Close(); err != nil {
|
|
err = gerror.Wrapf(err, `zlib.Writer.Close failed`)
|
|
return in.Bytes(), err
|
|
}
|
|
return in.Bytes(), nil
|
|
}
|
|
|
|
// UnZlib decompresses `data` with zlib algorithm.
|
|
func UnZlib(data []byte) ([]byte, error) {
|
|
if data == nil || len(data) < 13 {
|
|
return data, nil
|
|
}
|
|
var (
|
|
out bytes.Buffer
|
|
bytesReader = bytes.NewReader(data)
|
|
zlibReader, err = zlib.NewReader(bytesReader)
|
|
)
|
|
if err != nil {
|
|
err = gerror.Wrapf(err, `zlib.NewReader failed`)
|
|
return nil, err
|
|
}
|
|
if _, err = io.Copy(&out, zlibReader); err != nil {
|
|
err = gerror.Wrapf(err, `io.Copy failed`)
|
|
return nil, err
|
|
}
|
|
return out.Bytes(), nil
|
|
}
|