Golang的archive/zip库如何压缩解压文件 演示多文件打包与读取

import (
    "archive/zip"
    "io"
    "os"
    "path/filepath"
    "fmt"
)

func compress(zipFile string, files []string) error {
    newZipFile, err := os.Create(zipFile)
    if err != nil {
        return err
    }
    defer newZipFile.Close()

    zipWriter := zip.NewWriter(newZipFile)
    defer zipWriter.Close()

    // 遍历要压缩的文件
    for _, file := range files {
        if err = addFileToZip(zipWriter, file); err != nil {
            return err
        }
    }
    return nil
}

func addFileToZip(zipWriter *zip.Writer, filename string) error {

    file, err := os.Open(filename)
    if err != nil {
        return err
    }
    defer file.Close()

    // 获取文件信息
    info, err := file.Stat()
    if err != nil {
        return err
    }

    header, err := zip.FileInfoHeader(info)
    if err != nil {
        return err
    }

    // 使用相对路径作为zip内的文件名
    header.Name = filename

    // 如果是目录，则需要设置权限
    if info.IsDir() {
        header.Name += "/"
    } else {
        header.Method = zip.Deflate
    }

    writer, err := zipWriter.CreateHeader(header)
    if err != nil {
        return err
    }

    if info.IsDir() {
        return nil
    }

    _, err = io.Copy(writer, file)
    return err
}

func main() {
    files := []string{"file1.txt", "file2.txt", "dir1"} // 假设已经存在这些文件和目录
    zipFile := "archive.zip"

    // 创建示例文件和目录
    os.WriteFile("file1.txt", []byte("This is file1."), 0644)
    os.WriteFile("file2.txt", []byte("This is file2."), 0644)
    os.Mkdir("dir1", 0755)
    os.WriteFile("dir1/file3.txt", []byte("This is file3 in dir1."), 0644)


    err := compress(zipFile, files)
    if err != nil {
        fmt.Println("压缩失败:", err)
        return
    }
    fmt.Println("压缩成功!")
}

func decompress(zipFile string, destDir string) error {
    reader, err := zip.OpenReader(zipFile)
    if err != nil {
        return err
    }
    defer reader.Close()

    for _, file := range reader.File {
        if err = decompressFile(file, destDir); err != nil {
            return err
        }
    }
    return nil
}

func decompressFile(file *zip.File, destDir string) error {
    filePath := filepath.Join(destDir, file.Name)

    if file.FileInfo().IsDir() {
        os.MkdirAll(filePath, file.Mode())
        return nil
    }

    // 创建目录
    if err := os.MkdirAll(filepath.Dir(filePath), 0755); err != nil {
        return err
    }

    // 打开文件
    reader, err := file.Open()
    if err != nil {
        return err
    }
    defer reader.Close()

    writer, err := os.Create(filePath)
    if err != nil {
        return err
    }
    defer writer.Close()

    _, err = io.Copy(writer, reader)
    return err
}

func main() {
    zipFile := "archive.zip"
    destDir := "extracted"

    // 创建解压目录
    os.Mkdir(destDir, 0755)

    err := decompress(zipFile, destDir)
    if err != nil {
        fmt.Println("解压缩失败:", err)
        return
    }
    fmt.Println("解压缩成功!")
}