114 lines
2.8 KiB
Go
114 lines
2.8 KiB
Go
package zip
|
||
|
||
import (
|
||
"archive/zip"
|
||
"fmt"
|
||
"io"
|
||
"os"
|
||
"path"
|
||
)
|
||
|
||
func ZipFiles(filename string, files []string) error {
|
||
fmt.Println("start zip file......")
|
||
//创建输出文件目录
|
||
newZipFile, err := os.Create(filename)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
defer newZipFile.Close()
|
||
//创建空的zip档案，可以理解为打开zip文件，准备写入
|
||
zipWriter := zip.NewWriter(newZipFile)
|
||
defer zipWriter.Close()
|
||
// Add files to zip
|
||
for _, file := range files {
|
||
if err = AddFileToZip(zipWriter, file); err != nil {
|
||
return err
|
||
}
|
||
}
|
||
return nil
|
||
}
|
||
|
||
func AddFileToZip(zipWriter *zip.Writer, filename string) error {
|
||
//打开要压缩的文件
|
||
fileToZip, err := os.Open(filename)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
defer fileToZip.Close()
|
||
//获取文件的描述
|
||
info, err := fileToZip.Stat()
|
||
if err != nil {
|
||
return err
|
||
}
|
||
//FileInfoHeader返回一个根据fi填写了部分字段的Header，可以理解成是将fileinfo转换成zip格式的文件信息
|
||
header, err := zip.FileInfoHeader(info)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
header.Name = filename
|
||
/*
|
||
预定义压缩算法。
|
||
archive/zip包中预定义的有两种压缩方式。一个是仅把文件写入到zip中。不做压缩。一种是压缩文件然后写入到zip中。默认的Store模式。就是只保存不压缩的模式。
|
||
Store unit16 = 0 //仅存储文件
|
||
Deflate unit16 = 8 //压缩文件
|
||
*/
|
||
header.Method = zip.Store
|
||
//创建压缩包头部信息
|
||
writer, err := zipWriter.CreateHeader(header)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
//将源复制到目标，将fileToZip 写入writer 是按默认的缓冲区32k循环操作的，不会将内容一次性全写入内存中,这样就能解决大文件的问题
|
||
_, err = io.Copy(writer, fileToZip)
|
||
return err
|
||
}
|
||
|
||
// Decompressor 解压
|
||
func Decompressor(zipFilePath string, targetDir string, filename string) error {
|
||
reader, err := zip.OpenReader(zipFilePath)
|
||
if nil != err {
|
||
fmt.Println(err)
|
||
return err
|
||
}
|
||
defer reader.Close()
|
||
|
||
_ = os.MkdirAll(targetDir, 0777)
|
||
names := []string{}
|
||
for _, f := range reader.File {
|
||
err := func() error {
|
||
if f.FileInfo().IsDir() {
|
||
_ = os.MkdirAll(path.Join(targetDir, f.Name), f.Mode())
|
||
return nil
|
||
}
|
||
suffix := path.Ext(f.Name)
|
||
//fmt.Println(f.Name)
|
||
//fmt.Println(path.Join(targetDir, f.Name))
|
||
writeFile, err := os.OpenFile(path.Join(targetDir, filename+suffix), os.O_WRONLY|os.O_CREATE, f.Mode())
|
||
if nil != err {
|
||
return err
|
||
}
|
||
defer writeFile.Close()
|
||
|
||
readFile, err := f.Open()
|
||
if nil != err {
|
||
return err
|
||
}
|
||
defer readFile.Close()
|
||
|
||
n, err := io.Copy(writeFile, readFile)
|
||
if nil != err {
|
||
return err
|
||
}
|
||
if false {
|
||
names = append(names, f.Name)
|
||
fmt.Printf("解压文件: %s 大小: %v", f.Name, n)
|
||
}
|
||
return nil
|
||
}()
|
||
if nil != err {
|
||
return err
|
||
}
|
||
}
|
||
return nil
|
||
}
|