golang中archive/zip包

一、ZIP的操作

壓縮、解壓

二、GO中壓縮操作

1、建立zip檔案

2、向這個zip檔案裡寫入要壓縮的檔案的頭部資訊和檔案資訊

package main

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

func main() {
    err := Zip("demo.zip", []string{})
    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
        }
        // 寫入檔案的頭資訊
        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
}           

參考連結：

https://broqiang.com/posts/46#top