103 lines
2.1 KiB
Go
103 lines
2.1 KiB
Go
package util
|
|
|
|
import (
|
|
"archive/zip"
|
|
"github.com/eyebluecn/tank/code/tool/result"
|
|
"io"
|
|
"log"
|
|
"os"
|
|
"path/filepath"
|
|
"strings"
|
|
)
|
|
|
|
//将srcPath压缩到destPath。
|
|
func Zip(srcPath string, destPath string) error {
|
|
|
|
//统一处理\\成/
|
|
srcPath = UniformPath(srcPath)
|
|
|
|
if PathExists(destPath) {
|
|
panic(result.BadRequest("%s 已经存在了", destPath))
|
|
}
|
|
|
|
// 创建准备写入的文件
|
|
fileWriter, err := os.Create(destPath)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer func() {
|
|
err := fileWriter.Close()
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
}()
|
|
|
|
// 通过 fileWriter 来创建 zip.Write
|
|
zipWriter := zip.NewWriter(fileWriter)
|
|
defer func() {
|
|
// 检测一下是否成功关闭
|
|
if err := zipWriter.Close(); err != nil {
|
|
log.Fatalln(err)
|
|
}
|
|
}()
|
|
|
|
//上一个文件夹路径
|
|
baseDirPath := GetDirOfPath(srcPath) + "/"
|
|
// 下面来将文件写入 zipWriter ，因为有可能会有很多个目录及文件，所以递归处理
|
|
err = filepath.Walk(srcPath, func(path string, fileInfo os.FileInfo, errBack error) (err error) {
|
|
if errBack != nil {
|
|
return errBack
|
|
}
|
|
|
|
//统一处理\\成/
|
|
path = UniformPath(path)
|
|
|
|
// 通过文件信息，创建 zip 的文件信息
|
|
fileHeader, err := zip.FileInfoHeader(fileInfo)
|
|
if err != nil {
|
|
return
|
|
}
|
|
|
|
// 替换文件信息中的文件名
|
|
fileHeader.Name = strings.TrimPrefix(path, baseDirPath)
|
|
|
|
// 目录前要加上/
|
|
if fileInfo.IsDir() {
|
|
fileHeader.Name += "/"
|
|
}
|
|
|
|
// 写入文件信息，并返回一个 Write 结构
|
|
writer, err := zipWriter.CreateHeader(fileHeader)
|
|
if err != nil {
|
|
return
|
|
}
|
|
|
|
// 检测，如果不是标准文件就只写入头信息，不写入文件数据到 writer
|
|
// 如目录，也没有数据需要写
|
|
if !fileHeader.Mode().IsRegular() {
|
|
return nil
|
|
}
|
|
|
|
// 打开要压缩的文件
|
|
fileToBeZip, err := os.Open(path)
|
|
defer func() {
|
|
err = fileToBeZip.Close()
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
}()
|
|
if err != nil {
|
|
return
|
|
}
|
|
|
|
// 将打开的文件 Copy 到 writer
|
|
_, err = io.Copy(writer, fileToBeZip)
|
|
if err != nil {
|
|
return
|
|
}
|
|
|
|
return nil
|
|
})
|
|
return err
|
|
}
|