使用Go的archive/zip和net/http实现Web文件压缩解压:接收ZIP上传并解压至指定目录,支持multipart表单解析与流式读取。

在Golang开发Web应用时,经常需要处理用户上传的文件压缩包或提供打包下载功能。实现文件的压缩与解压不仅能节省带宽,还能提升用户体验。Go语言标准库中的 archive/zip 包提供了完整的ZIP文件操作支持,结合 net/http 可轻松实现Web端的压缩与解压功能。
1. 实现ZIP文件在线解压功能
当用户上传一个ZIP文件时,服务端可以即时解压并读取内容。以下是一个通过HTTP接口接收ZIP文件并解压到指定目录的示例:
func unzipHandler(w http.ResponseWriter, r *http.Request) {
// 只允许POST方法
if r.Method != "POST" {
http.Error(w, "仅支持POST请求", http.StatusMethodNotAllowed)
return
}
// 解析multipart表单,限制内存使用(例如32MB)
err := r.ParseMultipartForm(32 << 20)
if err != nil {
http.Error(w, "解析表单失败", http.StatusBadRequest)
return
}
file, handler, err := r.FormFile("zipfile")
if err != nil {
http.Error(w, "获取文件失败", http.StatusBadRequest)
return
}
defer file.Close()
// 创建临时文件保存上传的ZIP
tempFile, err := os.CreateTemp("", "*.zip")
if err != nil {
http.Error(w, "创建临时文件失败", http.StatusInternalServerError)
return
}
defer os.Remove(tempFile.Name()) // 处理完成后删除
defer tempFile.Close()
// 将上传内容写入临时文件
_, err = io.Copy(tempFile, file)
if err != nil {
http.Error(w, "保存文件失败", http.StatusInternalServerError)
return
}
// 打开ZIP文件进行解压
zipReader, err := zip.OpenReader(tempFile.Name())
if err != nil {
http.Error(w, "打开ZIP失败", http.StatusBadRequest)
return
}
defer zipReader.Close()
targetDir := "./uploads/" // 解压目标目录
for _, f := range zipReader.File {
filePath := filepath.Join(targetDir, f.Name)
// 防止路径遍历攻击
if !strings.HasPrefix(filePath, filepath.Clean(targetDir)+string(os.PathSeparator)) {
http.Error(w, "非法文件路径", http.StatusBadRequest)
return
}
if f.FileInfo().IsDir() {
os.MkdirAll(filePath, os.ModePerm)
continue
}
// 确保父目录存在
os.MkdirAll(filepath.Dir(filePath), os.ModePerm)
out, err := os.OpenFile(filePath, os.O_WRONLY|os.O_CREATE|os.O_TRUNC, f.Mode())
if err != nil {
http.Error(w, "创建文件失败: "+filePath, http.StatusInternalServerError)
return
}
rc, err := f.Open()
if err != nil {
out.Close()
http.Error(w, "读取ZIP内文件失败", http.StatusInternalServerError)
return
}
_, err = io.Copy(out, rc)
rc.Close()
out.Close()
if err != nil {
http.Error(w, "写入文件失败", http.StatusInternalServerError)
return
}
}
w.Write([]byte("解压成功"))
}
2. 动态打包多个文件为ZIP并下载
有时需要将多个文件(如日志、图片等)打包成ZIP供用户下载。可以通过 zip.Writer 将文件写入HTTP响应流中,无需生成本地临时文件。
func downloadZipHandler(w http.ResponseWriter, r *http.Request) {
// 设置响应头,触发浏览器下载
w.Header().Set("Content-Type", "application/zip")
w.Header().Set("Content-Disposition", `attachment; filename="files.zip"`)
zipWriter := zip.NewWriter(w)
defer zipWriter.Close()
filesToZip := []string{"./data/file1.txt", "./data/image.png"}
for _, filePath := range filesToZip {
fileInfo, err := os.Stat(filePath)
if err != nil || fileInfo.IsDir() {
continue
}
file, err := os.Open(filePath)
if err != nil {
continue
}
defer file.Close()
// 在ZIP中创建新文件
fw, err := zipWriter.Create(filepath.Base(filePath))
if err != nil {
file.Close()
continue
}
_, err = io.Copy(fw, file)
file.Close()
if err != nil {
return // 写入失败直接中断
}
}
// 注意:不要手动调用w.Write()或其他写操作,以免破坏ZIP结构
}
3. 安全与性能注意事项
在生产环境中使用压缩功能需注意以下几点:
立即学习“go语言免费学习笔记(深入)”;
- 防止ZIP炸弹:限制上传文件大小,避免解压后占用过多磁盘空间。可设置最大解压体积阈值,或逐个检查每个文件头中的未压缩大小。
- 路径遍历防护:解压时验证文件路径是否超出目标目录,防止恶意ZIP利用 ../ 跳出目录写入系统文件。
- 内存控制:大文件应流式处理,避免一次性加载整个ZIP到内存。使用 io.Pipe 可实现边压缩边输出。
- 并发安全:多个用户同时打包下载时,确保临时资源隔离,避免文件名冲突。
4. 支持在线预览压缩包内容
不实际解压的情况下查看ZIP内文件列表,可用于前端展示压缩包结构:
func listZipHandler(w http.ResponseWriter, r *http.Request) {
path := r.URL.Query().Get("file")
if path == "" {
http.Error(w, "缺少文件参数", http.StatusBadRequest)
return
}
reader, err := zip.OpenReader(path)
if err != nil {
http.Error(w, "无法读取ZIP文件", http.StatusBadRequest)
return
}
defer reader.Close()
var files []string
for _, f := range reader.File {
files = append(files, f.Name)
}
json.NewEncoder(w).Encode(map[string]interface{}{
"files": files,
"total": len(files),
})
}
基本上就这些。使用Golang的标准库就能高效实现Web场景下的文件压缩与解压,代码简洁且运行稳定,适合集成进各类文件服务系统中。关键是处理好边界情况和安全性问题,才能保证功能可靠可用。










