如何使用Golang实现文件压缩下载_Golang 文件压缩与传输实践

package main

import (
    "archive/zip"
    "io"
    "net/http"
    "os"
)

func downloadZipHandler(w http.ResponseWriter, r *http.Request) {
    // 设置响应头，触发浏览器下载
    w.Header().Set("Content-Type", "application/zip")
    w.Header().Set("Content-Disposition", `attachment; filename="files.zip"`)

    // 创建 zip writer，直接写入 HTTP 响应
    zipWriter := zip.NewWriter(w)
    defer zipWriter.Close()

    // 要压缩的文件列表
    files := []string{"file1.txt", "file2.log", "config.json"}

    for _, fileName := range files {
        file, err := os.Open(fileName)
        if err != nil {
            http.Error(w, "无法打开文件: "+fileName, http.StatusInternalServerError)
            return
        }
        defer file.Close()

        // 获取文件信息，保留原始文件名
        fileInfo, err := file.Stat()
        if err != nil {
            http.Error(w, "无法读取文件信息: "+fileName, http.StatusInternalServerError)
            return
        }

        // 在 ZIP 中创建对应文件头
        fileHeader, err := zip.FileInfoHeader(fileInfo)
        if err != nil {
            http.Error(w, "创建文件头失败", http.StatusInternalServerError)
            return
        }
        fileHeader.Name = fileName // 保持原始文件名
        fileHeader.Method = zip.Deflate

        // 创建 ZIP 中的文件写入器
        writer, err := zipWriter.CreateHeader(fileHeader)
        if err != nil {
            http.Error(w, "创建压缩条目失败", http.StatusInternalServerError)
            return
        }

        // 将文件内容拷贝到压缩流
        _, err = io.Copy(writer, file)
        if err != nil {
            http.Error(w, "写入压缩数据失败", http.StatusInternalServerError)
            return
        }
    }
}

func main() {
    http.HandleFunc("/download", downloadZipHandler)
    http.ListenAndServe(":8080", nil)
}