Go语言zip文件的读写操作

2023-05-14 08:46

短信预约 -IT技能 免费直播动态提醒

Go语言的标准库提供了对几种压缩格式的支持，其中包括 gzip，因此 Go 程序可以无缝地读写 .gz 扩展名的 gzip 压缩文件或非 .gz 扩展名的非压缩文件。此外标准库也提供了读和写 .zip 文件、tar 包文件（.tar 和 .tar.gz），以及读 .bz2 文件（即 .tar .bz2 文件）的功能。

本节我们将主要介绍 zip 归档文件的读写操作。

创建 zip 归档文件

Go语言提供了 archive/zip 包来操作压缩文件，下面通过一个简单的的示例演示如何使用Go语言来创建一个 zip 文件，示例代码如下：

package main
import (
    "archive/zip"
    "bytes"
    "fmt"
    "os"
)
func main() {
    // 创建一个缓冲区用来保存压缩文件内容
    buf := new(bytes.Buffer)
    // 创建一个压缩文档
    w := zip.NewWriter(buf)
    // 将文件加入压缩文档
    var files = []struct {
        Name, Body string
    }{
        {"Golang.txt", "http://jb51.net/golang/"},
    }
    for _, file := range files {
        f, err := w.Create(file.Name)
        if err != nil {
            fmt.Println(err)
        }
        _, err = f.Write([]byte(file.Body))
        if err != nil {
            fmt.Println(err)
        }
    }
    // 关闭压缩文档
    err := w.Close()
    if err != nil {
        fmt.Println(err)
    }
    // 将压缩文档内容写入文件
    f, err := os.OpenFile("file.zip", os.O_CREATE|os.O_WRONLY, 0666)
    if err != nil {
        fmt.Println(err)
    }
    buf.WriteTo(f)
}

运行上面的文件会在当前目录下生成 file.zip 文件，如下图所示：

读取 zip 归档文件

读取一个 .zip 归档文件与创建一个归档文件一样简单，只是如果归档文件中包含带有路径的文件名，就必须重建目录结构。

示例代码如下所示：

package main
import (
    "archive/zip"
    "fmt"
    "io"
    "os"
)
func main() {
    // 打开一个zip格式文件
    r, err := zip.OpenReader("file.zip")
    if err != nil {
        fmt.Printf(err.Error())
    }
    defer r.Close()
    // 迭代压缩文件中的文件，打印出文件中的内容
    for _, f := range r.File {
        fmt.Printf("文件名: %s\n", f.Name)
        rc, err := f.Open()
        if err != nil {
            fmt.Printf(err.Error())
        }
        _, err = io.CopyN(os.Stdout, rc, int64(f.UncompressedSize64))
        if err != nil {
            fmt.Printf(err.Error())
        }
        rc.Close()
    }
}

运行结果如下：