答案：本文介绍Go语言中使用archive/zip包实现文件压缩与解压的方法，包含compressFiles和decompressZip两个函数示例，分别演示将多个文件压缩为zip及解压到指定目录的流程，并强调路径安全、流式处理、权限保留和错误处理等注意事项。
在Go语言开发中，处理文件压缩与解压是常见的需求，比如日志归档、数据导出、网络传输优化等场景。Golang标准库提供了archive/zip和compress/gzip等包，可以方便地实现压缩与解压功能。本文通过实际示例，带你快速掌握如何使用Golang进行文件的zip压缩与解压。
使用 zip 实现文件压缩
要将一个或多个文件打包成 zip 格式，可以使用 archive/zip 包。核心思路是创建一个 zip.Writer，然后逐个将文件写入归档中。
以下是一个简单的压缩函数示例：
func compressFiles(filename string, files []string) error {
outFile, err := os.Create(filename)
if err != nil {
return err
}
defer outFile.Close()
zipWriter := zip.NewWriter(outFile)
defer zipWriter.Close()
for _, filePath := range files {
if err := addFileToZip(zipWriter, filePath); 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
}
// 使用Deflate压缩算法，可减小体积
header.Method = zip.Deflate
writer, err := zipWriter.CreateHeader(header)
if err != nil {
return err
}
_, err = io.Copy(writer, file)
return err
}
调用方式：
立即学习“go语言免费学习笔记（深入）”；
compressFiles("output.zip", []string{"file1.txt", "file2.txt"})
实现 zip 文件解压
解压 zip 文件需要读取归档中的每个文件，并将其内容写入磁盘。使用 zip.Reader 遍历文件列表，逐个提取。
以下是一个基础的解压函数：
func decompressZip(zipPath, targetDir string) error {
reader, err := zip.OpenReader(zipPath)
if err != nil {
return err
}
defer reader.Close()
for _, file := range reader.File {
filePath := filepath.Join(targetDir, file.Name)
if file.FileInfo().IsDir() {
os.MkdirAll(filePath, os.ModePerm)
continue
}
if err := os.MkdirAll(filepath.Dir(filePath), os.ModePerm); err != nil {
return err
}
outFile, err := os.Create(filePath)
if err != nil {
return err
}
rc, err := file.Open()
if err != nil {
outFile.Close()
return err
}
_, err = io.Copy(outFile, rc)
rc.Close()
outFile.Close()
if err != nil {
return err
}
}
return nil
}
使用方法：
decompressZip("output.zip", "./extracted/")
注意事项与优化建议
在实际项目中，需要注意以下几点以提升稳定性和兼容性：
-
路径安全：解压时需防止路径穿越攻击（如文件名为
../../../etc/passwd），应校验文件路径是否在目标目录内。 - 大文件处理：避免一次性加载整个文件到内存，始终使用流式读写。
-
权限保留：Linux下可尝试恢复文件权限（通过
file.Mode()设置）。 - 错误处理：对每个IO操作做错误检查，确保资源及时释放。
基本上就这些。Golang通过标准库就能完成实用的压缩解压功能，无需引入第三方依赖，适合嵌入到服务端工具或CLI程序中。掌握这些基础操作后，你可以进一步封装为通用模块，支持目录递归压缩、进度反馈等功能。
