使用Golang生成壓縮文件的詳細教程

更新時間：2024年01月30日 10:48:33 作者：夢想畫家

與其他語言一樣,內(nèi)置標準庫默認支持文件壓縮功能,本文介紹Golang如何創(chuàng)建壓縮文件,增加一個或多個文件生成壓縮文件,也可以在壓縮文件中創(chuàng)建文件夾,用于對文件進行分類管理,Golang標準庫 archive/zip提供了創(chuàng)建和讀取壓縮文件功能,需要的朋友可以參考下

方法一

使用gin框架，獲取前端傳過來的文件壓縮成zip，存儲路徑入數(shù)據(jù)庫中，代碼簡單示例如下：

@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("參數(shù)有誤",c)
		return
	}
	//創(chuàng)建目錄和指定文件名
	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("創(chuàng)建失敗",c)
		retuen
	}
	//創(chuàng)建一個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
			}
			//創(chuàng)建一個壓縮包里面的文件和文件名，這樣解壓后會有一個files目錄，目錄里面有壓縮的文件
			f, _ := writer.Create("flies" + rf.Name())
			//將文件壓縮成zip
			if _,err := io.Copy(f,open);err != nil {
				response.Fail("壓縮失敗",c)
				retuen
			}
		}
	}
	//將壓縮路徑存放數(shù)據(jù)庫
	...
	response.OK("壓縮成功",c)
}

方法二

壓縮實現(xiàn)過程

創(chuàng)建壓縮歸檔文件

首先需要創(chuàng)建歸檔文件，與其他普通文件一樣。使用os包的os.Create函數(shù)：

func Create(name string) (*File, error)

該方法創(chuàng)建或刪除給定名稱文件。如果文件已存在則刪除重新創(chuàng)建，如果不存在，創(chuàng)建文件，模式為0666。創(chuàng)建成功返回File用于讀寫，關(guān)聯(lián)文件描述符為O_RDWR，反之報錯，類型為*PathError。

初始化歸檔文件zip.Writer

使用 archive/zip 包中的zip.NewWriter 函數(shù)，用于寫數(shù)據(jù)（文件或目錄）至最終的壓縮文件。
語法如下：

func NewWriter(w io.Writer) *Writer

NewWriter 返回Writer寫

使用zip.Writer.Create增加文件壓縮文件

上面已創(chuàng)建了zip.Writer，可以增加文件和目錄至壓縮文件，使用zip.Writer.Create函數(shù)：

func (w *Writer) Create(name string) (io.Writer, error)

方法通過文件名稱增加文件至壓縮文件，返回Writer用于寫文件內(nèi)容，文件內(nèi)容將被壓縮。文件名稱必須為相對路徑，不能以驅(qū)動器字母(C:)或斜杠開頭，僅允許正斜杠。如果增加目錄，需要在名稱后面增加尾斜杠。在下一次調(diào)用Create, CreateHeader, 或 Close方法之前，文件內(nèi)容必須被寫入io.Writer.

使用 io.Copy 或 io.Writer.Write 寫文件內(nèi)容

zip.Writer.Create返回io.Writer，用于寫數(shù)據(jù)，所以任何文件內(nèi)容可以流入或?qū)懭朐揥riter，也可以使用io.Copy函數(shù)。

func Copy(dst Writer, src Reader) (written int64, err error)

Copy函數(shù)從src至dst，直達EOF或遇到錯誤。返回拷貝的字節(jié)數(shù)以及錯誤（如果有錯誤發(fā)生）。

使用zip.Writer.Close關(guān)閉壓縮文件

所有文件和目錄寫入至壓縮文件后，需要通過zip.Writer.Close方法關(guān)閉寫生成壓縮文件，即寫所有數(shù)據(jù)至底層數(shù)據(jù)流。

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