读取http的主体。不修改请求状态的请求？

Question

我有一个实现http的类型。Handler接口，在其ServeHTTP方法中，检查传入的HTTP请求，采取一些操作，然后将请求转发到反向代理处理程序（httputil.NewSingleHostReverseProxy）。

只要我只检查基本请求属性（如URL或标头），就可以正常工作。

当我想检查传入POST请求的主体时，例如通过调用req。ParseForm（）然后使用req。Form属性，一旦请求传递到反向代理，我就会遇到错误：

http:proxy错误：http:Request。ContentLength=687，正文长度为0

我认为这是因为查看HTTP请求的主体会导致req。身体读卡器要排空的流，这意味着代理处理程序无法再次读取该流。

我一直在玩io之类的东西。复制（）和bufio。Peek（），但我真的没有什么进展。

有没有办法偷看HTTP请求主体（并使用req的内置解析）。ParseForm等），同时将原始请求对象保持在其原始状态，以便可以将其传递给反向代理？

Answer 1

最近我用了一种不同的方法。有一个可用的GetBody方法，您可以通过该方法获取请求正文的新副本，因此不必执行以下操作：

doStuff(r.Body)

你可以改为：

body, _ := r.GetBody()
doStuff(body)
// r.Body is unmodified

这允许您在检查请求主体的同时仍保留它，以便稍后进行进一步处理

Answer 2

尝试读取缓冲区，然后使用缓冲区备份两个新读取器，一个供您使用，一个供后续消费者使用。例如，假设我们要修改以下代码：

doStuff(r.Body) // r is an http.Request

我们可以做到：

buf, _ := ioutil.ReadAll(r.Body)
rdr1 := ioutil.NopCloser(bytes.NewBuffer(buf))
rdr2 := ioutil.NopCloser(bytes.NewBuffer(buf))

doStuff(rdr1)
r.Body = rdr2 // OK since rdr2 implements the io.ReadCloser interface

// Now the program can continue oblivious to the fact that
// r.Body was ever touched.

请注意，*字节。缓冲区没有Close（）错误方法，因此它不实现io。ReadCloser界面。因此，我们必须包装*字节。调用ioutil时缓冲区值。NoCloser。