Go语言压缩实战（使用compress/zip包向ZIP文件中添加文件）

准备工作

首先，确保你已经安装了 Go 环境（建议 1.16 或更高版本）。然后创建一个新的项目目录，并初始化一个 Go 模块：

mkdir zip-democd zip-demogo mod init zip-demo

基本原理

要向 ZIP 文件中添加文件，我们需要：

1. 创建或打开一个 ZIP 文件用于写入（使用 os.Create）
2. 使用 zip.NewWriter 创建一个 ZIP 写入器
3. 调用 Create 方法创建 ZIP 中的新文件条目
4. 将源文件的内容写入该条目
5. 关闭所有资源

完整示例代码

下面是一个完整的 Go 程序，演示如何将本地文件添加到新的 ZIP 文件中：

package mainimport (	"archive/zip"	"fmt"	"io"	"os")// 将单个文件添加到 ZIP 中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	}	// 在 ZIP 中创建同名文件	header, err := zip.FileInfoHeader(info)	if err != nil {		return err	}	writer, err := zipWriter.CreateHeader(header)	if err != nil {		return err	}	// 将文件内容复制到 ZIP 条目中	_, err = io.Copy(writer, file)	return err}func main() {	// 创建新的 ZIP 文件	zipFile, err := os.Create("output.zip")	if err != nil {		fmt.Println("创建 ZIP 文件失败:", err)		return	}	defer zipFile.Close()	// 创建 ZIP 写入器	zipWriter := zip.NewWriter(zipFile)	defer zipWriter.Close()	// 假设当前目录下有 test.txt 文件	// 你可以替换成任意你想添加的文件路径	err = addFileToZip(zipWriter, "test.txt")	if err != nil {		fmt.Println("添加文件失败:", err)		return	}	fmt.Println("文件已成功添加到 output.zip！")}

如何测试这个程序？

1. 在项目根目录下创建一个 test.txt 文件，随便写点内容。

2. 运行程序：

go run main.go

3. 程序执行后，你会在目录中看到一个 output.zip 文件，解压后就能看到你添加的 test.txt。

进阶：批量添加多个文件

如果你需要添加多个文件，可以简单地循环调用 addFileToZip 函数：

files := []string{"file1.txt", "file2.log", "image.png"}for _, filename := range files {	err := addFileToZip(zipWriter, filename)	if err != nil {		fmt.Printf("添加 %s 失败: %v\n", filename, err)		// 可选择继续或退出	}}

常见问题与注意事项

• 文件路径：ZIP 中的文件路径由你传给 Create 或 CreateHeader 的名称决定。如果你想在 ZIP 中保留目录结构，可以在文件名前加上路径，例如 "docs/readme.txt"。
• 重复添加：如果多次添加同名文件，后面的会覆盖前面的。
• 大文件处理：对于非常大的文件，io.Copy 是流式处理，不会一次性加载全部内容到内存，所以很安全。
• 错误处理：务必检查每一步的错误，避免 ZIP 文件损坏。

总结

通过本文，你已经学会了如何使用 Go 语言的 compress/zip包 向 ZIP 文件中添加文件。这项技能在生成日志归档、打包资源、实现备份功能等场景中非常实用。掌握 Go语言 的 ZIP 操作，能让你的程序更加强大和灵活。

记住关键步骤：创建 ZIP 文件 → 创建写入器 → 创建条目 → 写入内容 → 关闭资源。多加练习，你很快就能熟练运用 ZIP文件操作 技术！

关键词回顾：Go语言、compress/zip包、ZIP文件操作、添加文件到ZIP