archive/zip

压缩实现过程

创建压缩归档文件

osos.Create
func Create(name string) (*File, error)
O_RDWR*PathError

初始化归档文件zip.Writer

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 mainimport ("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.zipLength      Date    Time    Name
---------  ---------- -----   ----50  1980-00-00 00:00   csv/test.csv16  1980-00-00 00:00   txt/test.txt
---------                     -------66                     2 files