Golang多次读取请求正文[重复]

Question

我正在编写自己的登录中间件。基本上，我需要记录请求和响应的正文。我面临的问题是，当我阅读正文时，它变成了空的，我无法阅读两次。 我知道它发生是因为它是 ReadCloser 类型。有没有办法将身体倒回开始？

Answer 1

检查和模拟请求正文

当您第一次读取正文时，您必须存储它，因此一旦您完成它，您可以设置一个新的io.ReadCloser 作为从原始数据构造的请求正文。因此，当您在链中前进时，下一个处理程序可以读取相同的主体。

一种选择是使用ioutil.ReadAll() 读取整个正文，这会将正文作为字节切片。

您可以使用bytes.NewBuffer() 从字节切片中获取io.Reader。

最后缺少的部分是将io.Reader 设为io.ReadCloser，因为bytes.Buffer 没有Close() 方法。为此，您可以使用 ioutil.NopCloser()，它包装了一个 io.Reader，并返回一个 io.ReadCloser，其添加的 Close() 方法将是一个无操作（什么都不做）。

请注意，您甚至可以修改用于创建“新”正文的字节片的内容。您可以完全控制它。

^{但必须小心，因为如果您只修改数据，可能还有其他 HTTP 字段（如内容长度和校验和）可能会变得无效。如果后续处理程序检查这些，您还需要修改它们！}

检查/修改响应正文

如果你还想阅读响应正文，那么你必须包装你得到的http.ResponseWriter，并在链上传递包装器。这个包装器可以缓存发送的数据，您可以在之后即时检查（当后续处理程序写入时）。

这里有一个简单的ResponseWriter 包装器，它只是缓存数据，所以在后续处理程序返回后就可以使用：

type MyResponseWriter struct {
    http.ResponseWriter
    buf *bytes.Buffer
}

func (mrw *MyResponseWriter) Write(p []byte) (int, error) {
    return mrw.buf.Write(p)
}

请注意，MyResponseWriter.Write() 只是将数据写入缓冲区。您还可以选择即时检查它（在Write() 方法中）并将数据立即写入包装/嵌入的ResponseWriter。您甚至可以修改数据。您拥有完全的控制权。

^{但必须再次小心，因为后续处理程序也可能发送与响应数据相关的 HTTP 响应标头（例如长度或校验和），如果您更改响应数据，这些标头也可能变得无效。}

完整示例

将各个部分放在一起，这是一个完整的工作示例：

func loginmw(handler http.Handler) http.Handler {
    return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        body, err := ioutil.ReadAll(r.Body)
        if err != nil {
            log.Printf("Error reading body: %v", err)
            http.Error(w, "can't read body", http.StatusBadRequest)
            return
        }

        // Work / inspect body. You may even modify it!

        // And now set a new body, which will simulate the same data we read:
        r.Body = ioutil.NopCloser(bytes.NewBuffer(body))

        // Create a response wrapper:
        mrw := &MyResponseWriter{
            ResponseWriter: w,
            buf:            &bytes.Buffer{},
        }

        // Call next handler, passing the response wrapper:
        handler.ServeHTTP(mrw, r)

        // Now inspect response, and finally send it out:
        // (You can also modify it before sending it out!)
        if _, err := io.Copy(w, mrw.buf); err != nil {
            log.Printf("Failed to send out response: %v", err)
        }
    })
}

Answer 2

我可以使用 Request 包中的 GetBody。

在 net/http 中 request.go 的源代码中查看此评论：

GetBody 定义了一个可选的函数来返回一个新的 身体。当重定向需要时，它用于客户端请求 不止一次地阅读身体。仍然使用 GetBody 需要设置 Body。 对于服务器请求，它是未使用的。”

GetBody func() (io.ReadCloser, error)

这样你就可以得到body请求而不用让它为空。

示例：

getBody := request.GetBody
copyBody, err := getBody()
if err != nil {
    // Do something return err
}
http.DefaultClient.Do(request)