golang复用http.request.body的方法示例
作者:ppmoon
问题及场景
业务当中有需要分发http.request.body的场景。比如微信回调消息只能指定一个地址,所以期望可以复制一份消息发给其他服务。由服务B和接收微信回调的服务A一起处理微信回调信息。
本文将详细介绍golang复用http.request.body的相关内容,分享出来供大家参考学习,下面话不多说了,来一起看看详细的介绍吧
解决思路
最开始考虑的是直接转发http.request。使用ReverseProxy直接将http.request由服务A转发给服务B。但是微信涉及到验证等问题,完全调整好非常麻烦。所以转换思路,打算将http.request.body的内容直接post给服务B。
可是http.request是readcloser。我们将http.request readAll的时候讲无法再次读取http.request里面的信息。
如何才能将http.request.body复制使用呢?
其中c表示的是http的上下文
// 把request的内容读取出来 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.然后再将变量里面的数据使用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来进行处理了。
总结
以上就是这篇文章的全部内容了,希望本文的内容对大家的学习或者工作具有一定的参考学习价值,如果有疑问大家可以留言交流,谢谢大家对脚本之家的支持。