Golang项目中的文件压缩与解压实现方法

答案：使用archive/zip包可实现文件压缩与解压。压缩时遍历文件路径，写入ZIP并保持目录结构；解压时逐项还原文件，注意路径安全与权限处理。

在Golang项目中处理文件压缩与解压，常用的是archive/zip包来操作ZIP格式。下面介绍如何实现文件的压缩和解压功能，适合实际项目使用。

压缩多个文件为ZIP

要将多个文件或目录打包成一个ZIP文件，可以使用archive/zip配合os和io包进行写入操作。

基本思路是创建一个ZIP输出文件，然后遍历需要压缩的路径，逐个将文件写入ZIP归档中，并保持目录结构。

注意：写入时需正确设置文件头信息，尤其是文件权限和路径分隔符（应使用/）。

示例代码：

func compressFiles(zipFile string, files []string) error {
    outFile, err := os.Create(zipFile)
    if err != nil {
        return err
    }
    defer outFile.Close()

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

    for _, filePath := range files {
        err := filepath.Walk(filePath, func(path string, info os.FileInfo, err error) error {
            if err != nil {
                return err
            }

            // 创建相对路径
            relPath, _ := filepath.Rel(".", path)
            relPath = filepath.ToSlash(relPath)

            if info.IsDir() {
                // 添加目录条目
                _, err := zipWriter.Create(relPath + "/")
                return err
            }

            // 添加文件
            fileInZip, err := zipWriter.Create(relPath)
            if err != nil {
                return err
            }

            file, err := os.Open(path)
            if err != nil {
                return err
            }
            defer file.Close()

            _, err = io.Copy(fileInZip, file)
            return err
        })
        if err != nil {
            return err
        }
    }
    return nil
}

解压ZIP文件到指定目录

解压操作需要读取ZIP文件中的每个条目，根据类型（文件或目录）创建对应结构，并将内容写入磁盘。

关键点包括：确保目标路径安全（防止路径穿越），还原文件权限，以及正确处理目录创建。

示例代码：

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

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

        // 防止路径穿越攻击
        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
        }

        outFile, err := os.OpenFile(filePath, os.O_WRONLY|os.O_CREATE|os.O_TRUNC, file.Mode())
        if err != nil {
            return err
        }

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

        _, err = io.Copy(outFile, rc)
        rc.Close()
        outFile.Close()

        if err != nil {
            return err
        }
    }
    return nil
}

使用建议与注意事项

在实际项目中使用压缩解压功能时，注意以下几点可提升稳定性和安全性：

• 路径处理统一用filepath.ToSlash()：避免Windows下出现反斜杠导致ZIP结构异常。
• 检查文件大小限制：防止解压超大文件或恶意压缩炸弹。
• 验证目标路径合法性：特别是解压时，禁止写入系统关键目录。
• 支持进度反馈（可选）：大文件压缩/解压时可通过回调函数通知进度。
• 错误处理要细致：如部分文件失败是否中断整体流程，需根据业务决定。

基本上就这些。Golang标准库对ZIP的支持足够应对大多数场景，无需引入外部依赖。只要注意路径、权限和资源释放，就能写出健壮的压缩解压逻辑。

本文转载于：互联网 如有侵犯，请联系zhengruancom@outlook.com删除。
免责声明：正软商城发布此文仅为传递信息，不代表正软商城认同其观点或证实其描述。