問題及場景
業(yè)務當中有需要分發(fā)http.request.body的場景。比如微信回調(diào)消息只能指定一個地址,所以期望可以復制一份消息發(fā)給其他服務。由服務B和接收微信回調(diào)的服務A一起處理微信回調(diào)信息。
解決思路
最開始考慮的是直接轉(zhuǎn)發(fā)http.request。使用ReverseProxy直接將http.request由服務A轉(zhuǎn)發(fā)給服務B。但是微信涉及到驗證等問題,完全調(diào)整好非常麻煩。所以轉(zhuǎn)換思路,打算將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來進行處理了。