加入收藏 | 设为首页 | 会员中心 | 我要投稿 李大同 (https://www.lidatong.com.cn/)- 科技、建站、经验、云计算、5G、大数据,站长网!
当前位置: 首页 > 大数据 > 正文

go语言制作的zip压缩程序

发布时间:2020-12-16 19:32:35 所属栏目:大数据 来源:网络整理
导读:可以压缩文件和目录。 package mainimport ( "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扩展

可以压缩文件和目录。

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,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
}

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

(编辑:李大同)

【声明】本站内容均来自网络,其相关言论仅代表作者个人观点,不代表本站立场。若无意侵犯到您的权利,请及时与联系站长删除相关内容!

    推荐文章
      热点阅读