Go语言中利用archive/zip包实现字节数据压缩教程

package main

import (
    "archive/zip"
    "bytes"
    "fmt"
    "log"
    "os"
)

// ZipFileEntry 结构体定义了要添加到Zip归档中的文件信息
type ZipFileEntry struct {
    Name string // 文件在Zip归档中的名称
    Body []byte // 文件的内容（字节数组）
}

// ZipBytesToArchive 将一组字节数据压缩并写入到指定的Zip文件路径
// zipFilePath: 目标Zip文件的路径
// files: 包含要压缩的每个文件信息的切片
func ZipBytesToArchive(zipFilePath string, files []ZipFileEntry) error {
    // 1. 创建一个缓冲区来存储Zip归档的字节数据
    buf := new(bytes.Buffer)

    // 2. 创建一个新的Zip写入器，它会将数据写入到buf中
    zipWriter := zip.NewWriter(buf)

    // 3. 遍历要添加到归档中的文件
    for _, file := range files {
        // 3.1 在Zip归档中创建一个新的文件条目
        // zip.Create会返回一个io.Writer，我们可以向其中写入文件内容
        zipFileEntryWriter, err := zipWriter.Create(file.Name)
        if err != nil {
            return fmt.Errorf("创建Zip文件条目 '%s' 失败: %w", file.Name, err)
        }

        // 3.2 将文件内容写入到Zip文件条目中
        _, err = zipFileEntryWriter.Write(file.Body)
        if err != nil {
            return fmt.Errorf("写入文件内容 '%s' 失败: %w", file.Name, err)
        }
    }

    // 4. 关闭Zip写入器。这一步非常重要，它会完成Zip归档的最终写入和元数据更新。
    // 务必检查此处的错误，因为Zip文件损坏的常见原因就是未正确关闭。
    err := zipWriter.Close()
    if err != nil {
        return fmt.Errorf("关闭Zip写入器失败: %w", err)
    }

    // 5. 将包含Zip归档数据的缓冲区内容写入到物理文件
    // os.WriteFile是Go 1.16+推荐的替代ioutil.WriteFile的方法
    // 0644表示文件所有者可读写，其他人只读
    err = os.WriteFile(zipFilePath, buf.Bytes(), 0644)
    if err != nil {
        return fmt.Errorf("将Zip数据写入文件 '%s' 失败: %w", zipFilePath, err)
    }

    return nil // 成功完成
}

func main() {
    fmt.Println("开始执行Zip压缩示例...")

    // 定义要压缩的文件数据
    filesToZip := []ZipFileEntry{
        {"readme.txt", []byte("这是一个包含文本文件的Zip归档。\n欢迎使用Go语言进行数据压缩。")},
        {"gopher.txt", []byte("Gopher名字:\n乔治\n杰弗里\n冈萨洛\n格洛丽亚")},
        {"todo.txt", []byte("1. 获取动物处理许可证。\n2. 编写更多示例代码。\n3. 学习更多Go语言特性。")},
        {"binary_data.bin", []byte{0xDE, 0xAD, 0xBE, 0xEF, 0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07, 0x08}}, // 示例二进制数据
    }

    zipFileName := "example_archive.zip"
    err := ZipBytesToArchive(zipFileName, filesToZip)
    if err != nil {
        log.Fatalf("Zip压缩失败: %v", err) // 使用log.Fatalf在发生错误时终止程序
    }

    fmt.Printf("Zip文件 '%s' 已成功创建。\n", zipFileName)
}