📜  如何在 Golang 中解压缩文件?(1)

📅  最后修改于: 2023-12-03 15:24:10.921000             🧑  作者: Mango

如何在 Golang 中解压缩文件?

有时候,我们需要在 Golang 中解压缩一些文件,这可能是因为我们需要恢复备份文件、下载远程文件或解压缩压缩文件。下面,我们将介绍如何在 Golang 中使用内置的 archive/zip 模块进行文件解压缩操作。

准备工作

在开始使用 archive/zip 模块之前,我们需要确保我们已经将 Golang 安装在我们的计算机上。另外,您可能需要安装一些实用工具,如 7zipWinZip 等解压缩软件。

解压缩文件

首先,我们需要导入 archive/zip 模块。

import (
    "archive/zip"
    "fmt"
    "os"
)

我们可以使用以下代码来打开一个 Zip 文件并将其解压缩到指定目录中。

func unzip(zipFile string, dest string) error {
    r, err := zip.OpenReader(zipFile)
    if err != nil {
        return err
    }
    defer r.Close()
    for _, f := range r.File {
        // Store filename/path for returning and using later on
        fpath := filepath.Join(dest, f.Name)

        // Check for ZipSlip. More Info: http://bit.ly/2MsjAWE
        if !strings.HasPrefix(fpath, filepath.Clean(dest)+string(os.PathSeparator)) {
            return fmt.Errorf("illegal file path: %s", fpath)
        }

        if f.FileInfo().IsDir() {
            // Make Folder
            os.MkdirAll(fpath, os.ModePerm)
            continue
        }

        // Make File
        if err = os.MkdirAll(filepath.Dir(fpath), os.ModePerm); err != nil {
            return err
        }

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

        rc, err := f.Open()
        if err != nil {
            return err
        }

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

        // Close the file without defer to close before next iteration of loop
        outFile.Close()
        rc.Close()

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

该函数接受 Zip 文件路径和目标目录路径作为输入,并将包含的所有文件解压缩到目标目录中。

解压缩示例

下面是解压缩文件的简单示例。

package main

import (
    "fmt"
    "os"
)

func main() {
    zipFile := "example.zip"
    dest := "extracted"

    err := unzip(zipFile, dest)
    if err != nil {
        fmt.Println(err)
        os.Exit(1)
    }

    fmt.Println("Extraction Completed!")
}
结论

现在您已经知道如何在 Golang 中使用 archive/zip 模块进行文件解压缩操作了。记住,确保您已经安装了 Golang 并且已经安装了一些解压缩软件作为准备工作。