Golang 中的 archive/zip 包用于处理 ZIP 格式的压缩文件,提供了一系列用于创建、读取和解压缩 ZIP 格式文件的函数和类型,使用起来非常方便。
实现压缩功能
1、首先需要创建一个 zip 文件。
zip 文件也是一个文件,首先需要创建一个基础的 zip 文件,使用 os.Create() 函数创建即可。
package mainimport "os"func main() {//第一步,创建 zip 文件zipFile, err := os.Create("archive.zip")if err != nil {panic(err)}defer zipFile.Close()
}
2、使用 zip.NewWriter() 函数创建一个新的 *Writer 对象,用于将待压缩的文件写入到 zip 文件。
package mainimport ("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 mainimport ("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 文件中添加一个文件,返回一个待压缩的文件内容应写入的 Writerw, 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 mainimport ("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 文件中添加一个文件,返回一个待压缩的文件内容应写入的 Writerw, 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 mainimport ("archive/zip"
)func main() {// 第一步,打开 zip 文件zipFile, err := zip.OpenReader("archive.zip")if err != nil {panic(err)}defer zipFile.Close()
}
2、 遍历 zip 中的文件,将压缩包的中的文件读取到本地文件
package mainimport ("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.Nameif 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()}
}