欢迎您访问新疆栾骏商贸有限公司,公司主营电子五金轴承产品批发业务!
全国咨询热线: 400-8878-609

新闻资讯

技术教程

Go如何在Go中实现文件下载_Go文件下载实现思路

作者:P粉6029986702026-01-16 00:00:00
Go中文件下载需手动读取并设置Content-Disposition等Header以控制下载行为,支持断点续传需用http.ServeContent,客户端下载要处理重定向和大小限制,中文文件名需按RFC 5987编码或使用ASCII替代。

HTTP服务端直接返回文件流

Go标准库的http.ServeFile适合静态文件直传,但缺乏控制权(比如无法添加自定义Header、无法做权限校验、无法记录下载日志)。更稳妥的做法是手动读取文件并写入ResponseWriter

关键点:必须设置Content-Disposition Header让浏览器触发下载,且要区分是否强制下载(attachment)还是尝试内联(inline);同时建议设置Content-TypeContent-Length提升兼容性与性能。

func downloadHandler(w http.ResponseWriter, r *http.Request) {
    filename := r.URL.Query().Get("name")
    if filename == "" {
        http.Error(w, "missing 'name'", http.StatusBadRequest)
        return
    }

    // 安全检查:防止路径遍历
    if strings.Contains(filename, "..") || strings.HasPrefix(filename, "/") {
        http.Error(w, "invalid filename", http.StatusForbidden)
        return
    }

    filepath := "./uploads/" + filename
    file, err := os.Open(filepath)
    if err != nil {
        http.Error(w, "file not found", http.StatusNotFound)
        return
    }
    defer file.Close()

    // 获取文件信息以设置 Content-Length
    fi, _ := file.Stat()
    w.Header().Set("Content-Type", "application/octet-stream")
    w.Header().Set("Content-Disposition", `attachment; filename="`+filename+`"`)
    w.Header().Set("Content-Length", fmt.Sprintf("%d", fi.Size()))

    // 直接拷贝到响应体,避免内存加载整文件
    io.Copy(w, file)
}

使用http.ServeContent支持断点续传

当需要支持大文件下载或断点续传(如用户暂停后重连),不能用io.Copy简单转发。必须用http.ServeContent,它会自动解析Range请求头、返回206 Partial Content,并处理边界条件。

注意:http.ServeContent要求传入一个http.FileInfo接口实现,且ModTime()需返回准确时间——否则可能跳过缓存协商或返回错误状态码。

  • 务必用os.Stat获取真实文件信息,不要自己构造FileInfo
  • serveContentmodtime参数必须与FileInfo.ModTime()一致,否则ETag/Last-Modified校验失效
  • 若文件内容动态生成(非磁盘文件),需自行实现http.File接口

客户端发起下载请求时如何处理重定向和大文件

http.Client下载文件时,默认不自动跟随重定向(CheckRedirect默认为nil),而很多下载链接是302跳转到CDN地址。不处理会导致Body为空。

另一个常见问题是未限制响应体大小,遇到恶

意服务返回超大响应体可能OOM。应结合io.LimitReader或检查Content-Length提前拒绝。

client := &http.Client{
    CheckRedirect: func(req *http.Request, via []*http.Request) error {
        return nil // 允许重定向
    },
}

resp, err := client.Get("https://example.com/file.zip")
if err != nil {
    log.Fatal(err)
}
defer resp.Body.Close()

if resp.StatusCode != http.StatusOK {
    log.Fatalf("unexpected status: %s", resp.Status)
}

// 可选:限制最大下载体积(例如 1GB)
limitedBody := io.LimitReader(resp.Body, 1024*1024*1024)

out, err := os.Create("downloaded.zip")
if err != nil {
    log.Fatal(err)
}
defer out.Close()

io.Copy(out, limitedBody)

文件名中文乱码与安全截断

浏览器对Content-Disposition中中文filename的支持不一:filename="中文.pdf"在Chrome可行,但在旧版Safari或IE中会乱码或截断。RFC 5987推荐用filename*=UTF-8''encoded格式,但Go标准库不自动编码,需手动处理。

更实际的方案是:服务端统一用ASCII文件名(如UUID),并在URL query里带原始名称供前端显示;或用url.PathEscape编码后拼进Header(兼容性更好)。

  • 永远校验客户端传入的filename参数,过滤/..、空字节等
  • 避免直接用r.URL.Query().Get("filename")拼路径,应先path.Clean再检查是否仍在允许目录下
  • 如果必须透出原始中文名,用fmt.Sprintf(`filename*=UTF-8''%s`, url.PathEscape(origName))构造filename*字段

真正难的不是写几行io.Copy,而是把路径校验、Header构造、重定向策略、大小限制、字符编码这些细节串成一条不出错的链路——漏掉任意一环,上线后都可能变成线上事故。