Golang 操作压缩文件的实用指南

了解压缩文件操作

在 Golang 中，我们可以使用内置的 "archive/zip" 和 "archive/tar" 包来处理压缩文件。这些包提供了一组简单而强大的函数和类型，可用于创建、读取和操作各种常见的压缩格式，如 ZIP 和 TAR。

使用 "archive/zip" 包处理 ZIP 文件

"archive/zip" 包允许我们在 Golang 中创建和读取 ZIP 文件。要创建一个 ZIP 文件，我们可以使用 "Create" 函数并提供一个文件名作为参数。然后，我们可以使用 "Create" 函数返回的 Writer 来向 ZIP 文件中添加文件或目录。下面是一个简单的例子：

```go package main import ( "archive/zip" "os" ) func main() { file, _ := os.Create("example.zip") defer file.Close() writer := zip.NewWriter(file) defer writer.Close() // 添加文件到 ZIP fileToZip, _ := os.Open("file.txt") defer fileToZip.Close() fileInfo, _ := fileToZip.Stat() header, _ := zip.FileInfoHeader(fileInfo) header.Name = "file.txt" writer.WriteHeader(header) io.Copy(writer, fileToZip) } ```

要读取 ZIP 文件并提取其中的文件，我们可以使用 "OpenReader" 函数来打开 ZIP 文件。然后，我们可以遍历打开文件返回的 Reader 中的所有文件，并将它们解压缩到指定的目录。下面是一个简单的读取 ZIP 文件的例子：

```go package main import ( "archive/zip" "fmt" "io" "os" ) func main() { zipFile, _ := os.Open("example.zip") defer zipFile.Close() reader, _ := zip.OpenReader(zipFile) defer reader.Close() for _, file := range reader.File { extractedFilePath := fmt.Sprintf("extracted/%s", file.Name) if file.FileInfo().IsDir() { os.MkdirAll(extractedFilePath, os.ModePerm) continue } extractedFile, _ := os.OpenFile(extractedFilePath, os.O_WRONLY|os.O_CREATE|os.O_TRUNC, file.Mode()) defer extractedFile.Close() fileReader, _ := file.Open() defer fileReader.Close() io.Copy(extractedFile, fileReader) } } ```

使用 "archive/tar" 包处理 TAR 文件

"archive/tar" 包允许我们在 Golang 中创建和读取 TAR 文件。要创建一个 TAR 文件，我们可以使用 "Create" 函数并提供一个文件名作为参数。然后，我们可以使用 "Create" 函数返回的 Writer 来向 TAR 文件中添加文件或目录。下面是一个简单的例子：

```go package main import ( "archive/tar" "os" ) func main() { file, _ := os.Create("example.tar") defer file.Close() writer := tar.NewWriter(file) defer writer.Close() // 添加文件到 TAR fileToTar, _ := os.Open("file.txt") defer fileToTar.Close() fileInfo, _ := fileToTar.Stat() header := &tar.Header{ Name: "file.txt", Mode: int64(fileInfo.Mode()), Size: fileInfo.Size(), } writer.WriteHeader(header) io.Copy(writer, fileToTar) } ```

要读取 TAR 文件并提取其中的文件，我们可以使用 "OpenReader" 函数来打开 TAR 文件。然后，我们可以遍历打开文件返回的 Reader 中的所有文件，并将它们解压缩到指定的目录。下面是一个简单的读取 TAR 文件的例子：

```go package main import ( "archive/tar" "fmt" "io" "os" ) func main() { tarFile, _ := os.Open("example.tar") defer tarFile.Close() reader := tar.NewReader(tarFile) for { header, err := reader.Next() if err == io.EOF { break } extractedFilePath := fmt.Sprintf("extracted/%s", header.Name) switch header.Typeflag { case tar.TypeDir: os.MkdirAll(extractedFilePath, os.ModePerm) case tar.TypeReg: extractedFile, _ := os.OpenFile(extractedFilePath, os.O_WRONLY|os.O_CREATE|os.O_TRUNC, header.FileInfo().Mode()) defer extractedFile.Close() io.Copy(extractedFile, reader) } } } ```

总结

通过使用 "archive/zip" 和 "archive/tar" 包，我们可以方便地在 Golang 中进行压缩文件的创建、读取和操作。无论是处理 ZIP 文件还是 TAR 文件，这些包都为我们提供了简单且易于使用的 API。希望本文可以帮助你更好地理解和应用 Golang 中的压缩文件操作。