Go语言zip归档文件的读写操作

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", "https://www.xinbaoku.com/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()
    }
}
运行结果如下:

go run main.go
文件名: Golang.txt
https://www.xinbaoku.com/golang/