如何在不缓冲的情况下记录请求正文（通过中间件）？

当前位置： > > > > 如何在不缓冲的情况下记录请求正文（通过中间件）？

来源：stackoverflow
2024-04-19 21:12:28
0浏览
收藏

今天将给大家带来《如何在不缓冲的情况下记录请求正文（通过中间件）？》，感兴趣的朋友请继续看下去吧！以下内容将会涉及到等等知识点，如果你是正在学习Golang或者已经是大佬级别了，都非常欢迎也希望大家都能给我建议评论哈~希望能帮助到大家！

问题内容

假设我有这个简单的服务器（为了简洁起见，省略了错误处理）：

func main() {
    http.handlefunc("/", helloserver)
    http.listenandserve(":8080", nil)
}

func helloserver(w http.responsewriter, r *http.request) {
    b, _ := ioutil.readall(r.body)
    fmt.fprintf(w, "the request is: %s", string(b))
}

我需要添加一个简单的中间件来读取请求正文并记录它。我已经有了这个工作解决方案：

func requestLogger(next http.HandlerFunc) http.HandlerFunc {
    return func(w http.ResponseWriter, r *http.Request) {
        pr, pw := io.Pipe()
        tee := io.TeeReader(r.Body, pw)
        r.Body = pr
        go func() {
            body, _ := ioutil.ReadAll(tee)
            defer pw.Close()
            log.Printf("This is the logged request: %s", string(body))
        }()
        next(w, r)
    }
}

我通过反复试验直观地得出了这个解决方案，所以我不确定这是否适用于所有情况。这些是我对自己的解决方案的疑问：

1. 为什么我必须手动关闭 pipe？ ioutil.readall() 在 r.body 上遇到 eof 时不会关闭它。
2. 为什么这不会阻塞？存在循环引用，因为 teereader 从与通过管道 (r.body = pr) 写入的同一源 (r.body) 读取数据。根据文档，写入 pipewriter 会阻塞，直到满足一个或多个读取，而对于 teereader，写入必须在读取完成之前完成。

解决方案

这不是记录的行为。另外，io.Reader 不是 io.ReadCloser。

终于介绍完啦！小伙伴们，这篇关于《如何在不缓冲的情况下记录请求正文（通过中间件）？》的介绍应该让你收获多多了吧！欢迎大家收藏或分享给更多需要学习的朋友吧~公众号也会发布Golang相关知识，快来关注吧！