這篇文章主要介紹了go語言制作的zip壓縮程序,其主體思路是首先創建一個讀寫緩沖,然后用壓縮器包裝該緩沖,用Walk方法來將所有目錄下的文件寫入zip,有需要的小伙伴參考下。
可以壓縮文件和目錄。
- package main
- import (
- "archive/zip"
- "bytes"
- "fmt"
- "io/ioutil"
- "os"
- "path/filepath"
- )
- func main() {
- if err := compress(`gopkg`, `gopkg.zip`); err != nil {
- fmt.Println(err)
- }
- }
- // 參數frm可以是文件或目錄,不會給dst添加.zip擴展名
- func compress(frm, dst string) error {
- buf := bytes.NewBuffer(make([]byte, 0, 10*1024*1024)) // 創建一個讀寫緩沖
- myzip := zip.NewWriter(buf) // 用壓縮器包裝該緩沖
- // 用Walk方法來將所有目錄下的文件寫入zip
- err := filepath.Walk(frm, func(path string, info os.FileInfo, err error) error {
- var file []byte
- if err != nil {
- return filepath.SkipDir
- }
- header, err := zip.FileInfoHeader(info) // 轉換為zip格式的文件信息
- if err != nil {
- return filepath.SkipDir
- }
- header.Name, _ = filepath.Rel(filepath.Dir(frm), path)
- if !info.IsDir() {
- // 確定采用的壓縮算法(這個是內建注冊的deflate)
- header.Method = 8
- file, err = ioutil.ReadFile(path) // 獲取文件內容
- if err != nil {
- return filepath.SkipDir
- }
- } else {
- file = nil
- }
- // 上面的部分如果出錯都返回filepath.SkipDir
- // 下面的部分如果出錯都直接返回該錯誤
- // 目的是盡可能的壓縮目錄下的文件,同時保證zip文件格式正確
- w, err := myzip.CreateHeader(header) // 創建一條記錄并寫入文件信息
- if err != nil {
- return err
- }
- _, err = w.Write(file) // 非目錄文件會寫入數據,目錄不會寫入數據
- if err != nil { // 因為目錄的內容可能會修改
- return err // 最關鍵的是我不知道咋獲得目錄文件的內容
- }
- return nil
- })
- if err != nil {
- return err
- }
- myzip.Close() // 關閉壓縮器,讓壓縮器緩沖中的數據寫入buf
- file, err := os.Create(dst) // 建立zip文件
- if err != nil {
- return err
- }
- defer file.Close()
- _, err = buf.WriteTo(file) // 將buf中的數據寫入文件
- if err != nil {
- return err
- }
- return nil
- }
以上所述就是本文的全部內容了,希望大家能夠喜歡。
新聞熱點
疑難解答