go语言制作的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
}

以上所述就是本文的全部内容了，希望大家能够喜欢。