golang zip add files

使用Golang对文件进行ZIP压缩的方法

Go语言（Golang）是一个由Google开发的开源编程语言，它依赖于强大而灵活的标准库，提供了丰富的功能和工具。其中之一是对文件进行ZIP压缩和解压缩的支持。本文将向您展示如何使用Golang对文件进行ZIP压缩。

首先，我们需要引入`archive/zip`包来处理ZIP相关的操作。在您的代码中加入以下代码行：

import "archive/zip"

压缩单个文件

要压缩单个文件，请按照以下步骤进行操作：

1. 打开一个新的ZIP文件，您可以使用`Create`函数来创建一个新的ZIP文件。
2. 通过调用`Create`函数返回的返回值的`Create`方法，将文件写入到ZIP文件中。
3. 关闭ZIP文件以确保所有更改都已保存。

下面是一个例子，展示了如何将文件添加到ZIP文件中：

func zipFile(filename string){
    newZipFile, err := os.Create("newZip.zip")
    if err != nil {
        panic(err)
    }
    defer newZipFile.Close()

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

    fileToAdd, err := os.Open(filename)
    if err != nil {
        panic(err)
    }
    defer fileToAdd.Close()

    fileInfo, err := fileToAdd.Stat()
    if err != nil {
        panic(err)
    }
    
    header, err := zip.FileInfoHeader(fileInfo)
    if err != nil {
        panic(err)
    }
    
    writer, err := zipWriter.CreateHeader(header)
    if err != nil {
        panic(err)
    }
    
    _, err = io.Copy(writer, fileToAdd)
    if err != nil {
        panic(err)
    }
}

在上面的代码示例中，我们首先创建一个新的ZIP文件，并确保在函数结束时关闭它。然后，我们使用`zip.NewWriter`函数创建一个ZIP写入器，以便将文件写入到ZIP文件中。

接下来，我们打开要添加到ZIP文件的文件并检查是否有错误。通过调用`FileInfoHeader`函数，我们可以使用文件的元数据创建一个ZIP文件头对象。最后，我们调用`zipWriter.CreateHeader`函数创建一个新的ZIP文件条目，并使用`io.Copy`函数将文件的内容复制到ZIP文件中。

压缩多个文件

要压缩多个文件，我们可以通过循环迭代文件列表，并重复上述添加单个文件的过程。下面是一个展示如何压缩多个文件的例子：

func zipFiles(filenames []string){
    newZipFile, err := os.Create("newZip.zip")
    if err != nil {
        panic(err)
    }
    defer newZipFile.Close()

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

    for _, filename := range filenames {
        fileToAdd, err := os.Open(filename)
        if err != nil {
            panic(err)
        }
        defer fileToAdd.Close()

        fileInfo, err := fileToAdd.Stat()
        if err != nil {
            panic(err)
        }
        
        header, err := zip.FileInfoHeader(fileInfo)
        if err != nil {
            panic(err)
        }
        
        writer, err := zipWriter.CreateHeader(header)
        if err != nil {
            panic(err)
        }
        
        _, err = io.Copy(writer, fileToAdd)
        if err != nil {
            panic(err)
        }
    }
}

在上面的代码示例中，我们首先创建一个新的ZIP文件，并确保在函数结束时关闭它。然后，我们使用`zip.NewWriter`函数创建一个ZIP写入器，以便将文件写入到ZIP文件中。

接下来，我们循环迭代文件名列表，并为每个文件创建一个新的ZIP条目。通过调用`zipWriter.CreateHeader`函数来创建ZIP文件条目，并使用`io.Copy`函数将文件的内容复制到ZIP文件中。

压缩文件夹

如果您想要压缩整个文件夹，只需将文件夹路径作为参数传递给`zipFiles`函数即可。下面是一个示例：

func zipFolder(folderPath string){
    var filenames []string
    
    err := filepath.Walk(folderPath, func(path string, info os.FileInfo, err error) error {
        if err != nil {
            return err
        }
        
        if !info.IsDir() {
            filenames = append(filenames, path)
        }
        
        return nil
    })
    if err != nil {
        panic(err)
    }

    zipFiles(filenames)
}

在上面的代码示例中，我们使用`filepath.Walk`函数遍历指定的文件夹，并将每个文件的路径添加到一个字符串切片中。然后，我们调用`zipFiles`函数来压缩这些文件。

总结

本文向您介绍了如何使用Golang对文件进行ZIP压缩。我们展示了如何压缩单个文件、多个文件以及整个文件夹。您可以根据自己的需求和场景灵活运用这些方法。希望这篇文章能对您有所帮助！