一尘不染

Golang 多次读取请求正文

go

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


阅读 203

收藏
2021-12-13

共1个答案

一尘不染

检查和模拟请求正文

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

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

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

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