17370845950

本文介绍如何使用 `io.copy` 将上游 http 响应直接流式写入 `http.responsewriter`，避免内存积压，实现零拷贝式代理转发。

在构建反向代理、文件网关或 API 聚合层时，常需将外部服务返回的响应（如图片、PDF、视频流）原样透传给客户端。若先读取全部响应体到内存（如 ioutil.ReadAll(resp.Body)），不仅浪费内存，还可能因大文件导致 OOM 或显著延迟。Go 标准库提供了更优雅的流式方案：直接管道化（piping）http.Response.Body 到 http.ResponseWriter。

核心工具是 io.Copy，它以固定缓冲区（默认 32KB）循环读取 Reader 并写入 Writer，全程无须加载全部数据到内存：

func pipeResponse(w http.ResponseWriter, r *http.Request) {
    // 1. 发起上游请求（生产环境建议复用 http.Client 并设置超时）
    resp, err := http.Get("https://example.com/large-file.pdf")
    if err != nil {
        http.Error(w, "Failed to fetch resource: "+err.Error(), http.StatusBadGateway)
        return
    }
    defer resp.Body.Close() // 确保资源释放

    // 2. 复制关键响应头（Content-Type、Content-Length 等）
    // 注意：不要盲目复制所有 header（如 Connection、Transfer-Encoding），应有选择地透传
    for name, values := range resp.Header {
        for _, value := range values {
            w.Header().Add(name, value)
        }
    }

    // 3. 设置状态码（可选，默认 200；若需保留上游状态码，用 w.WriteHeader(resp.StatusCode)）
    w.WriteHeader(resp.StatusCode)

    // 4. 流式传输主体内容
    _, err = io.Copy(w, resp.Body)
    if err != nil {
        
// 客户端断连时 io.Copy 可能返回 io.ErrUnexpectedEOF 或 net/http.ErrAbortHandler，
        // 通常可忽略（无需额外错误处理），但日志记录有助于调试
        log.Printf("Stream copy interrupted: %v", err)
    }
}

⚠️ 关键注意事项：

• 不要调用 resp.Body.Close() 后再 io.Copy —— 示例中 defer resp.Body.Close() 放在 io.Copy 之后才安全；若提前关闭，会导致 io.Copy 读取空内容。
• 谨慎透传 Header：避免传递 Connection、Keep-Alive、Transfer-Encoding 等 hop-by-hop 字段（HTTP/1.1 规范要求中间件必须移除），否则可能引发客户端解析错误。推荐仅透传语义性字段（如 Content-Type, Content-Disposition, Cache-Control）。
• 超时与重试：生产环境务必为 http.Client 配置 Timeout 或 Context，防止上游挂起阻塞整个 goroutine。
• 状态码处理：io.Copy 不影响 HTTP 状态码。若需精确还原上游状态码，应在 io.Copy 前显式调用 w.WriteHeader(resp.StatusCode)。

✅ 为什么不用 io.Pipe？
io.Pipe() 用于在两个 goroutine 间创建配对的 PipeReader/PipeWriter，适用于需要异步解耦读写场景（如边下载边解密）。而本例中 resp.Body 和 http.ResponseWriter 已是就绪的 io.Reader 和 io.Writer，直接 io.Copy 更简洁、零额外 goroutine 开销，是标准且最优解。

总结：io.Copy(w, r.Body) 是 Go 中流式代理的基石操作——轻量、可靠、内存友好。搭配合理的 Header 过滤与错误处理，即可构建高性能、低延迟的 HTTP 响应透传服务。