在Golang中基于Gin框架实现不落地存储直接生成并提供文件下载

核心实现逻辑 #

不需要在服务端创建磁盘文件，直接将生成的文件内容写入HTTP响应流，通过配置对应响应头触发浏览器下载行为即可，全程只有内存操作，无磁盘IO。

基础实现代码（适合小文件） #

替换你原来的磁盘操作逻辑，直接写入字节流即可：

import (
    "net/http"
    "strconv"

    "github.com/gin-gonic/gin"
)

func serveFile(c *gin.Context) {
    // 此处替换为你实际生成文件内容的逻辑，支持任意二进制内容（文本、Excel、压缩包等均可）
    fileContent := []byte("hello world")
    downloadFileName := "hello.txt"

    // 配置响应头，触发浏览器下载
    c.Header("Content-Disposition", "attachment; filename="+downloadFileName)
    c.Header("Content-Length", strconv.Itoa(len(fileContent)))

    // 直接将内容写入响应，第二个参数为对应文件的MIME类型
    c.Data(http.StatusOK, "text/plain", fileContent)
}

流式实现代码（适合大文件） #

如果生成的文件体积较大，为了避免占用过多内存，可以使用流式写入方式，边生成内容边返回：

import (
    "fmt"
    "io"
    "net/http"

    "github.com/gin-gonic/gin"
)

func serveBigFile(c *gin.Context) {
    downloadFileName := "big.txt"
    // 配置响应头，触发浏览器下载
    c.Header("Content-Disposition", "attachment; filename="+downloadFileName)
    // 按实际文件类型修改MIME值
    c.Header("Content-Type", "text/plain")

    // 流式写入响应
    c.Stream(func(w io.Writer) bool {
        // 此处替换为你动态生成内容的逻辑，生成一块就写入一块
        for i := 0; i < 10000; i++ {
            if _, err := fmt.Fprintf(w, "第%d行测试内容\n", i); err != nil {
                // 写入出错时终止流
                return false
            }
        }
        // 所有内容写入完成后结束流
        return false
    })
}

补充说明 #

如果需要返回其他格式的文件，只需要修改Content-Type为对应MIME类型即可，例如CSV为text/csv、Excel为application/vnd.openxmlformats-officedocument.spreadsheetml.sheet、ZIP为application/zip。

内容的提问来源于stack exchange，提问作者Nuzurie