使用Golang生成压缩文件的详细教程
这是一个简单的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生成压缩文件的资料请关注脚本之家其它相关文章!
最新评论