本文介绍如何在 Gin 框架中构建反向代理中间件，并通过 httputil.ReverseProxy.ModifyResponse 钩子拦截、读取并安全修改上游微服务返回的 HTTP 响应（如添加/修改 Header、重写 Body 等）。
本文介绍如何在 gin 框架中构建反向代理中间件，并通过 `httputil.reverseproxy.modifyresponse` 钩子拦截、读取并安全修改上游微服务返回的 http 响应（如添加/修改 header、重写 body 等）。
在微服务架构中，常需由 Go 后端（如 Gin 应用）作为网关代理客户端请求至下游微服务，并在响应返回给客户端前进行统一处理——例如注入追踪头（X-Request-ID）、脱敏敏感字段、添加 CORS 支持、或动态重写响应体。原生 net/http/httputil.NewSingleHostReverseProxy 已提供强大基础能力，而 ModifyResponse 字段正是官方推荐的响应定制入口。
以下是一个生产就绪的 Gin 反向代理中间件示例，支持安全修改响应 Header 与 Body：
import (
"io"
"log"
"net/http"
"net/http/httputil"
"net/url"
"strings"
"github.com/gin-gonic/gin"
)
func ReverseProxy(target string) gin.HandlerFunc {
u, err := url.Parse(target)
if err != nil {
log.Fatalf("invalid proxy target %s: %v", target, err)
}
proxy := httputil.NewSingleHostReverseProxy(u)
// ✅ 关键：注册 ModifyResponse 钩子，用于修改上游响应
proxy.ModifyResponse = func(r *http.Response) error {
// 示例 1：添加自定义响应头
r.Header.Set("X-Proxy-By", "Gin-Reverse-Proxy")
r.Header.Set("X-Upstream-Status", strings.TrimSpace(r.Status))
// 示例 2：安全重写响应体（需确保 Content-Length 和 Transfer-Encoding 一致）
if r.Body != nil && strings.Contains(r.Header.Get("Content-Type"), "application/json") {
// 注意：必须先读取原始 body，再构造新 response body
body, err := io.ReadAll(r.Body)
r.Body.Close()
if err != nil {
return err
}
// 示例：为 JSON 响应注入顶层字段（谨慎用于简单场景）
modifiedBody := []byte(`{"proxy_timestamp":` + strconv.FormatInt(time.Now().Unix(), 10) + `,"data":` + string(body) + `}`)
// 替换响应体并更新长度头
r.Body = io.NopCloser(bytes.NewReader(modifiedBody))
r.ContentLength = int64(len(modifiedBody))
r.Header.Set("Content-Length", strconv.Itoa(len(modifiedBody)))
}
return nil
}
// ⚠️ 重要：禁用默认的 Director 修改（避免重复 Host 设置），或按需定制
proxy.Director = func(req *http.Request) {
req.URL.Scheme = u.Scheme
req.URL.Host = u.Host
req.Host = u.Host // 显式设置 Host，避免被 Gin 自动覆盖
}
return func(c *gin.Context) {
// 防止 Gin 自动写入 header/body 干扰代理流
c.Status(http.StatusOK)
proxy.ServeHTTP(c.Writer, c.Request)
}
}
? 关键注意事项：
-
ModifyResponse在响应头已写入但响应体尚未发送时触发，是修改 Header 的安全时机；但修改 Body 需手动读取并重设r.Body，务必同步更新Content-Length或移除该头以支持 chunked 编码。 - 不要直接修改
c.Writer（如c.Header()或c.Data()），这会破坏代理流；所有变更必须通过r *http.Response完成。 - 若需访问原始请求上下文（如
c.Param、c.Get("user_id")），需提前在Director或中间件中将数据存入req.Context()，再在ModifyResponse中通过r.Request.Context()提取（ModifyResponse接收的是代理后的*http.Response，其Request字段保留了上下文）。 - 对于大文件或流式响应（如视频、长连接 SSE），避免
io.ReadAll，应使用io.Copy+io.TeeReader等流式处理方式。
✅ 总结：ReverseProxy.ModifyResponse 是 Gin 场景下轻量、高效、符合 Go HTTP 生态的最佳实践方案。合理利用它，你无需引入第三方代理库，即可构建具备响应增强能力的企业级 API 网关层。
