登录
推荐 文章 Go 技术 课程 下载 专题 AI
首页 >  Golang >  Go问答

Go archive/zip RegisterCompressor 为什么没有被调用

来源:17golang原创

时间:2026-09-26 19:21:46 467浏览 收藏

直接答案:RegisterCompressor 只是在注册表里把压缩器绑定到一个 method ID;只有后续创建的条目真的选择了这个 ID,并且走的是会压缩的写入接口,工厂函数才会被调用。最常见原因是 FileHeader.Method 仍为零(即 Store)、注册了自定义 ID 却使用 Create(),或者实际调用了绕过压缩的 CreateRaw()/Copy()。

先核对条目的 Method

Writer.RegisterCompressor(method, comp) 按方法编号匹配。FileHeader.Method 为零时使用 Store,不会选择为 Deflate 或其他编号注册的压缩器。若使用 Create(name),标准库固定选择 Deflate;因此注册自定义编号后再调用 Create,也不会命中。

写入方式实际方法来源何时调用注册压缩器
Create(name)固定 Deflate注册项的 ID 是 zip.Deflate
CreateHeader(h)h.MethodMethod 与注册 ID 完全一致
AddFS(fsys)使用 Deflate覆盖了当前 Writer 的 Deflate
CreateRaw(h)写入已压缩原始字节不会调用
Copy(f)复制原始形式不会调用
archive zip 写入 API 与压缩方法编号的静态匹配关系图
图1:先看 API 与 method ID 的绑定。Create 和 AddFS 选择 Deflate,CreateHeader 读取 FileHeader.Method,Store 与自定义 ID 必须各自命中对应注册项;这是静态关系说明图。

确认没有走 Raw 或 Copy 旁路

CreateRaw() 接收的就是原始压缩数据,标准库不会再次压缩。Writer.Copy() 也会直接复制源文件的原始形式,明确绕过解压、压缩和校验。这两个接口适合无损搬运条目,却不会触发 Compressor。

另一个容易混淆的点是作用域:w.RegisterCompressor 只属于那个 Writer 实例。若把压缩器注册到 writerA,实际条目却由 writerB 创建,writerB 只会查自己的注册表,再回退到包级注册表。

ZIP Writer 私有注册表、包级回退与原始写入旁路结构图
图2:看对象归属与旁路。Writer A 的私有注册不会影响 Writer B;CreateRaw 和 Copy 直接处理原始形式,不经过 Compressor 工厂。这是对象关系说明图,不是执行截图。

注册必须早于对应条目创建

压缩器是在创建条目 writer 时按 Method 查找的。若先调用 CreateHeader(),再注册压缩器,当前条目已经完成选择;新注册只可能影响之后创建的条目。生产代码应在 zip.NewWriter 之后集中完成注册,再开始添加任何文件。

包级 zip.RegisterCompressor 和 Writer 级注册不要混着排查。Writer 找到自己的注册项时直接使用;只有当前 Writer 没找到指定 method,才回退到包级注册表。为了减少全局状态,通常优先 Writer 级注册。

用最小探针确认是否命中

下面用 atomic.Bool 记录工厂函数是否被选择。示例覆盖 Deflate,因此 Create() 或 Method 为 Deflate 的 CreateHeader() 都能命中;探针只用于排查,确认后可以移除。

func writeAndProbe(dst io.Writer, body []byte) (bool, error) {
    var called atomic.Bool
    zw := zip.NewWriter(dst)

    // 注册到实际用于写条目的同一个 Writer,并覆盖 Deflate
    zw.RegisterCompressor(zip.Deflate, func(out io.Writer) (io.WriteCloser, error) {
        called.Store(true) // 工厂被选择时记录一次,不在这里统计写入次数
        return flate.NewWriter(out, flate.BestSpeed)
    })

    header := &zip.FileHeader{
        Name:   "data.txt",
        Method: zip.Deflate, // 必须与注册时的 method ID 一致
    }
    entry, err := zw.CreateHeader(header)
    if err != nil {
        _ = zw.Close() // 创建失败时释放归档状态
        return called.Load(), err
    }
    if _, err := entry.Write(body); err != nil {
        _ = zw.Close() // 保留原始写入错误
        return called.Load(), err
    }
    if err := zw.Close(); err != nil {
        return called.Load(), err // Close 负责刷新压缩数据和中央目录
    }
    return called.Load(), nil
}

如果探针仍为 false,优先打印或检查“注册 ID、header.Method、Writer 对象地址、创建 API”四项,而不是反复修改压缩函数内部逻辑。Compressor 工厂本身还必须支持被多个 goroutine 同时调用;每次返回的 writer 则只由一个 goroutine 使用。

常见问题

覆盖 zip.Deflate 会影响所有 ZIP Writer 吗?

调用 w.RegisterCompressor 只影响当前 Writer;调用包级 zip.RegisterCompressor 才会改变包级回退项。

FileHeader.Method 不赋值为什么不会压缩?

Method 为零表示 Store。若要压缩,必须设置为 zip.Deflate 或另一个已注册的非零方法编号。

声明:本文转载于:17golang原创 如有侵犯,请联系study_golang@163.com删除
相关阅读
更多>
最新阅读
更多>
课程推荐
更多>