在 defer 中延迟关闭 gzip writer 可能会导致数据丢失

在Golang实战开发的过程中，我们经常会遇到一些这样那样的问题，然后要卡好半天，等问题解决了才发现原来一些细节知识点还是没有掌握好。今天golang学习网就整理分享《在 defer 中延迟关闭 gzip writer 可能会导致数据丢失》，聊聊，希望可以帮助到正在努力赚钱的你。

我使用 golang gzip.newwriter 来压缩切片，并使用 defer close() 来关闭 writer。但是当读取压缩数据时，它将返回unexpected eof。代码是：

func main() {

    a := []byte{'a', 'b', 'c', 'd', 'e', 'f'}
    zippeddata, err := zipdata(a)
    if err != nil {
        panic(err)
    }

    unzippeddata, err := unzipdata(zippeddata)
    if err != nil {
        panic(err)
    }
    fmt.printf("%v\n", unzippeddata)

}

zip 函数是：

func zipdata(origindata []byte) ([]byte, error) {
    var bf bytes.buffer
    gw := gzip.newwriter(&bf)

    defer gw.close()

    _, err := gw.write(origindata)
    if err != nil {
        return nil, errors.new(fmt.sprintf("gzip data err: %v", err))
    }

    err = gw.flush()
    if err != nil {
        return nil, err
    }
    // if i rm 'defer gw.close()' and call 'gw.close()' here, it'll be ok

    logs.debug("before gzip len: %v", len(origindata))
    logs.debug("gzip len: %v", bf.len())
    return bf.bytes(), nil
}

上面的zip函数使用defer gw.close()来关闭gw。

解压函数是：

func unzipData(zippedData []byte) ([]byte, error) {
    dst := make([]byte, len(zippedData))
    copy(dst, zippedData)

    reader, err := gzip.NewReader(bytes.NewBuffer(dst))
    if err != nil {
        return nil, errors.New(fmt.Sprintf("unzip err :%v", err))
    }

    defer reader.Close()

    data, err := ioutil.ReadAll(reader)
    if err != nil {
        return nil, errors.New(fmt.Sprintf("read err :%v", err))
    }
    return data, err
}

为什么 defer gw.close() 会出现 意外的 eof？

解决方案

使用 defer，您会丢失 gzip 页脚。根据 Clos​​e 文档：

因此，即使 Flush 刷新任何缓冲数据，它也不会写入页脚。通过延迟关闭，您将获取不包含页脚的字节数组并将其返回，然后将页脚写入输出。

返回之前关闭编写器。

以上就是本文的全部内容了，是否有顺利帮助你解决问题？若是能给你带来学习上的帮助，请大家多多支持golang学习网！更多关于Golang的相关知识，也可关注golang学习网公众号。