golang 压缩tar

2025-01-19 19:16:30 编程来源：ZONE.CI 全球网 0 阅读模式

在现代软件开发过程中，文件的压缩和解压通常是一个很常见的需求。而Tar.Gz是一种非常流行的压缩格式，它不仅可以压缩多个文件和文件夹，还可以保持文件夹结构和权限。本文将向您介绍如何用Golang来进行Tar.Gz文件的压缩和解压，希望能为您提供一些帮助。

使用Golang压缩文件到Tar.Gz格式

首先我们需要导入`archive/tar`和`compress/gzip`两个包来处理Tar压缩和Gzip压缩。接下来，我们将通过代码来演示如何将文件或文件夹压缩到Tar.Gz格式。首先，我们创建一个空的Tar文件：

tarFile, err := os.Create("example.tar")
if err != nil {
    log.Fatal(err)
}
defer tarFile.Close()

然后，我们创建一个Gzip写入器，将其与创建的Tar文件绑定：

gzWriter := gzip.NewWriter(tarFile)
defer gzWriter.Close()

接着，我们使用Tar包装上一步骤创建的Gzip写入器：

tarWriter := tar.NewWriter(gzWriter)
defer tarWriter.Close()

现在，我们可以将要压缩的文件或文件夹添加到Tar文件中：

fileToCompress, err := os.Open("/path/to/file")
if err != nil {
    log.Fatal(err)
}
defer fileToCompress.Close()

fileInfo, err := fileToCompress.Stat()
if err != nil {
    log.Fatal(err)
}

header, err := tar.FileInfoHeader(fileInfo, "")
if err != nil {
    log.Fatal(err)
}

err = tarWriter.WriteHeader(header)
if err != nil {
    log.Fatal(err)
}

_, err = io.Copy(tarWriter, fileToCompress)
if err != nil {
    log.Fatal(err)
}

重复以上步骤，您可以将多个文件或文件夹添加到同一个Tar文件中。最后，我们关闭Tar写入器和Gzip写入器，以确保所有的数据都被刷新到文件中：

tarWriter.Close()
gzWriter.Close()

使用Golang解压Tar.Gz文件

接下来，我们将展示如何使用Golang解压缩Tar.Gz文件。我们需要导入与压缩相同的两个包：`archive/tar`和`compress/gzip`。

首先，我们打开要解压的Tar文件：

fileToDecompress, err := os.Open("example.tar.gz")
if err != nil {
    log.Fatal(err)
}
defer fileToDecompress.Close()

然后，我们创建一个Gzip读取器，并将其与打开的Tar文件绑定：

gzReader, err := gzip.NewReader(fileToDecompress)
if err != nil {
    log.Fatal(err)
}
defer gzReader.Close()

然后，我们使用Tar包装上一步骤创建的Gzip读取器：

tarReader := tar.NewReader(gzReader)

现在，我们可以遍历Tar文件中的每个文件或文件夹，并将其解压到指定的目录：

for {
    header, err := tarReader.Next()
    if err == io.EOF {
        break
    }
    if err != nil {
        log.Fatal(err)
    }

    filePath := filepath.Join("/path/to/destination", header.Name)

    switch header.Typeflag {
    case tar.TypeDir:
        err = os.MkdirAll(filePath, os.ModePerm)
        if err != nil {
            log.Fatal(err)
        }
    case tar.TypeReg:
        outputFile, err := os.Create(filePath)
        if err != nil {
            log.Fatal(err)
        }
        defer outputFile.Close()

        _, err = io.Copy(outputFile, tarReader)
        if err != nil {
            log.Fatal(err)
        }
    default:
        log.Fatalf("unknown type: %v in %s", header.Typeflag, header.Name)
    }
}

最后，我们关闭Gzip读取器，以确保所有的数据都被清理和释放：