【Go语言】Go进行 无密码zip 文件的压缩和解压以及带密码zip文件压缩和解压（详细版）

Go 和许多其他编程语言一样，支持从标准库中直接进行 zip 文件的压缩和解压。在 Go 中利用标准库包 archive/zip 完成 zip 文件的创建和提取。golang zip包的解压有官方的zip包（archive/zip），但是官方给的zip解压包代码只有解压不带密码的zip包，如果我们要解压带密码的zip就做不了, 解决方案：使用 github.com/yeka/zip。

一、压缩操作

1.1 第一步：创建一个 zip 基础文件

zip 文件也是一个文件，我们要做的第一件事是创建为一个简单的文件作为 zip 文件，就像在 Go 中处理任何其他文件一样。使用 os 第三方包的 os.Create() 函数创建一个文件对象。

 func Create(name string) (*File, error)

1.2 第二步：初始化 zip.Writer

使用 archive/zip包中的 zip.NewWriter 函数初始化一个 zip.Writer，用于将数据（文件和目录）写入 zip 文件。

    func NewWriter(w io.Writer) *Writer

1.3 第三步：使用 zip.Writer.Create创建一个 io.Writer

一旦创建完 zip writer 后，便可以使用 zip.Writer.Create 向 zip 文件中添加一个文件或目录。它返回一个待压缩的文件内容应写入的 Writer，文件内容将使用 Deflate 方法进行压缩。

    func (w *Writer) Create(name string) (io.Writer, error)

1.4 第四步：使用 io.Copy或 io.Writer.Write 写数据到到 zip 文件

zip.Writer.Create 函数返回一个 io.Writer ，因此任何文件内容都可以通过流式写入该 writer。使用 io.Copy 或调用 writer 的 Write 方法。

   func Copy(dst Writer, src Reader) (written int64, err error)

1.5 第五步：使用 zip.Writer.Close 关闭 zip 文件

将所有文件和目录写入 zip 文件后，可以通过 zip.Writer.Close 方法关闭 zip writer，将所有数据写入指向基础 zip 文件的数据流。注意，它不会关闭指向基础 zip 文件的 Writer。

    func (w *Writer) Close() error

1.6 压缩无密码zip包（源码案例）

package main

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

func main_zip_file() {
	fmt.Println("creating zip archive...")
	// 1、创建 zip 基础文件
	archive, err := os.Create("D:\\CODEFile01\\GoPro\\unZipPro\\src\\data\\archive.zip")
	if err != nil {
		panic(err)
	}
	defer archive.Close()

	// 2、初始化zip写入对象
	zipWriter := zip.NewWriter(archive)
	fmt.Println("opening first file...")
	f1, err := os.Open("D:\\CODEFile01\\GoPro\\unZipPro\\src\\data\\test.csv")
	if err != nil {
		panic(err)
	}
	defer f1.Close()
	fmt.Println("writing first file to archive...")

	// 3、使用 zip.Writer.Create 创建一个 io.Writer,使用Create() 向zip文件中添加一个文件或目录
	w1, err := zipWriter.Create("test.csv")
	if err != nil {
		panic(err)
	}

	// 4、使用 io.Copy 或 io.Writer.Write 写数据到到zip文件
	if _, err := io.Copy(w1, f1); err != nil {
		panic(err)
	}

	fmt.Println("opening second file...")
	f2, err := os.Open("D:\\CODEFile01\\GoPro\\unZipPro\\src\\data\\test.txt")
	if err != nil {
		panic(err)
	}
	defer f2.Close()

	fmt.Println("writing second file to archive...")
	w2, err := zipWriter.Create("test.txt")
	if err != nil {
		panic(err)
	}
	if _, err := io.Copy(w2, f2); err != nil {
		panic(err)
	}
	fmt.Println("closing zip archive...")

	// 5、使用 zip.Writer.Close 关闭 zip 文件
	zipWriter.Close()
}

二、解压操作

2.1 第一步：使用 zip.OpenReader 打开 zip 文件

要想解压 zip 文件我们可能需要做的第一件事是打开它。我们可以使用 archive/zip 包提供的函数 zip.OpenReader 来打开 zip 文件，以 io.ReadCloser 的形式返回一个 zip.Reader 的实例。

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

2.2 第二步：循环访问 zip 中的文件

zip.OpenReader 返回一个 zip.Reader 实例，其包含一个 zip.File 切片。

  type Reader struct {
  	File []*File
  	Comment string
  	// contains filtered or unexported fields

2.3 第三步：使用 zip.File.Open方法读取 zip 中文件的内容

   func (f *File) Open() (io.ReadCloser, error)

2.4 第四步：使用 io.Copy 或 io.Writer.Write 保存解压后的文件内容**

   func Copy(dst Writer, src Reader) (written int64, err error)

2.5 第五步：使用 zip.Reader.Close 关闭 zip 文件**

读取完 zip 存档中所有文件的内容并保存到指定位置后，要用接口 ReadCloser 中的方法 Close 关闭文件句柄。

    type ReadCloser interface {
    	Reader
    	Closer
    }
    
    // Closer is the interface that wraps the basic Close method.
    // The behavior of Close after the first call is undefined.
    // Specific implementations may document their own behavior.
    type Closer interface {
    	Close() error
    }

2.6 解压不带密码zip压缩包（源码案例）

package main

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

func main() {
	dst := "D:\\CODEFile01\\GoPro\\unZipPro\\src\\data_output"
	// 1、使用zip.OpenReader打开zip文件
	archive, err := zip.OpenReader("D:\\CODEFile01\\GoPro\\unZipPro\\src\\data\\DYSH_20210104_Index.zip")
	if err != nil {
		panic(err)
	}
	defer archive.Close()

	// 2、循环访问 zip 中的文件 zip.File 切片
	for _, f := range archive.File {
		filePath := filepath.Join(dst, f.Name)
		fmt.Println("unzipping file ", filePath)

		if !strings.HasPrefix(filePath, filepath.Clean(dst)+string(os.PathSeparator)) {
			fmt.Println("invalid file path")
			return
		}
		if f.FileInfo().IsDir() {
			fmt.Println("creating directory...")
			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)
		}

		// 3、使用 zip.File.Open 方法读取 zip 中文件的内容
		fileInArchive, err := f.Open()
		if err != nil {
			panic(err)
		}

		// 4、使用 io.Copy 或 io.Writer.Write 保存解压后的文件内容
		if _, err := io.Copy(dstFile, fileInArchive); err != nil {
			panic(err)
		}

		// 5、使用 zip.Reader.Close 关闭 zip 文件
		dstFile.Close()
		fileInArchive.Close()
	}
}

三、针对带密码的zip操作

解决方案：使用 github.com/yeka/zip

3.1 解压带密码zip压缩包

关键代码：

if f.IsEncrypted() {
	f.SetPassword(passwd)
}

其实就是将go官方zip包 archive/zip做一个替换即可。

3.2 解压带密码zip压缩包（源码案例）

package main

import (
	//"archive/zip"
	"fmt"
	"io"
	"os"
	"path/filepath"
	"strings"
	"github.com/yeka/zip"
)

func main() {
	dst := "D:\\CODEFile01\\GoPro\\unZipPro\\src\\data_output"
	passwd := "DataYes"
	// 1、使用zip.OpenReader打开zip文件
	archive, err := zip.OpenReader("D:\\CODEFile01\\GoPro\\unZipPro\\src\\data\\DYSH_20210104_Index.zip")
	if err != nil {
		panic(err)
	}
	defer archive.Close()

	// 2、循环访问 zip 中的文件 zip.File 切片
	for _, f := range archive.File {
		if f.IsEncrypted() {
			f.SetPassword(passwd)
		}
		filePath := filepath.Join(dst, f.Name)
		fmt.Println("unzipping file ", filePath)

		if !strings.HasPrefix(filePath, filepath.Clean(dst)+string(os.PathSeparator)) {
			fmt.Println("invalid file path")
			return
		}
		if f.FileInfo().IsDir() {
			fmt.Println("creating directory...")
			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)
		}

		// 3、使用 zip.File.Open 方法读取 zip 中文件的内容
		fileInArchive, err := f.Open()
		if err != nil {
			panic(err)
		}

		// 4、使用 io.Copy 或 io.Writer.Write 保存解压后的文件内容
		if _, err := io.Copy(dstFile, fileInArchive); err != nil {
			panic(err)
		}

		// 5、使用 zip.Reader.Close 关闭 zip 文件
		dstFile.Close()
		fileInArchive.Close()
	}
}

以上便是针对带密码的zip压缩包解压方法！！！