問題及場景

業(yè)務當中有需要分發(fā)http.request.body的場景。比如微信回調(diào)消息只能指定一個地址，所以期望可以復制一份消息發(fā)給其他服務。由服務B和接收微信回調(diào)的服務A一起處理微信回調(diào)信息。

本文將詳細介紹golang復用http.request.body的相關內(nèi)容，分享出來供大家參考學習，下面話不多說了，來一起看看詳細的介紹吧

解決思路

最開始考慮的是直接轉發(fā)http.request。使用ReverseProxy直接將http.request由服務A轉發(fā)給服務B。但是微信涉及到驗證等問題，完全調(diào)整好非常麻煩。所以轉換思路，打算將http.request.body的內(nèi)容直接post給服務B。

可是http.request是readcloser。我們將http.request readAll的時候講無法再次讀取http.request里面的信息。

如何才能將http.request.body復制使用呢？

其中c表示的是http的上下文

 // 把request的內(nèi)容讀取出來
 var bodyBytes []byte
 if c.Request.Body != nil {
  bodyBytes, _ = ioutil.ReadAll(c.Request.Body)
 }
 // 把剛剛讀出來的再寫進去
 c.Request.Body = ioutil.NopCloser(bytes.NewBuffer(bodyBytes))

1.我們先將body從http.request里面讀取出來，保存到一個變量里面。

2.然后再將變量里面的數(shù)據(jù)使用ioutil.NopCloser方法寫回到http.request里面。

https://golang.org/pkg/io/ioutil/#NopCloser

NopCloser returns a ReadCloser with a no-op Close method wrapping the provided Reader r.

NopCloser用一個無操作的Close方法包裝Reader r返回一個ReadCloser接口。

這樣我們就可以再次使用c.request來進行處理了。

總結

以上就是這篇文章的全部內(nèi)容了，希望本文的內(nèi)容對大家的學習或者工作具有一定的參考學習價值，如果有疑問大家可以留言交流，謝謝大家對腳本之家的支持。