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

从现在开始，努力学习吧！本文《Golang多次读取请求正文[重复]》主要讲解了golang等等相关知识点，我会在golang学习网中持续更新相关的系列文章，欢迎大家关注并积极留言建议。下面就先一起来看一下本篇正文内容吧，希望能帮到你！

问题内容

这个问题在这里已经有了答案 ：

[在 Go 中以字符串形式访问 HTTP 响应](http://localhost:8080/questions/38673673/access-http- response-as-string-in-go) 4 个答案

1 年前关闭。

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

正确答案

检查和模拟请求正文

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

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

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

最后缺少的部分是制作io.Readeran io.ReadCloser，因为bytes.Buffer没有Close()方法。为此，您可以使用ioutil.NopCloser()which 包装 anio.Reader并返回 an 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)
        }
    })
}

本篇关于《Golang多次读取请求正文[重复]》的介绍就到此结束啦，但是学无止境，想要了解学习更多关于Golang的相关知识，请关注golang学习网公众号！