Go archive/zip 怎么为指定方法注册自定义压缩器

直接答案：创建 zip.Writer 后调用 w.RegisterCompressor(method, comp)，再把同一个方法编号写入 FileHeader.Method。comp 必须返回一个向底层 writer 写入压缩数据的 io.WriteCloser；若只想影响当前归档，优先使用 Writer 级注册，而不是包级全局注册。

先把注册范围缩到当前 Writer

我在调整 ZIP 压缩等级时，最容易忽略的不是压缩算法，而是注册作用域。zip.RegisterCompressor 写入包级注册表；w.RegisterCompressor 只覆盖当前 Writer。后者更适合服务端请求、测试和不同归档采用不同参数的场景，也不会悄悄改变进程里其他 ZIP 写入任务。

func newFastZIP(dst io.Writer) *zip.Writer {
    zw := zip.NewWriter(dst)

    // 只覆盖当前 Writer 的 Deflate 实现，不修改包级全局注册表
    zw.RegisterCompressor(zip.Deflate, func(out io.Writer) (io.WriteCloser, error) {
        return flate.NewWriter(out, flate.BestSpeed)
    })
    return zw
}

官方定义的 Compressor 是一个工厂函数：输入底层 io.Writer，返回新的压缩 io.WriteCloser。工厂本身必须能被多个 goroutine 安全调用；但每次返回的 writer 只会由一个 goroutine 使用。

Method 必须和注册编号一致

只注册还不够。Create() 固定使用内置 Deflate；要明确选择方法，应构造 FileHeader 并调用 CreateHeader()。下面把 Deflate 的实现换成快速等级，同时仍保持常见 ZIP 工具可读取。

func writeReport(dst io.Writer, body []byte) error {
    zw := newFastZIP(dst)
    header := &zip.FileHeader{
        Name:   "report.txt",
        Method: zip.Deflate, // 必须匹配已注册的 method ID
    }

    entry, err := zw.CreateHeader(header)
    if err != nil {
        _ = zw.Close() // 创建失败时也释放归档写入状态
        return err
    }
    if _, err := entry.Write(body); err != nil {
        _ = zw.Close() // 写入失败时保留原始错误
        return err
    }
    return zw.Close() // 写入中央目录，并检查最终刷新错误
}

CreateHeader 会接管 header，调用后不要再修改它。ZIP Writer 的 Close 会完成归档并写中央目录，但不会关闭传入的底层 writer，因此文件或网络连接仍由调用方管理。

自定义 method ID 还要解决读取端

如果只是覆盖 zip.Deflate 的参数，读取端仍按方法 8 解压，兼容性最好。若注册一个新的私有方法编号，生成的归档只有认识该编号的读取端才能打开；Go 端需要为对应 Reader 注册匹配的 Decompressor。找不到实现时会得到不支持压缩算法的错误。

用三组指标决定是否值得替换

我不会因为 BestCompression 这个名字就直接上线。更稳妥的做法是用同一批代表性文件比较：归档总字节数、每次写入耗时，以及 go test -benchmem 给出的分配量。文本、日志与已经压缩的图片表现差异很大，必须按真实数据分组。

还要把 Close 的耗时和错误计入测试，因为压缩器可能在关闭时才刷新尾部数据。判断标准可以很简单：若体积下降不足以抵消 CPU 与延迟增加，就保留默认 Deflate；若数据已经压缩，优先考虑 Store，避免重复压缩。

常见问题

Writer 级注册和包级注册谁优先？

当前 Writer 找到指定方法的注册项时使用它；找不到时才回退到包级注册表。需要局部参数时使用 Writer 级注册更清晰。

为什么写完内容还必须检查 Close？

压缩 writer 的 Close 需要刷新待处理数据，ZIP Writer 的 Close 还要写中央目录。忽略关闭错误可能得到内容不完整或无法正常读取的归档。