golang中archive/zip包

简介： golang使用archive/zip包、golang进行zip操作

一、ZIP的操作

压缩、解压

二、GO中压缩操作

1、创建zip文件
2、向这个zip文件里写入要压缩的文件的头部信息和文件信息

package main

import (
    "os"
    "log"
    "archive/zip"
    "io"
    "fmt"
)

func main() {
    err := Zip("demo.zip", []string{"static/downloads/test-data.xlsx"})
    if err != nil {
        fmt.Println(err)
    }
}

func Zip(zipFile string, fileList []string) error {
    // 创建 zip 包文件
    fw, err := os.Create(zipFile)
    if err != nil {
        log.Fatal()
    }
    defer fw.Close()

    // 实例化新的 zip.Writer
    zw := zip.NewWriter(fw)
    defer func() {
        // 检测一下是否成功关闭
        if err := zw.Close(); err != nil {
            log.Fatalln(err)
        }
    }()

    for _, fileName := range fileList {
        fr, err := os.Open(fileName)
        if err != nil {
            return err
        }
        fi, err := fr.Stat()
        if err != nil {
            return err
        }
        defer fr.Close()

        // 写入文件的头信息
        fh, err := zip.FileInfoHeader(fi)
        w, err := zw.CreateHeader(fh)
        if err != nil {
            return err
        }
        // 写入文件内容
        _, err = io.Copy(w, fr)
        if err != nil {
            return err
        }
    }
    return nil
}

三、GO中解缩操作
1、遍历读取zip内压缩文件信息
2、根据zip内压缩文件的数据创建文件，并写入对应数据

package main

import (
    "archive/zip"
    "os"
    "io"
    "fmt"
)

func main() {
    err := Unzip("demo.zip")
    if err != nil {
        fmt.Println(err)
    }
}

func Unzip(zipFile string) error {
    zr, err := zip.OpenReader(zipFile)
    defer zr.Close()
    if err != nil {
        return err
    }

    for _, file := range zr.File {
        // 如果是目录，则创建目录
        if file.FileInfo().IsDir() {
            if err = os.MkdirAll(file.Name, file.Mode());err != nil {
                return err
            }
            continue
        }
        // 获取到 Reader
        fr, err := file.Open()
        if err != nil {
            return err
        }

        fw, err := os.OpenFile(file.Name, os.O_CREATE|os.O_RDWR|os.O_TRUNC, file.Mode())
        if err != nil {
            return err
        }
        _, err = io.Copy(fw, fr)
        if err != nil {
            return err
        }
        fw.Close()
        fr.Close()
    }
    return nil
}