首页  >  文章  >  后端开发  >  如何在 Go 中提取 TAR.GZ 文件?

如何在 Go 中提取 TAR.GZ 文件?

Patricia Arquette
Patricia Arquette原创
2024-11-03 20:54:29349浏览

How to Extract a TAR.GZ File in Go?

在 Go 中解压 TAR.GZ 文件

在 Go 中解压 TAR.GZ 文件涉及使用 gzip 包读取压缩文件并使用 archive/tar 包将其内容解析为 tarball。这是代码的更正版本:

<code class="go">package main

import (
    "archive/tar"
    "compress/gzip"
    "fmt"
    "io"
    "log"
    "os"
)

func ExtractTarGz(gzipStream io.Reader) {
    uncompressedStream, err := gzip.NewReader(gzipStream)
    if err != nil {
        log.Fatal("ExtractTarGz: NewReader failed")
    }

    tarReader := tar.NewReader(uncompressedStream)

    for true {
        header, err := tarReader.Next()

        if err == io.EOF {
            break
        }

        if err != nil {
            log.Fatalf("ExtractTarGz: Next() failed: %s", err.Error())
        }

        switch header.Typeflag {
        case tar.TypeDir:
            if err := os.Mkdir(header.Name, 0755); err != nil {
                log.Fatalf("ExtractTarGz: Mkdir() failed: %s", err.Error())
            }
        case tar.TypeReg:
            outFile, err := os.Create(header.Name)
            if err != nil {
                log.Fatalf("ExtractTarGz: Create() failed: %s", err.Error())
            }
            if _, err := io.Copy(outFile, tarReader); err != nil {
                log.Fatalf("ExtractTarGz: Copy() failed: %s", err.Error())
            }
            outFile.Close()

        default:
            log.Fatalf(
                "ExtractTarGz: uknown type: %s in %s",
                header.Typeflag,
                header.Name)
        }

    }
}

func main() {
    r, err := os.Open("./file.tar.gz")
    if err != nil {
        fmt.Println("error")
    }
    ExtractTarGz(r)
}</code>

原始代码的主要问题是写入后没有关闭输出文件。这可能会导致数据损坏和“打开文件过多”错误。

更正代码中的改进:

  • 更正代码会关闭输出文件写入后。
  • 如果文件无法打开,它会在控制台打印一条错误消息。

以上是如何在 Go 中提取 TAR.GZ 文件?的详细内容。更多信息请关注PHP中文网其他相关文章!

声明:
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn