答案：使用archive/zip包可实现文件压缩与解压。压缩时遍历文件路径，写入ZIP并保持目录结构；解压时逐项还原文件，注意路径安全与权限处理。
在Golang项目中处理文件压缩与解压，常用的是archive/zip包来操作ZIP格式。下面介绍如何实现文件的压缩和解压功能，适合实际项目使用。
压缩多个文件为ZIP
要将多个文件或目录打包成一个ZIP文件，可以使用archive/zip配合os和io包进行写入操作。
基本思路是创建一个ZIP输出文件，然后遍历需要压缩的路径，逐个将文件写入ZIP归档中，并保持目录结构。
注意：写入时需正确设置文件头信息，尤其是文件权限和路径分隔符（应使用/）。
示例代码：
立即学习“go语言免费学习笔记（深入）”；
func compressFiles(zipFile string, files []string) error {
outFile, err := os.Create(zipFile)
if err != nil {
return err
}
defer outFile.Close()
zipWriter := zip.NewWriter(outFile)
defer zipWriter.Close()
for _, filePath := range files {
err := filepath.Walk(filePath, func(path string, info os.FileInfo, err error) error {
if err != nil {
return err
}
// 创建相对路径
relPath, _ := filepath.Rel(".", path)
relPath = filepath.ToSlash(relPath)
if info.IsDir() {
// 添加目录条目
_, err := zipWriter.Create(relPath + "/")
return err
}
// 添加文件
fileInZip, err := zipWriter.Create(relPath)
if err != nil {
return err
}
file, err := os.Open(path)
if err != nil {
return err
}
defer file.Close()
_, err = io.Copy(fileInZip, file)
return err
})
if err != nil {
return err
}
}
return nil
}
解压ZIP文件到指定目录
解压操作需要读取ZIP文件中的每个条目，根据类型（文件或目录）创建对应结构，并将内容写入磁盘。
关键点包括：确保目标路径安全（防止路径穿越），还原文件权限，以及正确处理目录创建。
示例代码：
立即学习“go语言免费学习笔记（深入）”；
func decompressZip(zipFile, dest string) error {
reader, err := zip.OpenReader(zipFile)
if err != nil {
return err
}
defer reader.Close()
for _, file := range reader.File {
filePath := filepath.Join(dest, file.Name)
filePath = filepath.Clean(filePath)
// 防止路径穿越攻击
if !strings.HasPrefix(filePath, filepath.Clean(dest)+string(os.PathSeparator)) {
return fmt.Errorf("invalid file path: %s", filePath)
}
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.OpenFile(filePath, os.O_WRONLY|os.O_CREATE|os.O_TRUNC, file.Mode())
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
}
使用建议与注意事项
在实际项目中使用压缩解压功能时，注意以下几点可提升稳定性和安全性：
-
路径处理统一用
filepath.ToSlash()：避免Windows下出现反斜杠导致ZIP结构异常。 - 检查文件大小限制：防止解压超大文件或恶意压缩炸弹。
- 验证目标路径合法性：特别是解压时，禁止写入系统关键目录。
- 支持进度反馈（可选）：大文件压缩/解压时可通过回调函数通知进度。
- 错误处理要细致：如部分文件失败是否中断整体流程，需根据业务决定。
基本上就这些。Golang标准库对ZIP的支持足够应对大多数场景，无需引入外部依赖。只要注意路径、权限和资源释放，就能写出健壮的压缩解压逻辑。
