golang復(fù)用http.request.body的方法示例

問題及場景

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

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

解決思路

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

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

如何才能將http.request.body復(fù)制使用呢？

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

 // 把request的內(nèi)容讀取出來
 var bodyBytes []byte
 if c.Request.Body != nil {
  bodyBytes, _ = ioutil.ReadAll(c.Request.Body)
 }
 // 把剛剛讀出來的再寫進(jìn)去
 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來進(jìn)行處理了。

總結(jié)

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