Golang如何实现文件压缩解压实践archive/zip标准库操作-Golang-PHP中文网

Golang如何实现文件压缩解压实践archive/zip标准库操作

P粉602998670

发布： 2025-07-20 08:09:02

原创

342人浏览过

在 go 中使用 archive/zip 包可实现 zip 文件的压缩和解压。1. 创建 zip 文件需依次创建文件、初始化写入器、遍历源文件并添加至 zip；2. 解压 zip 需打开 zip 文件、遍历条目并根据类型创建目录或写入文件；3. 注意路径安全、权限控制、压缩级别限制及大文件处理。上述方法适用于大多数基础场景，但需注意资源释放与路径校验以避免安全隐患。

Golang如何实现文件压缩解压实践archive/zip标准库操作

在 Golang 中，如果你需要对文件进行压缩或解压操作，最常用的方式就是使用标准库中的 archive/zip 包。它不需要额外依赖，直接可用，适合大多数基本的 ZIP 文件处理需求。

下面我们就来看看如何用这个包实现常见的压缩和解压功能。

如何创建一个 ZIP 压缩文件

要压缩文件，核心是使用 zip.Create() 创建 ZIP 文件，然后逐个添加文件进去。大致步骤如下：

立即学习“go语言免费学习笔记（深入）”；

知我AI

一款多端AI知识助理，通过一键生成播客/视频/文档/网页文章摘要、思维导图，提高个人知识获取效率；自动存储知识，通过与知识库聊天，提高知识利用效率。

101

查看详情

创建目标 ZIP 文件
使用 zipWriter := zip.NewWriter(file) 初始化写入器
遍历需要压缩的文件，为每个文件创建一个 zip.FileHeader
打开源文件并复制内容到 ZIP 写入器中

func compressFiles(zipPath string, filesToCompress []string) error {
    newZipFile, err := os.Create(zipPath)
    if err != nil {
        return err
    }
    defer newZipFile.Close()

    zipWriter := zip.NewWriter(newZipFile)
    defer zipWriter.Close()

    for _, filename := range filesToCompress {
        fileToZip, err := os.Open(filename)
        if err != nil {
            return err
        }
        defer fileToZip.Close()

        // 获取文件信息，用于构造 ZIP 条目
        info, err := fileToZip.Stat()
        if err != nil {
            return err
        }

        header, err := zip.FileInfoHeader(info)
        if err != nil {
            return err
        }

        // 设置压缩方式
        header.Method = zip.Deflate

        writer, err := zipWriter.CreateHeader(header)
        if err != nil {
            return err
        }

        _, err = io.Copy(writer, fileToZip)
        if err != nil {
            return err
        }
    }
    return nil
}

登录后复制

这里需要注意的是，header.Name 默认是从路径中提取的文件名。如果你想改变 ZIP 里的路径结构，可以手动设置 header.Name 的值。

如何解压一个 ZIP 文件

解压的过程相对简单一些：打开 ZIP 文件、遍历其中的条目，然后依次读取并写入本地磁盘。

使用 zipReader, _ := zip.OpenReader(path) 打开 ZIP 文件
遍历 zipReader.File 中的每一个条目
如果是目录，就创建对应目录；如果是文件，就打开并写入数据

func extractZip(zipPath string, targetDir string) error {
    reader, err := zip.OpenReader(zipPath)
    if err != nil {
        return err
    }
    defer reader.Close()

    for _, f := range reader.File {
        path := filepath.Join(targetDir, f.Name)

        if f.FileInfo().IsDir() {
            os.MkdirAll(path, os.ModePerm)
            continue
        }

        if err := os.MkdirAll(filepath.Dir(path), os.ModePerm); err != nil {
            return err
        }

        rc, err := f.Open()
        if err != nil {
            return err
        }
        defer rc.Close()

        outFile, err := os.OpenFile(path, os.O_WRONLY|os.O_CREATE|os.O_TRUNC, f.Mode())
        if err != nil {
            return err
        }
        defer outFile.Close()

        io.Copy(outFile, rc)
    }
    return nil
}

登录后复制

这段代码能处理 ZIP 中包含目录结构的情况。注意，在打开 ZIP 文件条目时，要记得关闭资源（比如 rc.Close()），避免资源泄露。

注意事项与常见问题

路径问题：ZIP 文件中的路径可能带有前缀或者层级结构，解压时如果不做处理可能会覆盖当前目录之外的内容。建议在解压前校验路径是否合法，防止“路径穿越”。
权限问题：压缩和解压都需要对目标路径有写权限，否则会报错。
压缩级别控制：archive/zip 默认使用 Deflate 算法，但不支持设置压缩等级，如需更高级别的控制，可以考虑第三方库。
大文件处理：如果压缩或解压的是非常大的文件，建议使用流式处理，避免一次性加载全部内容导致内存占用过高。