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", "http://www.cxmall.cn/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