Golang读取请求正文

Question

我正在编写自己的logginMiddleware.基本上,我需要记录请求的主体和响应.我遇到的问题是,当我读到身体时,它会变空,我无法读取它两次.我知道它发生是因为它是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)
        }
    })
}