Golang 如何做一个文件压缩解压工具_Golang zip 包操作项目实践-Golang-PHP中文网

Golang 如何做一个文件压缩解压工具_Golang zip 包操作项目实践

P粉602998670

发布： 2025-11-17 10:09:21

原创

723人浏览过

使用Golang的archive/zip包可轻松实现文件压缩解压。首先通过os.Create创建目标ZIP文件，再用zip.NewWriter生成写入器，遍历源文件并调用FileInfoHeader获取文件头，利用filepath.ToSlash确保路径兼容性，普通文件需读取内容写入ZIP，目录则创建对应条目；解压时用zip.OpenReader打开压缩包，遍历其中文件，检查路径合法性防止穿越攻击，目录直接创建，文件则逐个解压到指定位置，并注意关闭资源；实际应用中建议设置压缩级别、流式处理大文件、添加进度提示与错误处理，确保工具稳定安全。

golang 如何做一个文件压缩解压工具_golang zip 包操作项目实践

用 Golang 做一个文件压缩解压工具并不复杂，标准库中的 archive/zip 包提供了完整的支持。通过它，你可以轻松实现文件的打包、压缩、解压和读取操作。下面结合实际项目场景，一步步带你使用 zip 包完成常见功能。

创建 ZIP 压缩文件

要将一个或多个文件压缩成 ZIP 包，需要创建一个 zip.Writer，然后逐个写入文件内容。

基本步骤：

打开目标 ZIP 文件（os.Create）
用 zip.NewWriter 创建写入器
遍历待压缩的文件路径，逐个添加到 ZIP 中
每个文件需创建 zip.FileHeader，并写入原始数据
记得关闭 writer 和文件句柄

示例代码片段：

func compressFile(src, dst string) error {
    outFile, err := os.Create(dst)
    if err != nil {
        return err
    }
    defer outFile.Close()

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

    return filepath.Walk(src, func(path string, info os.FileInfo, err error) error {
        if err != nil {
            return err
        }

        header, _ := zip.FileInfoHeader(info)
        header.Name = filepath.ToSlash(path[len(src):]) // 保持相对路径

        if info.IsDir() {
            header.Name += "/"
            zipWriter.CreateHeader(header)
            return nil
        }

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

        content, err := os.ReadFile(path)
        if err != nil {
            return err
        }
        _, err = writer.Write(content)
        return err
    })
}

登录后复制

解压 ZIP 文件

解压的核心是读取 ZIP 包中的每个文件，并还原到指定目录。

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

笔目鱼英文论文写作器

写高质量英文论文，就用笔目鱼

查看详情

关键点：

使用 zip.OpenReader 打开 ZIP 文件
遍历其中的 *zip.File 对象
如果是目录，创建对应文件夹
如果是普通文件，读取内容并写入本地
注意处理路径安全问题，避免路径穿越攻击

示例代码片段：

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

    for _, file := range reader.File {
        filePath := filepath.Join(dest, file.Name)

        // 防止路径穿越
        if !strings.HasPrefix(filePath, filepath.Clean(dest)+string(os.PathSeparator)) {
            return fmt.Errorf("invalid file path: %s", filePath)
        }

        if file.FileInfo().IsDir() {
            os.MkdirAll(filePath, os.ModePerm)
            continue
        }

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

        inFile, err := file.Open()
        if err != nil {
            return err
        }

        outFile, err := os.Create(filePath)
        if err != nil {
            inFile.Close()
            return err
        }

        io.Copy(outFile, inFile)

        outFile.Close()
        inFile.Close()
    }
    return nil
}

登录后复制