Golang 中的 archive/zip 包详解（三）：常用函数

Golang 中的 archive/zip 包用于处理 ZIP 格式的压缩文件，提供了一系列用于创建、读取和解压缩 ZIP 格式文件的函数和类型，使用起来非常方便，本文讲解下常用函数。

zip.OpenReader

定义如下：

func OpenReader(name string) (*ReadCloser, error) 

打开一个 ZIP 文件，并返回一个 zip.ReadCloser 类型的对象。简单使用示例如下：

package main

import (
	"archive/zip"
	"fmt"
	"log"
)

func main() {
	zipFile := "path/to/myfile.zip"
	reader, err := zip.OpenReader(zipFile)
	if err != nil {
		log.Fatal(err)
	}
	defer reader.Close()

	// 读取 ZIP 文件中的内容
	for _, file := range reader.File {
		fmt.Println("文件名:", file.Name)
		fmt.Println("文件大小:", file.UncompressedSize64)
	}
}

zip.NewReader

定义如下：

func NewReader(r io.ReaderAt, size int64) (*Reader, error)

创建一个新的 zip.Reader 类型的对象用于读取ZIP文件，简单使用示例如下：

package main

import (
	"archive/zip"
	"fmt"
	"log"
	"strings"
)

func main() {
	zipData := getZipData() // 假设获取到了 ZIP 文件的字节数据
	reader, err := zip.NewReader(strings.NewReader(string(zipData)), int64(len(zipData)))
	if err != nil {
		log.Fatal(err)
	}

	// 现在你可以使用 reader 对象来读取 ZIP 文件中的内容了
	for _, file := range reader.File {
		fmt.Println("文件名:", file.Name)
		fmt.Println("文件大小:", file.UncompressedSize64)
	}
}

func getZipData() []byte {
	// 假设这里是获取 ZIP 文件的字节数据的逻辑
	// 这里只是示例，你需要根据实际情况来获取 ZIP 文件的数据
	zipData := []byte{80, 75, 3, 4, 20, 0, 8, 8, 0, 0, 0, 0, 90, 90, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0}
	return zipData
}

zip.NewWriter

定义如下：

func NewWriter(w io.Writer) *Writer 

创建一个新的 zip.Writer 类型的对象用于写入 ZIP 文件。使用示例可以参考之前的第一篇文章《Golang 中的 archive/zip 包详解（一）：实现 ZIP 压缩与解压》。

zip.FileInfoHeader

定义如下：

func FileInfoHeader(fi fs.FileInfo) (*FileHeader, error) 

将文件信息转换为 zip.FileHeader 结构体，以便设置 ZIP 文件中每个文件的属性，简单使用示例如下：

package main

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

func main() {
	fileName := "file.txt"
	fileInfo, err := os.Stat(fileName)
	if err != nil {
		log.Fatal(err)
	}

	header, err := zip.FileInfoHeader(fileInfo)
	if err != nil {
		log.Fatal(err)
	}

	fmt.Println("文件名:", header.Name)
	fmt.Println("文件大小:", header.UncompressedSize64)
}

zip.RegisterCompressor

定义如下：

func RegisterCompressor(method uint16, comp Compressor) 

用来注册自定义的压缩方法。

zip.RegisterDecompressor

定义如下：

func RegisterDecompressor(method uint16, dcomp Decompressor) 

用来注册自定义的解压缩方法。

小结

以上是 archive/zip 包中的一些常用函数，通过使用这些函数，可以方便地创建、读取和修改 ZIP 文件。