Golang Zip压缩|极客笔记

Golang Zip压缩

在软件开发中，文件的压缩和解压缩是非常常见的操作。其中，Zip格式是一种常用的压缩格式，它可以将多个文件或目录打包为一个单一的文件，方便存储和传输。在Golang中，我们可以使用标准库中的archive/zip来实现Zip文件的压缩和解压缩操作。

本文将详细介绍在Golang中如何使用archive/zip包来进行Zip文件的压缩操作。我们将分为以下几个部分进行讲解：

导入必要的包
创建和写入Zip文件
压缩单个文件
压缩多个文件
打包文件夹
运行示例代码
总结

1. 导入必要的包

首先，我们需要导入archive/zip和os包，以便在Golang中使用Zip文件的功能。代码示例如下：

package main

import (
    "archive/zip"
    "os"
)

2. 创建和写入Zip文件

在Golang中，可以通过archive/zip包提供的zip.NewWriter方法来创建一个Zip文件并向其中写入内容。首先，我们需要创建一个Zip文件，然后逐个向其中添加文件或目录，并最终将Zip文件保存到磁盘上。

func createZipFile(zipFileName string) error {
    // 创建一个新的Zip文件
    zipFile, err := os.Create(zipFileName)
    if err != nil {
        return err
    }
    defer zipFile.Close()

    // 创建一个Zip写入器
    zipWriter := zip.NewWriter(zipFile)

    // 关闭Zip写入器
    defer zipWriter.Close()

    return nil
}

3. 压缩单个文件

接下来，我们可以编写一个函数来将单个文件压缩到Zip文件中。在这个函数中，我们需要打开要压缩的源文件，将其内容写入到Zip文件中，并设置文件的一些元数据，如文件名和权限等。

func compressSingleFile(zipWriter *zip.Writer, sourceFile string) error {
    // 打开要压缩的源文件
    fileToZip, err := os.Open(sourceFile)
    if err != nil {
        return err
    }
    defer fileToZip.Close()

    // 获取源文件的文件信息
    fileInfo, err := fileToZip.Stat()
    if err != nil {
        return err
    }

    // 创建一个Zip文件头
    zipHeader, err := zip.FileInfoHeader(fileInfo)
    if err != nil {
        return err
    }

    // 设置Zip文件头的文件名
    zipHeader.Name = sourceFile

    // 将Zip文件头写入Zip文件
    writer, err := zipWriter.CreateHeader(zipHeader)
    if err != nil {
        return err
    }

    // 将源文件内容写入Zip文件
    _, err = io.Copy(writer, fileToZip)
    if err != nil {
        return err
    }

    return nil
}

4. 压缩多个文件

有时候，我们需要将多个文件一起压缩到Zip文件中。下面是一个函数，用于将多个文件压缩到同一个Zip文件中。

func compressMultipleFiles(zipWriter *zip.Writer, sourceFiles []string) error {
    for _, file := range sourceFiles {
        err := compressSingleFile(zipWriter, file)
        if err != nil {
            return err
        }
    }

    return nil
}

5. 打包文件夹

除了压缩多个文件之外，有时候我们也需要将整个文件夹打包为一个Zip文件。下面是一个函数，用于实现将一个文件夹及其所有内容压缩到一个Zip文件中。

func compressFolder(zipWriter *zip.Writer, folderPath string) error {
    // 遍历文件夹中的所有文件
    files, err := ioutil.ReadDir(folderPath)
    if err != nil {
        return err
    }

    for _, file := range files {
        filePath := filepath.Join(folderPath, file.Name())

        if file.IsDir() {
            // 递归压缩子文件夹
            err := compressFolder(zipWriter, filePath)
            if err != nil {
                return err
            }
        } else {
            // 压缩文件
            err := compressSingleFile(zipWriter, filePath)
            if err != nil {
                return err
            }
        }
    }

    return nil
}

6. 运行示例代码

下面我们将结合上面的函数示例来展示如何在Golang中进行Zip文件的压缩操作。在本示例中，我们将压缩testfile.txt和testdata文件夹，并将它们写入到output.zip文件中。

package main

import (
    "archive/zip"
    "io"
    "os"
    "path/filepath"
    "io/ioutil"
)

func main() {
    // 创建一个新的Zip文件
    zipFileName := "output.zip"
    zipFile, err := os.Create(zipFileName)
    if err != nil {
        panic(err)
    }
    defer zipFile.Close()

    // 创建一个Zip写入器
    zipWriter := zip.NewWriter(zipFile)
    defer zipWriter.Close()

    // 压缩单个文件
    err = compressSingleFile(zipWriter, "testfile.txt")
    if err != nil {
        panic(err)
    }

    // 压缩文件夹
    err = compressFolder(zipWriter, "testdata")
    if err != nil {
        panic(err)
    }

    println("文件压缩成功：", zipFileName)
}

运行上述示例代码，将在同级目录下生成一个名为output.zip的Zip文件，其中包含了testfile.txt和testdata文件夹的内容。通过解压这个Zip文件，你可以看到被压缩的文件和文件夹。