如何在Go语言中利用net/http/httputil实现轻量级反向代理-Golang-PHP中文网

因为go标准库默认过滤非安全头，authorization、x-forwarded-for等关键头被静默丢弃；必须在director中手动补全，如req.header.set("authorization", req.header.get("authorization"))并正确处理x-forwarded-for追加逻辑。

如何在go语言中利用net/http/httputil实现轻量级反向代理

为什么直接用 `httputil.NewSingleHostReverseProxy` 会丢请求头？

默认情况下，httputil.NewSingleHostReverseProxy 会过滤掉部分客户端请求头（比如 Connection、Keep-Alive、Proxy-Authenticate 等），这是为了安全和协议合规——但实际开发中，你常需要透传 Authorization、X-Forwarded-For、自定义的 X-Trace-ID 等。不改默认行为，后端服务就收不到关键上下文。

解决办法是重写 Director 函数，并手动补全被删的头字段：

proxy := httputil.NewSingleHostReverseProxy(u)
proxy.Director = func(req *http.Request) {
    req.URL.Scheme = u.Scheme
    req.URL.Host = u.Host
    // 必须显式恢复被过滤的头
    req.Header.Add("X-Forwarded-For", req.RemoteAddr)
    if clientIP := req.Header.Get("X-Real-IP"); clientIP != "" {
        req.Header.Set("X-Forwarded-For", clientIP)
    }
    // 若需透传 Authorization，取消注释下一行
    // req.Header.Set("Authorization", req.Header.Get("Authorization"))
}

如何让反向代理支持 WebSocket 升级？

httputil.ReverseProxy 默认不处理 Upgrade 请求，遇到 GET + Upgrade: websocket 会直接返回 501 Not Implemented。这不是 bug，而是设计上要求你显式启用。

必须在 Transport 中允许升级，并在 Director 中保留相关头：

设置 proxy.Transport 的 FlushInterval 避免缓冲阻塞
在 Director 中复制 Connection 和 Upgrade 头
添加 Upgrade 到 hopHeaders 黑名单里（否则会被自动删除）

示例关键片段：

proxy.Transport = &http.Transport{
    FlushInterval: 10 * time.Millisecond,
}
proxy.Director = func(req *http.Request) {
    req.URL.Scheme = u.Scheme
    req.URL.Host = u.Host
    // 关键：透传升级头
    if upgrade := req.Header.Get("Upgrade"); upgrade != "" {
        req.Header.Set("Upgrade", upgrade)
        req.Header.Set("Connection", "Upgrade")
    }
}
// 移除 hop-by-hop 头时跳过 Upgrade 和 Connection
hopHeaders := []string{"Connection", "Upgrade", "Keep-Alive", "Proxy-Authenticate", "Proxy-Authorization", "Te", "Trailers", "Transfer-Encoding", "WWW-Authenticate"}
for _, h := range hopHeaders {
    if h == "Upgrade" || h == "Connection" {
        continue
    }
    req.Header.Del(h)
}

`Director` 函数里修改 `req.URL.Path` 为什么没生效？

常见误区：以为改了 req.URL.Path 就能重写路径，结果后端收到的还是原始路径。根本原因是 ReverseProxy 在转发前会把 req.URL 整体拼成目标地址，而 Path 只是其中一部分；如果 req.URL.RawPath 非空，它会优先使用 RawPath，导致 Path 修改被忽略。

Go语言（Golang）1.26.0

Go语言（Golang）1.26.0版本提供 Go 官方 Windows amd64 MSI 安装包下载入口，版本号 1.26.0，可用于旧项目维护、兼容性测试和指定版本开发环境配置。

下载

正确做法是统一操作 RawPath，或清空它再设 Path：

若路径含 URL 编码字符（如 %2F），务必用 req.URL.EscapedPath() 而非 req.URL.Path
修改前先清空 RawPath：req.URL.RawPath = ""
路径重写建议用 strings.TrimPrefix + path.Join，避免手动拼接出错

例如将 /api/v1/ 代理到后端根路径：

proxy.Director = func(req *http.Request) {
    req.URL.Scheme = u.Scheme
    req.URL.Host = u.Host
    req.URL.Path = strings.TrimPrefix(req.URL.Path, "/api/v1")
    req.URL.RawPath = "" // 强制使用 Path 字段
}

代理响应体过大时为什么连接会提前关闭？

httputil.ReverseProxy 默认使用 io.Copy 转发响应体，但没做流控或超时控制。当后端返回大文件（比如 100MB 日志）且客户端网络慢时，容易触发底层 TCP 连接超时或 net/http 的默认 WriteTimeout，表现为“connection reset by peer”或响应截断。

应对方式不是加大超时，而是分块拷贝并加健康检查：

给 proxy.Transport 设置 ResponseHeaderTimeout 和 IdleConnTimeout
用 io.CopyBuffer 替代 io.Copy，指定 32KB 缓冲区提升吞吐
在 ModifyResponse 中检查 Content-Length，对超大响应加 Content-Disposition 提示下载

简单缓冲替换示例：

originalDirector := proxy.Director
proxy.Director = func(req *http.Request) {
    originalDirector(req)
}
proxy.ModifyResponse = func(resp *http.Response) error {
    // 可在此注入 header 或拦截错误响应
    return nil
}
// 实际转发时需自定义 ServeHTTP 才能换 io.CopyBuffer，标准 proxy.ServeHTTP 不可插拔
// 所以更推荐：直接调用 proxy.Transport.RoundTrip 并自己 copy

真正要稳住大响应，得绕过 proxy.ServeHTTP，自己实现转发逻辑——这点容易被文档带偏，ReverseProxy 的封装便利性是以牺牲细粒度控制为代价的。