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

Go archive/zip.Writer.Copy 的原始数据搬运与压缩开销控制

来源:17golang原创

时间:2026-11-03 18:28:39 341浏览 收藏

重打包 ZIP 时,如果大部分条目只需要换一个归档容器,先用 f.Open() 读取再写入 Writer.Create 会经历解压和再次压缩。Go 的 archive/zip.Writer.Copy 可以直接搬运已有条目的 raw form;未修改的文件走这条分支,需要替换内容的文件再切换到 CreateHeader。

要点速览
  • Writer.Copy 只适合从 zip.Reader 得到的、无需改内容的 *zip.File。
  • 需要改正文、文件名或压缩策略时,复制元数据后使用 CreateHeader;已有压缩字节才考虑 CreateRaw。
  • Writer.Close 会写中央目录,不能把它省略;Copy 绕过校验,输入可信度要由调用方负责。

先把三种写入路径分清

场景选择关键边界
条目原样保留Writer.Copy(f)直接复制 raw form,不解压、不再压缩、不做校验
修改内容或文件头CreateHeader写入的是未压缩内容,Writer 再按 header 处理
手上已有压缩字节CreateRawFileHeader 的方法、大小和 CRC 必须与字节一致

因此,Copy 的优化点不是“让压缩更快”,而是让不需要变化的条目避开一次解压和一次压缩。它也不是通用的 ZIP 修复器:官方文档明确说明 raw 搬运会绕过校验,输入本身不可信时不能把它当作完整性检查。

Copy 与 CreateHeader 的边界

下面的函数把 replace 中列出的文件改写,其余条目直接复制。示例故意把源文件和目标文件分开,避免边读边覆盖同一个 ZIP。

package main

import (
    "archive/zip"
    "fmt"
    "io"
    "os"
)

func repack(srcPath, dstPath string, replace map[string][]byte) error {
    // Reader 只负责读取源归档;OpenReader 会同时拿到文件句柄和条目列表。
    src, err := zip.OpenReader(srcPath)
    if err != nil {
        return fmt.Errorf("打开源 ZIP: %w", err)
    }
    defer src.Close()

    // 目标文件必须是另一条路径,避免覆盖仍在读取的源归档。
    dst, err := os.Create(dstPath)
    if err != nil {
        return fmt.Errorf("创建目标 ZIP: %w", err)
    }
    defer dst.Close()

    zw := zip.NewWriter(dst)
    for _, f := range src.File {
        data, needReplace := replace[f.Name]
        if !needReplace {
            // 未修改条目直接搬运压缩字节,跳过解压和再次压缩。
            if err := zw.Copy(f); err != nil {
                return fmt.Errorf("复制 %q: %w", f.Name, err)
            }
            continue
        }

        // 复制 FileHeader 保留名称、时间和已有元数据,再让 Writer 处理新内容。
        header := f.FileHeader
        out, err := zw.CreateHeader(&header)
        if err != nil {
            return fmt.Errorf("创建替换条目 %q: %w", f.Name, err)
        }
        if _, err := io.WriteString(out, string(data)); err != nil {
            return fmt.Errorf("写入替换条目 %q: %w", f.Name, err)
        }
    }

    // Close 会补写中央目录;成功返回前必须执行它。
    if err := zw.Close(); err != nil {
        return fmt.Errorf("关闭 ZIP Writer: %w", err)
    }
    return nil
}

这里的替换分支使用 CreateHeader,因为 data 是解压后的新内容。若只想改文件名,不能继续对原 *zip.File 调用 Copy 后再改名,而应复制 header、修改 header.Name,再通过 CreateHeader 写入内容。

Go archive zip 中 File、FileHeader、Writer.Copy 与 CreateHeader 的 raw 压缩字节关系说明图
图1:Writer.Copy 直接搬运 ZIP 条目 raw form,并与内容改写接口分开的结构说明图。

可替换条目的归档结构

批量迁移时可以把决策放在条目循环中:没变的条目走 Copy,需要内容替换的条目走 CreateHeader,两类最终都进入同一个目标 Writer。最后由 Close 写出中央目录,读者才能通过文件名索引到这些条目。

// 只有已经准备好压缩字节时才使用 CreateRaw;普通新内容不要走这里。
header := zip.FileHeader{
    Name:   "assets/data.bin",
    Method: zip.Deflate,
}
out, err := zw.CreateHeader(&header)
if err != nil {
    return err
}
// out 接收未压缩内容,Writer 会按 header.Method 处理压缩。
if _, err := out.Write(data); err != nil {
    return err
}

如果已有的是某个条目的压缩字节,而不是原始内容,才应使用 CreateRaw,并确保 Method、CRC32、CompressedSize64 和 UncompressedSize64 与这段字节相符。否则新归档的目录信息和实际数据会不一致。

Go ZIP 重打包中未修改条目、替换内容、FileHeader、目标 Writer 与中央目录的结构说明图
图2:ZIP 重打包中保留条目、替换条目与中央目录的关系说明图。

上线前检查这几个边界

  • 输入是否可信:Copy 绕过解压、压缩和校验,来源归档应先经过业务层的来源控制。
  • 是否修改内容:修改内容就用 CreateHeader;不要为了复用 Copy 而手工拼 ZIP 结构。
  • 是否关闭 Writer:检查 zw.Close() 返回值,再关闭目标文件。
  • 是否覆盖源文件:先写临时目标,成功后再按业务规则替换源文件。

相关问题

Writer.Copy 会验证源条目的 CRC 吗

不会。它直接搬运 raw form,绕过解压、压缩和校验;需要验证内容时应改用 Open 读取并自行处理错误。

修改 ZIP 文件名还能继续使用 Copy 吗

不能把 Copy 当成改名接口。应复制 FileHeader,修改 Name 后用 CreateHeader 写入解压后的内容。

为什么目标 ZIP 打不开

优先检查 Writer.Close 是否执行并返回成功,以及是否在 Writer 关闭前关闭了底层文件。中央目录没有写完时,归档通常是不完整的。

把“原样搬运”和“内容重写”分成两条明确路径,才是 archive/zip.Writer.Copy 控制压缩开销的核心:少做一次解压和再压缩,同时保留对需要变更条目的正常写入能力。

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