golang的zip压缩加密多文件完整例子
package main
import (
"bytes"
"fmt"
"io"
"os"
"path/filepath"
"github.com/mzky/zip"
)
func main() {
var array []string
array = append(array, "/mnt/k/v")
array = append(array, "/mnt/a/b")
err := Zip("/mnt/demo.zip", "1", array)
if err != nil {
fmt.Println(err)
}
// err = UnZip("/mnt/demo.zip", "1", "./") //解压相对路径
err = UnZip("/mnt/demo.zip", "1", "/")
if err != nil {
fmt.Println(err)
}
}
func IsZip(zipPath string) bool {
f, err := os.Open(zipPath)
if err != nil {
return false
}
defer f.Close()
buf := make([]byte, 4)
if n, err := f.Read(buf); err != nil || n < 4 {
return false
}
return bytes.Equal(buf, []byte("PK\x03\x04"))
}
// password值可以为空""
func Zip(zipPath, password string, fileList []string) error {
if len(fileList) < 1 {
return fmt.Errorf("将要压缩的文件列表不能为空")
}
fz, err := os.Create(zipPath)
if err != nil {
return err
}
zw := zip.NewWriter(fz)
defer zw.Close()
for _, fileName := range fileList {
fr, err := os.Open(fileName)
if err != nil {
return err
}
// 写入文件的头信息
var w io.Writer
if password != "" {
w, err = zw.Encrypt(fileName, password, zip.AES256Encryption)
} else {
w, err = zw.Create(fileName)
}
if err != nil {
return err
}
// 写入文件内容
_, err = io.Copy(w, fr)
if err != nil {
return err
}
}
return zw.Flush()
}
// password值可以为空""
// 当decompressPath值为"./"时，解压到相对路径
func UnZip(zipPath, password, decompressPath string) error {
if !IsZip(zipPath) {
return fmt.Errorf("压缩文件格式不正确或已损坏")
}
r, err := zip.OpenReader(zipPath)
if err != nil {
return err
}
defer r.Close()
for _, f := range r.File {
if password != "" {
if f.IsEncrypted() {
f.SetPassword(password)
} else {
return errors.New("must be encrypted")
}
}
fp := filepath.Join(decompressPath, f.Name)
dir, _ := filepath.Split(fp)
err = os.MkdirAll(dir, os.ModePerm)
if err != nil {
return err
}
w, err := os.Create(fp)
if nil != err {
return err
}
fr, err := f.Open()
if err != nil {
return err
}
_, err = io.Copy(w, fr)
if err != nil {
return err
}
w.Close()
}
return nil
}