如何在Golang项目中实现文件压缩与解压-Golang-PHP中文网

如何在Golang项目中实现文件压缩与解压

P粉602998670

发布： 2025-11-03 18:58:02

原创

868人浏览过

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

如何在golang项目中实现文件压缩与解压

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

压缩多个文件为ZIP

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

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

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

示例代码：

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

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文件中的每个条目，根据类型（文件或目录）创建对应结构，并将内容写入磁盘。

笔目鱼英文论文写作器

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

查看详情

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

示例代码：

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

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
}

登录后复制