Go中bytes.Reader的Gzip压缩与解压正确实现

在Go语言开发中，处理内存数据的Gzip压缩与解压时，有一个常见错误很容易遇到，那就是“unexpected EOF”异常。这个问题看似复杂，根源却十分简单：gzip.Writer没有显式调用Close()。本文会详细解析完整的实现过程，同时将最容易忽略的关键点彻底讲清楚。

先回顾一下你的操作场景：手里有一个*bytes.Reader，希望将其压缩为Gzip格式，之后再解压回原始数据。听起来很简单，对吗？但如果你只是简单地调用gzip.NewWriter，把数据写入，然后直接返回——很遗憾，解压时很可能收到一个“unexpected EOF”错误。问题到底出在哪里？

答案就是：gzip.Writer必须显式调用Close()，才能确保所有压缩数据——包括尾部校验和（CRC32）以及原始未压缩长度——被完整写入底层缓冲区。如果仅依靠defer writer.Close()或者干脆不调用，压缩流可能只写了正文数据，尾部信息缺失。解压器读取到末尾时发现头部信息与数据长度不匹配，自然会报错。

那么，一个健壮、可复用的实现应该怎么写？直接给出代码：

package main

import (
    "bytes"
    "compress/gzip"
    "fmt"
    "io"
)

type File struct {
    Name string
    Body *bytes.Reader
}

func (f *File) Zip() error {
    var buf bytes.Buffer
    gz := gzip.NewWriter(&buf)

    // 注意：此处不能仅靠 defer！必须在 WriteTo 后显式 Close
    defer gz.Close() // 用于 panic 安全兜底，但不可替代主动 Close

    _, err := f.Body.WriteTo(gz)
    if err != nil {
        return fmt.Errorf("failed to write to gzip writer: %w", err)
    }

    // ✅ 关键步骤：强制刷新并写入 gzip 尾部元数据
    if err := gz.Close(); err != nil {
        return fmt.Errorf("failed to close gzip writer: %w", err)
    }

    f.Body = bytes.NewReader(buf.Bytes())
    f.Name += ".gz"
    return nil
}

func (f *File) UnZip() error {
    gr, err := gzip.NewReader(f.Body)
    if err != nil {
        return fmt.Errorf("failed to create gzip reader: %w", err)
    }
    defer gr.Close() // 解压后务必关闭，释放资源并验证完整性

    var buf bytes.Buffer
    _, err = io.Copy(&buf, gr)
    if err != nil {
        return fmt.Errorf("failed to decompress: %w", err)
    }

    f.Body = bytes.NewReader(buf.Bytes())
    f.Name = f.Name[:len(f.Name)-3] // 移除 ".gz" 后缀（简单处理，生产环境建议更健壮的截断逻辑）
    return nil
}

? 核心注意事项：

• gzip.Writer.Close()不仅会释放资源，还会负责写入Gzip文件尾部（CRC32校验码和原始长度）。缺少这一步，解压器就无法验证数据完整性，unexpected EOF错误会直接出现。
• defer gz.Close()虽然能在函数返回前执行，但如果WriteTo之后还有其他逻辑（例如修改f.Body），它仍然可能晚于关键操作。因此推荐在WriteTo完成后立即显式调用gz.Close()，实现双重保障。
• 解压端的gzip.NewReader也别忘了调用Close()——它会校验尾部数据，如果数据被截断，Close()会返回错误。这是重要的完整性验证步骤，不是可有可无的善后操作。
• 数据搬运用io.Copy即可，简洁且高效。无需自己写循环Read，那样容易遗漏剩余数据。

总结一下：凡是涉及gzip.Writer的内存压缩场景，记住一条原则——“写入后立即Close()”。这样压缩流才能完整，解压过程才会可靠，unexpected EOF这个常见问题自然也就不会出现了。