如何在 HTTP 中间件处理程序之间重用 *http.Request 的请求体？

Question

我使用 go-chi 作为 HTTP 路由器，我想在另一个方法中重用一个方法

func Registration(w http.ResponseWriter, r *http.Request) {
    b, err := ioutil.ReadAll(r.Body) // if you delete this line, the user will be created   
    // ...other code

    // if all good then create new user
    user.Create(w, r)
}

...

func Create(w http.ResponseWriter, r *http.Request) {
  b, err := ioutil.ReadAll(r.Body)  
  // ...other code

  // ... there I get the problem with parse JSON from &b
}

user.Create返回错误"unexpected end of JSON input"

其实，在我执行ioutil.ReadAll
user.Create之后就不再解析JSON了，
在r.Body 中有一个空数组[]我该如何解决这个问题？

Answer 1

外部处理程序将请求正文读取到 EOF。当调用内部处理程序时，不再从主体中读取任何内容。

要解决此问题，请使用先前在外部处理程序中读取的数据恢复请求正文：

func Registration(w http.ResponseWriter, r *http.Request) {
    b, err := ioutil.ReadAll(r.Body) 
    // ...other code
    r.Body = ioutil.NopCloser(bytes.NewReader(b))
    user.Create(w, r)
}

函数bytes.NewReader() 在字节片上返回io.Reader。函数ioutil.NopCloser 将io.Reader 转换为r.Body 所需的io.ReadCloser。

Answer 2

最后，我能够通过这种方式恢复数据：

r.Body = ioutil.NopCloser(bytes.NewBuffer(b))