使用Golang生成压缩文件的详细教程
更新时间：2024年01月30日 10:48:33 作者：梦想画家
与其他语言一样,内置标准库默认支持文件压缩功能,本文介绍Golang如何创建压缩文件,增加一个或多个文件生成压缩文件,也可以在压缩文件中创建文件夹,用于对文件进行分类管理,Golang标准库 archive/zip提供了创建和读取压缩文件功能,需要的朋友可以参考下
这是一个简单的golang压缩文件小案例，可做很多的拓展，这里使用的库是archive/zip，在gopkg里面搜zip就行。
方法一
使用gin框架，获取前端传过来的文件压缩成zip，存储路径入数据库中，代码简单示例如下：
@Param file formData file true "上传文件"
func UploadToZip(c *gin.Context){
//获取前端传过来的文件，如果有多个文件: 使用form,_ := c.MultipartForm()
//files := form.File["file"]
//然后遍历files拿到每个文件
file,err := c.FormFile("file")
if err != nil {
response.Fail("参数有误",c)
return
}
//创建目录和指定文件名
err = c.SaveUploadedFile( file ,"files/" + file + file.Filename)
if err != nil {
response.Fail("存储文件失败",c)
return
}
//在指定的目录生成你要的zip文件
create, err := os.Create("flies/dome.zip")
defer create.Close()
if err != nil {
response.Fail("创建失败",c)
retuen
}
//创建一个zip流
writer := zip.NewWriter(create)
defer writer.Close()
//读取存储的目录，将里面的文件压缩
readFile ,err := os.ReadDir("files")
if err != nil {
response.Fail("读取文件失败",c)
retuen
}
//遍历目录，获取单个文件
for _,rf := range readFile {
//只压缩文件，目录不做压缩
if !rf.IsDir() {
//打开要压缩的文件
open,err := os.Open("flies" + rf.Name())
if err != nil {
response.Fail("打开文件失败",c)
retuen
}
//创建一个压缩包里面的文件和文件名，这样解压后会有一个files目录，目录里面有压缩的文件
f, _ := writer.Create("flies" + rf.Name())
//将文件压缩成zip
if _,err := io.Copy(f,open);err != nil {
response.Fail("压缩失败",c)
retuen
}
}
}
//将压缩路径存放数据库
...
response.OK("压缩成功",c)
}
方法二
压缩实现过程
创建压缩归档文件
首先需要创建归档文件，与其他普通文件一样。使用os包的os.Create函数：
func Create(name string) (*File, error)
该方法创建或删除给定名称文件。如果文件已存在则删除重新创建，如果不存在，创建文件，模式为0666。创建成功返回File用于读写，关联文件描述符为O_RDWR，反之报错，类型为*PathError。
初始化归档文件zip.Writer
使用 archive/zip 包 中的zip.NewWriter 函数，用于写数据（文件或目录）至最终的压缩文件。
语法如下：
func NewWriter(w io.Writer) *Writer
NewWriter 返回Writer写
使用zip.Writer.Create增加文件压缩文件
上面已创建了zip.Writer，可以增加文件和目录至压缩文件，使用zip.Writer.Create函数：
func (w *Writer) Create(name string) (io.Writer, error)
方法通过文件名称增加文件至压缩文件，返回Writer用于写文件内容，文件内容将被压缩。文件名称必须为相对路径，不能以驱动器字母(C:)或斜杠开头，仅允许正斜杠。如果增加目录，需要在名称后面增加尾斜杠。在下一次调用Create, CreateHeader, 或 Close方法之前，文件内容必须被写入io.Writer.
使用 io.Copy 或 io.Writer.Write 写文件内容
zip.Writer.Create返回io.Writer，用于写数据，所以任何文件内容可以流入或写入该Writer，也可以使用io.Copy函数。
func Copy(dst Writer, src Reader) (written int64, err error)
Copy函数从src至dst，直达EOF或遇到错误。返回拷贝的字节数以及错误（如果有错误发生）。
使用zip.Writer.Close关闭压缩文件
所有文件和目录写入至压缩文件后，需要通过zip.Writer.Close方法关闭写生成压缩文件，即写所有数据至底层数据流。
func (w *Writer) Close() error
实例代码
下面示例压缩两个文件（txt和csv文件）生成单个压缩文件。这两个文件分别在压缩文件的不同子目录中。
package main
import (
"archive/zip"
"fmt"
"io"
"os"
)
func main() {
fmt.Println("creating zip archive...")
archive, err := os.Create("archive.zip")
if err != nil {
panic(err)
}
defer archive.Close()
zipWriter := zip.NewWriter(archive)
fmt.Println("opening first file...")
f1, err := os.Open("test.csv")
if err != nil {
panic(err)
}
defer f1.Close()
fmt.Println("writing first file to archive...")
w1, err := zipWriter.Create("csv/test.csv")
if err != nil {
panic(err)
}
if _, err := io.Copy(w1, f1); err != nil {
panic(err)
}
fmt.Println("opening second file")
f2, err := os.Open("test.txt")
if err != nil {
panic(err)
}
defer f2.Close()
fmt.Println("writing second file to archive...")
w2, err := zipWriter.Create("txt/test.txt")
if err != nil {
panic(err)
}
if _, err := io.Copy(w2, f2); err != nil {
panic(err)
}
fmt.Println("closing zip archive...")
zipWriter.Close()
}
运行程序，生成日志：
creating zip archive... opening first file... writing first file to archive... opening second file writing second file to archive... closing zip archive...
最后解压压缩文件，和我们期望的一致：
$ unzip -l archive.zip
Archive: archive.zip
Length Date Time Name
--------- ---------- ----- ----
50 1980-00-00 00:00 csv/test.csv
16 1980-00-00 00:00 txt/test.txt
--------- -------
66 2 files
总结
以上就是使用Golang生成压缩文件的详细教程的详细内容，更多关于Golang生成压缩文件的资料请关注脚本之家其它相关文章！
相关文章
这篇文章主要介绍了golang 格式化输入输出操作，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2020-12-12
循环语句为了简化程序中有规律的重复性操作，需要用到循环语句，和其他大多数编程语言一样，GO的循环语句有for循环，不同的是没有while循环，而循环控制语句可以改变循环语句的执行过程，下面给大家介绍下go循环语句和循环控制语句的相关知识，一起看看吧2021-11-11
这篇文章主要介绍了Go Gin实现文件上传下载的示例代码，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2021-04-04
go语言 swagger 查询 json 字段注释的示例代码
在Go语言中,使用Swagger通过swag工具和gin-gonic框架生成API文档,涉及引入依赖、定义模型、添加注释等步骤,示例中展示了如何为接受查询参数的API端点添加注释,感兴趣的朋友跟随小编一起看看吧2024-09-09
这篇文章主要介绍了Golang使用CGO与Plugin技术运行加载C动态库，Golang 程序在运行时加载C动态库的技术，跳过了Golang项目编译阶段需要链接C动态库的过程，提高了Golang项目开发部署的灵活性2022-07-07
这篇文章主要介绍了MacOS下本地golang环境搭建详细教程,本文给大家介绍的非常详细，对大家的学习或工作具有一定的参考借鉴价值，需要的朋友可以参考下2020-09-09
这篇文章给大家介绍了Golang常用的几种密码加密方式,加密有两种方式,一种是直接加密,一种是盐值加密,直接加密指的是将原始密码直接进行加密,盐值加密则是在进行密码加密之前,文中有详细的代码示例,需要的朋友可以参考下2023-08-08
在 Go 语言中，有一种特殊的用法可能让许多人感到困惑，那就是空结构体，本文将对Go空结构体进行详解，准备一杯你最喜欢的饮料或茶，随着本文一探究竟吧2023-05-05
性能优化的前提是满足正确可靠、简洁清晰等质量因素，针对 Go语言特性，本文为大家整理了一些Go语言相关的性能优化建议，感兴趣的可以了解一下2023-07-07
go语言 xorm框架 postgresql 的用法及详细注解
这篇文章主要介绍了go语言 xorm框架 postgresql 的用法及详细注解，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2020-12-12
最新评论