如何使用Golang开发Web文件压缩与解压功能_Golang Web文件压缩处理实践

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("解压成功"))
}

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结构
}

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),
    })
}