如何在 Go 中安全地重写 HTTP 请求体

在Go中间件中动态修改HTTP请求体：标准实践与关键细节

在Go语言构建的HTTP服务中，有时需要在请求抵达核心业务逻辑之前，对请求体进行拦截和修改。无论是为了实现敏感信息的日志脱敏、进行A/B测试的路由分发，还是完成不同协议间的转换，这都涉及到一个核心操作：如何安全地重写*http.Request的Body。

直接修改原始的r.Body是行不通的。因为它是一个只读的io.ReadCloser接口，其底层数据可能已被部分读取或缓冲。正确的思路是整体替换：用一个全新的、符合接口规范的实例来替换r.Body字段，并同步更新所有相关的请求元数据。

✅ 标准替换步骤

这个过程可以分解为几个清晰的步骤，确保每一步都稳固可靠。

1. 构造新请求体内容：首先，需要准备好新的请求体数据。对于字符串内容，strings.NewReader()是轻量且只读的优选；如果后续还有写入需求，则可以考虑bytes.NewBufferString()。
2. 包装为 io.ReadCloser：HTTP请求体要求实现io.ReadCloser接口。使用io.NopCloser()（Go 1.16+版本在io包中，旧版本在ioutil包）可以方便地将一个io.Reader包装成具备空Close方法的ReadCloser。
3. 更新 ContentLength：这是至关重要的一步。必须显式地设置r.ContentLength = int64(len(newBodyString))。如果忘记这一步，下游处理器可能会因为声明的长度与实际内容长度不匹配而解析失败，甚至导致数据截断。
4. （可选）处理 TransferEncoding：如果原始请求使用了分块传输编码（Transfer-Encoding: chunked），而新的请求体是固定长度的，就需要清空这个字段：r.TransferEncoding = nil，以避免协议层面的冲突。

func requestModifier(next http.Handler) http.Handler {
    return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        // 示例：将所有 POST 请求体统一替换为 JSON
        if r.Method == "POST" {
            newBody := `{"modified": true, "timestamp": 1717023456}`
            // ✅ 安全替换 Body
            r.Body = io.NopCloser(strings.NewReader(newBody))
            r.ContentLength = int64(len(newBody))
            r.Header.Set("Content-Type", "application/json")
            // ⚠️ 若原请求含 TransferEncoding，需手动清理
            if len(r.TransferEncoding) > 0 {
                r.TransferEncoding = nil
            }
        }
        next.ServeHTTP(w, r) // 传递给下一中间件或 handler
    })
}

⚠️ 关键注意事项

掌握了标准步骤，还需要避开几个常见的“坑”，才能确保方案在生产环境中的健壮性。

• 不可复用原 r.Body：原始的r.Body通常设计为一次性读取流。尝试多次Read()可能只会得到io.EOF，而直接修改其内部缓冲区极易引发程序恐慌（panic）或数据混乱。
• 避免内存泄漏：当新请求体来自大字符串或文件时，内存管理需要留意。推荐使用bytes.NewReader()配合io.NopCloser()的组合，它比bytes.Buffer更节省资源，因为后者会分配额外的可写内存空间。
• 编码一致性：务必确保新内容的字符编码（例如UTF-8）与Content-Type头部声明的编码一致。否则，接收方在解析时很可能遇到乱码或异常。
• 中间件顺序敏感：如果请求链路上游的某个中间件已经读取过r.Body（比如调用了r.ParseForm()），那么此时r.Body可能已经处于关闭状态。因此，请求体重写的逻辑最好放在中间件链的最前端执行。

遵循以上实践，就能在完整保持HTTP协议语义的前提下，实现安全、可控的请求体重写。这套方法非常适用于API网关、请求审计、Mock服务等多种服务端场景，为架构的灵活性与可观测性提供了坚实的技术支撑。