golang復用http.request.body

問題及場景

業(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來進行處理了。

golang復用http-request-body

最后編輯于
?著作權歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成,瀏覽時請結(jié)合常識與多方信息審慎甄別。
平臺聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點,簡書系信息發(fā)布平臺,僅提供信息存儲服務。

相關閱讀更多精彩內(nèi)容

友情鏈接更多精彩內(nèi)容