Golang

关注公众号 jb51net

关闭
首页 > 脚本专栏 > Golang > Go ZIP压缩与解压

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()
	}
}

到此这篇关于Golang使用archive/zip包实现ZIP压缩与解压的文章就介绍到这了,更多相关Go ZIP压缩与解压内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家!

您可能感兴趣的文章:
阅读全文