使用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生成压缩文件的资料请关注脚本之家其它相关文章!

相关文章

  • Go语言中的自定义类型与类型别名详解

    Go语言中的自定义类型与类型别名详解

    文章介绍了Go语言中的自定义类型和类型别名,自定义类型使用type关键字创建,基于基本类型、结构体或接口,类型别名使用typealias=original语法,创建新名称引用原有类型,文章介绍的非常详细,感兴趣的朋友一起看看吧
    2026-04-04
  • go 代码的调试---打印调用堆栈的实例

    go 代码的调试---打印调用堆栈的实例

    下面小编就为大家带来一篇go 代码的调试---打印调用堆栈的实例。小编觉得挺不错的,现在就分享给大家,也给大家做个参考。一起跟随小编过来看看吧
    2017-10-10
  • Go语言中处理错误的技巧分享

    Go语言中处理错误的技巧分享

    编写 Go 语言程序时,有效地处理错误是至关重要的,Go 语言提供了一些强大的工具和模式来处理错误,本文将介绍这些方法,以便编写健壮的 Go 代码,更好地处理错误,需要的朋友可以参考下
    2023-09-09
  • Go语言学习之运算符使用详解

    Go语言学习之运算符使用详解

    这篇文章主要介绍了Go语言中常用运算符的使用,本文给大家介绍的非常详细,对大家的学习或工作具有一定的参考借鉴价值,需要的朋友可以参考下
    2022-04-04
  • golang反向代理设置host不生效的问题解决

    golang反向代理设置host不生效的问题解决

    在使用golang的httputil做反向代理的时候,发现一个奇怪的现象,上游网关必须要设置host才行,不设置host的话,golang服务反向代理请求下游会出现http 503错误,接下来通过本文给大家介绍golang反向代理设置host不生效问题,感兴趣的朋友一起看看吧
    2023-05-05
  • go语言数组及结构体继承和初始化示例解析

    go语言数组及结构体继承和初始化示例解析

    这篇文章主要为大家介绍了go语言数组及结构体继承和初始化示例解析,有需要的朋友可以借鉴参考下,希望能够有所帮助,祝大家多多进步早日升职加薪
    2022-04-04
  • Golang两行代码实现发送钉钉机器人消息

    Golang两行代码实现发送钉钉机器人消息

    创建一个钉钉机器人必须使用加签,本文通过Golang两行代码实现发送钉钉机器人消息,本文给大家介绍的非常详细,感兴趣的朋友跟随小编一起看看吧
    2021-12-12
  • Go语言设计模式之结构型模式

    Go语言设计模式之结构型模式

    本文主要聚焦在结构型模式(Structural Pattern)上,其主要思想是将多个对象组装成较大的结构,并同时保持结构的灵活和高效,从程序的结构上解决模块之间的耦合问题
    2021-06-06
  • golang的database.sql包和事务处理操作步骤

    golang的database.sql包和事务处理操作步骤

    事务是数据库操作中的一个逻辑单元,由一系列的数据库操作组成,这一系列操作要么全部执行并且提交,要么全部回滚,确保数据的一致性和完整性,这篇文章主要介绍了golang的database.sql包和事务处理操作步骤,需要的朋友可以参考下
    2025-04-04
  • Go实现字符串与数字的高效转换

    Go实现字符串与数字的高效转换

    在软件开发的世界里,数据类型转换是一项基础而重要的技能,尤其在Go语言这样类型严格的语言中,正确高效地进行类型转换对于性能优化和代码质量至关重要,本文给大家介绍了Go实现字符串与数字的高效转换,需要的朋友可以参考下
    2024-02-02

最新评论