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)
  }
}
// 參數(shù)frm可以是文件或目錄，不會給dst添加.zip擴展名
func compress(frm, dst string) error {
  buf := bytes.NewBuffer(make([]byte, 0, 10*1024*1024)) // 創(chuàng)建一個讀寫緩沖
  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) // 轉(zhuǎn)換為zip格式的文件信息
    if err != nil {
      return filepath.SkipDir
    }
    header.Name, _ = filepath.Rel(filepath.Dir(frm), path)
    if !info.IsDir() {
      // 確定采用的壓縮算法（這個是內(nèi)建注冊的deflate）
      header.Method = 8
      file, err = ioutil.ReadFile(path) // 獲取文件內(nèi)容
      if err != nil {
        return filepath.SkipDir
      }
    } else {
      file = nil
    }
    // 上面的部分如果出錯都返回filepath.SkipDir
    // 下面的部分如果出錯都直接返回該錯誤
    // 目的是盡可能的壓縮目錄下的文件，同時保證zip文件格式正確
    w, err := myzip.CreateHeader(header) // 創(chuàng)建一條記錄并寫入文件信息
    if err != nil {
      return err
    }
    _, err = w.Write(file) // 非目錄文件會寫入數(shù)據(jù)，目錄不會寫入數(shù)據(jù)
    if err != nil {    // 因為目錄的內(nèi)容可能會修改
      return err     // 最關(guān)鍵的是我不知道咋獲得目錄文件的內(nèi)容
    }
    return nil
  })
  if err != nil {
    return err
  }
  myzip.Close()        // 關(guān)閉壓縮器，讓壓縮器緩沖中的數(shù)據(jù)寫入buf
  file, err := os.Create(dst) // 建立zip文件
  if err != nil {
    return err
  }
  defer file.Close()
  _, err = buf.WriteTo(file) // 將buf中的數(shù)據(jù)寫入文件
  if err != nil {
    return err
  }
  return nil
}