Golang 中的 archive/zip 包详解（一）：实现 ZIP 压缩与解压

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

实现压缩功能

1、首先需要创建一个 zip 文件。

zip 文件也是一个文件，首先需要创建一个基础的 zip 文件，使用 os.Create() 函数创建即可。

package main

import "os"

func main() {
	//第一步，创建 zip 文件
	zipFile, err := os.Create("archive.zip")
	if err != nil {
		panic(err)
	}
	defer zipFile.Close()
}

2、使用 zip.NewWriter() 函数创建一个新的 *Writer 对象，用于将待压缩的文件写入到 zip 文件。

package main

import (
	"archive/zip"
	"os"
)

func main() {
	//第一步，创建 zip 文件
	zipFile, err := os.Create("archive.zip")
	if err != nil {
		panic(err)
	}
	defer zipFile.Close()

	//第二步，创建一个新的 *Writer 对象
	zipWriter := zip.NewWriter(zipFile)
	
}

3、向 zip 文件中添加第一个文件

创建完 zip writer 后，向 zip 文件中添加一个文件

package main

import (
	"archive/zip"
	"io"
	"os"
)

func main() {
	//第一步，创建 zip 文件
	zipFile, err := os.Create("archive.zip")
	if err != nil {
		panic(err)
	}
	defer zipFile.Close()

	//第二步，创建一个新的 *Writer 对象
	zipWriter := zip.NewWriter(zipFile)

	//第三步，向 zip 文件中添加第一个文件
	//
	//向 zip 文件中添加一个文件，返回一个待压缩的文件内容应写入的 Writer
	w, err := zipWriter.Create("json.go")
	if err != nil {
		panic(err)
	}
	// 打开待压缩的文件
	f, err := os.Open("bec/json.go")
	if err != nil {
		panic(err)
	}
	defer f.Close()

	//需要压缩更多文件到压缩包的话，只需要重复第三步即可，只需要更换下文件名

	if _, err := io.Copy(w, f); err != nil {
		panic(err)
	}
}

4、关闭 zip writer，将所有数据写入指向基础 zip 文件的数据流

package main

import (
	"archive/zip"
	"io"
	"os"
)

func main() {
	//第一步，创建 zip 文件
	zipFile, err := os.Create("archive.zip")
	if err != nil {
		panic(err)
	}
	defer zipFile.Close()

	//第二步，创建一个新的 *Writer 对象
	zipWriter := zip.NewWriter(zipFile)

	//第三步，向 zip 文件中添加第一个文件
	//
	//向 zip 文件中添加一个文件，返回一个待压缩的文件内容应写入的 Writer
	w, err := zipWriter.Create("json.go")
	if err != nil {
		panic(err)
	}
	// 打开待压缩的文件
	f, err := os.Open("bec/json.go")
	if err != nil {
		panic(err)
	}
	defer f.Close()

	//需要压缩更多文件到压缩包的话，只需要重复第三步即可，只需要更换下文件名

	if _, err := io.Copy(w, f); err != nil {
		panic(err)
	}

	// 第四步，关闭 zip writer，将所有数据写入指向基础 zip 文件的数据流
	zipWriter.Close()
}

解压 ZIP 包

1、 使用 zip.OpenReader 打开 zip 包

package main

import (
	"archive/zip"
)

func main() {
	// 第一步，打开 zip 文件
	zipFile, err := zip.OpenReader("archive.zip")
	if err != nil {
		panic(err)
	}
	defer zipFile.Close()
}

2、 遍历 zip 中的文件，将压缩包的中的文件读取到本地文件

package main

import (
	"archive/zip"
	"io"
	"os"
	"path/filepath"
)

func main() {
	// 第一步，打开 zip 文件
	zipFile, err := zip.OpenReader("archive.zip")
	if err != nil {
		panic(err)
	}
	defer zipFile.Close()

	// 第二步，遍历 zip 中的文件
	for _, f := range zipFile.File {
		filePath := f.Name
		if f.FileInfo().IsDir() {
			_ = os.MkdirAll(filePath, os.ModePerm)
			continue
		}
    // 创建对应文件夹
		if err := os.MkdirAll(filepath.Dir(filePath), os.ModePerm); err != nil {
			panic(err)
		}
    // 解压到的目标文件
		dstFile, err := os.OpenFile(filePath, os.O_WRONLY|os.O_CREATE|os.O_TRUNC, f.Mode())
		if err != nil {
			panic(err)
		}
		file, err := f.Open()
		if err != nil {
			panic(err)
		}
    // 写入到解压到的目标文件
		if _, err := io.Copy(dstFile, file); err != nil {
			panic(err)
		}
		dstFile.Close()
		file.Close()
	}
}