中间件中如何访问请求主体

Golang小白一枚，正在不断学习积累知识，现将学习到的知识记录一下，也是将我的所得分享给大家！而今天这篇文章《中间件中如何访问请求主体》带大家来了解一下##content_title##，希望对大家的知识积累有所帮助，从而弥补自己的不足，助力实战开发！

我需要在中间件中获取请求正文。像这样的东西：

return func(w http.ResponseWriter, req *http.Request) {

    data, err := handler(w, req)

    if err != nil {
        buf := new(bytes.Buffer)
        buf.ReadFrom(req.Body)
        s := buf.String()

但是我得到了 s == "" 以及带有正文的 post 请求。 为什么以及如何解决它？

解决方案

您只能读取请求正文一次，但可以读取一次并创建所读取内容的副本。使用 t 恤阅读器

https://golang.org/pkg/io/#example_TeeReader

或者这个答案：

https://stackoverflow.com/a/23077519/6376471

两者都包括将正文读入内存一次，然后制作副本。这样，当您在中间件中使用它时，您可以将副本重新分配给请求对象，然后再将其转发给处理程序。

给你这样的东西：

return func (w http.ResponseWriter, req *http.Request) {
    buf, err := ioutil.ReadAll(r.Body) // handle the error
    rdr1 := ioutil.NopCloser(bytes.NewBuffer(buf))

    req.Body = rdr1
    data, err := handler(w, req)
    if err != nil {
        s := buf.String()
        // ... trimmed
    }
}

这显然会将请求正文的所有副作用都放入内存中，如果请求正文非常大等怎么办。并且您的中间件正在从处理程序等中夺走一些控制权。

今天关于《中间件中如何访问请求主体》的内容介绍就到此结束，如果有什么疑问或者建议，可以在golang学习网公众号下多多回复交流；文中若有不正之处，也希望回复留言以告知！