go语言制作的zip压缩程序
程序员文章站
2022-05-15 08:21:08
可以压缩文件和目录。
package main
import (
"archive/zip"
"bytes"
"fmt"
"io/iou...
可以压缩文件和目录。
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 }
以上所述就是本文的全部内容了,希望大家能够喜欢。
上一篇: GOLANG版的冒泡排序和快速排序分享