Go语言的标准库提供了对几种压缩格式的支持,其中包括 gzip,因此 Go 程序可以无缝地读写 .gz 扩展名的 gzip 压缩文件或非 .gz 扩展名的非压缩文件。此外标准库也提供了读和写 .zip 文件、tar 包文件(.tar 和 .tar.gz),以及读 .bz2 文件(即 .tar .bz2 文件)的功能。
本节我们将主要介绍 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://www.cdsy.xyz/computer/programme/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 归档文件与创建一个归档文件一样简单,只是如果归档文件中包含带有路径的文件名,就必须重建目录结构。
示例代码如下所示:
- 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()
- }
- }
运行结果如下: