在 HTTP 处理实现中,检查请求主体对于处理某些请求至关重要,例如带有表单编码数据的 POST 请求。但是,使用 req.ParseForm() 等方法直接检查可能会破坏请求状态,从而导致将请求转发到反向代理时出错。
要解决此问题,可以在不离开的情况下查看请求正文原始请求对象中的任何跟踪。这可以为后续消费者(例如反向代理)维护原始状态。
一种有效的方法是将请求正文读入中间缓冲区并使用该缓冲区创建多个读取器。下面的代码片段演示了这一点:
// Read the request body into a buffer buf, _ := io.ReadAll(r.Body) // Create two readers from the buffer rdr1 := io.NopCloser(bytes.NewBuffer(buf)) rdr2 := io.NopCloser(bytes.NewBuffer(buf)) // Perform desired operations on the first reader doStuff(rdr1) // Reset the request body with the second reader r.Body = rdr2
在此代码中,bytes.NewBuffer(buf) 创建包装存储缓冲区的读取器。 io.NopCloser 确保 Close() 方法不会返回错误,这是读者实现 io.ReadCloser 接口所必需的。
代码片段修改了问题中提供的示例,其中 doStuff() 是处理请求正文的函数:
buf, _ := io.ReadAll(r.Body) rdr1 := io.NopCloser(bytes.NewBuffer(buf)) rdr2 := io.NopCloser(bytes.NewBuffer(buf)) doStuff(rdr1) r.Body = rdr2
经过此修改,可以在不影响原始请求状态的情况下检查请求正文。然后,反向代理可以继续处理具有未修改正文的请求。
以上是如何在不修改请求状态的情况下检查HTTP请求体?的详细内容。更多信息请关注PHP中文网其他相关文章!