一尘不染

Golang阅读请求正文

go

我正在编写自己的logginMiddleware。基本上,我需要记录请求和响应的正文。我面临的问题是,当我阅读正文时,它变成空的,无法两次阅读。我知道发生这种情况是因为它属于ReadCloser类型。有没有一种方法可以将身体重绕到开始?


阅读 342

收藏
2020-07-02

共1个答案

一尘不染

检查和嘲笑请求正文

初次阅读主体时,必须对其进行存储,以便在处理完该主体之后,可以将新io.ReadCloser的主体设置为根据原始数据构造的请求主体。因此,当您在链中前进时,下一个处理程序可以读取同一正文。

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

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

最后缺少的是使io.Readeran
io.ReadCloser,因为bytes.Buffer没有Close()方法。为此,您可以使用ioutil.NopCloser()包裹io.Reader,然后返回io.ReadCloser,其添加的Close()方法将是no-
op(不执行任何操作)。

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

但是,必须小心,因为如果仅修改数据,可能还有其他HTTP字段,例如content-
length和checksum,它们可能变得无效。如果后续处理程序检查了这些内容,那么您也需要对其进行修改!

检查/修改响应主体

如果您还想读取响应主体,则必须包装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)
        }
    })
}
2020-07-02