Go 文件压缩解压

在Go语言中,archive/zip包提供了创建、读取和解压缩ZIP格式文件的功能。

一、创建ZIP文件并添加内容----压缩

Go 复制代码
package main

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

func main() {
	// 创建一个缓冲区来存储ZIP文件的内容
	buf := new(bytes.Buffer)

	// 创建一个新的ZIP文件
	zipWriter := zip.NewWriter(buf)

	// 添加文件到ZIP文件中
	addFileToZip(zipWriter, "file1.txt", []byte("向file1写入内容"))
	addFileToZip(zipWriter, "file2.txt", []byte("向file2写入内容"))

	// 关闭ZIP文件
	err := zipWriter.Close()
	if err != nil {
		log.Fatal(err)
	}

	// 将ZIP文件内容写入磁盘
	file, err := os.Create("testFileZip.zip")
	if err != nil {
		log.Fatal(err)
	}
	defer file.Close()

	_, err = io.Copy(file, buf)
	if err != nil {
		log.Fatal(err)
	}

	fmt.Println("压缩文件创建成功!")
}

// 辅助函数,用于将字节数据作为文件添加到ZIP中
func addFileToZip(zipWriter *zip.Writer, fileName string, data []byte) error {
	header := &zip.FileHeader{
		Name:   fileName,
		Method: zip.Deflate,
	}

	writer, err := zipWriter.CreateHeader(header)
	if err != nil {
		return err
	}

	_, err = writer.Write(data)
	return err
}

二、解压缩ZIP文件----读取并解压缩一个ZIP文件

Go 复制代码
package main

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

func main() {
	//读取zip压缩文件
	reader, err := zip.OpenReader("testFileZip.zip")
	if err != nil {
		log.Fatal(err)
	}
	defer reader.Close()

	folder := "testFileZip"
	_, err = os.Stat(folder)              // os.Stat 函数来获取 folder 指定路径的状态信息。
	if err != nil && os.IsNotExist(err) { //判断是否存在
		err = os.Mkdir(folder, os.ModePerm) //不存在就创建,os.ModePerm是Go语言中定义的默认权限,通常用于创建文件或目录。
		if err != nil {
			log.Fatal(err)
		}
	}
	//提取压缩文件
	for _, file := range reader.File {
		fmt.Println("提取到的文件有:", file.Name)

		// 创建目标文件的路径
		targetPath := filepath.Join("testFileZip", file.Name)

		// 创建目标文件
		outputFile, err := os.Create(targetPath)
		if err != nil {
			log.Fatal(err)
		}
		defer outputFile.Close()

		// 读取并解压文件内容到目标文件
		fileReader, err := file.Open()
		if err != nil {
			log.Fatal(err)
		}
		defer fileReader.Close()

		_, err = io.Copy(outputFile, fileReader)
		if err != nil {
			log.Fatal(err)
		}
	}

	fmt.Println("已成功提取所有文件")
}
相关推荐
加油吧zkf6 小时前
Python入门:从零开始的完整学习指南
开发语言·前端·python
xqlily6 小时前
Kotlin:现代编程语言的革新者
android·开发语言·kotlin
csbysj20207 小时前
XSLT Apply:深入解析XSLT在XML转换中的应用
开发语言
不会kao代码的小王7 小时前
突破机房围墙:openEuler设备的公网管理实战指南
开发语言·数据库·笔记
CodeCraft Studio7 小时前
CAD文件处理控件Aspose.CAD教程:在 Python 中将 SVG 转换为 PDF
开发语言·python·pdf·svg·cad·aspose·aspose.cad
消失的旧时光-19437 小时前
人脸跟随 ( Channel 实现(缓存5条数据 + 2度过滤 + 平滑移动))
android·java·开发语言·kotlin
默默coding的程序猿7 小时前
1.单例模式有哪几种常见的实现方式?
java·开发语言·spring boot·spring·单例模式·设计模式·idea
MOON404☾8 小时前
Rust程序语言设计(5-8)
开发语言·后端·rust
Knight_AL9 小时前
浅拷贝与深拷贝详解:概念、代码示例与后端应用场景
android·java·开发语言
枫叶丹49 小时前
【Qt开发】输入类控件(六)-> QDial
开发语言·qt