improve package gcompress

This commit is contained in:
John
2020-05-01 01:47:02 +08:00
parent f7c2a51c9f
commit cc67f3d388
2 changed files with 31 additions and 11 deletions

View File

@ -19,9 +19,11 @@ import (
//
// Note that it returns error if given <level> is invalid.
func Gzip(data []byte, level ...int) ([]byte, error) {
var writer *gzip.Writer
var buf bytes.Buffer
var err error
var (
writer *gzip.Writer
buf bytes.Buffer
err error
)
if len(level) > 0 {
writer, err = gzip.NewWriterLevel(&buf, level[0])
if err != nil {
@ -41,8 +43,10 @@ func Gzip(data []byte, level ...int) ([]byte, error) {
// GzipFile compresses the file <src> to <dst> using gzip algorithm.
func GzipFile(src, dst string, level ...int) error {
var writer *gzip.Writer
var err error
var (
writer *gzip.Writer
err error
)
srcFile, err := gfile.Open(src)
if err != nil {
return err