首页 > 编程笔记 > Go语言笔记 阅读:5

Go语言读写zip文件(附带实例)

Go 语言的标准库支持多种压缩格式,其中包括写入和读取 zip 文件。

下面通过实例演示如何写入和读取 zip 文件。

【实例 1】写入 zip 文件。在当前程序所在的目录下,创建 file.zip 文件。在 zip 文件中,包含一个名为“C语言中文网教程”的文本文件。在这个文本文件中包含一个网址,即 https://c.biancheng.net/。
package main

import (
    "archive/zip"
    "bytes"
    "fmt"
    "os"
)

func main() {
    // 创建缓冲区,保存压缩文件内容
    buf := new(bytes.Buffer)

    // 创建一个压缩文档
    w := zip.NewWriter(buf)

    // 将文件加入压缩文档
    var files = []struct {
        Name, Body string
    }{
        {"C语言中文网教程.txt", "https://c.biancheng.net/"},
    }

    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 文件。双击 file.zip,即可看到 zip 文件中 C语言中文网教程.txt文件。

【实例 2】读取 zip 文件。读取实例 1 生成的 file.zip,把 file.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()
    }
}
运行结果如下:

文件名: C语言中文网教程.txt
https://c.biancheng.net/

相关文章