goproxy转发工作中碰到的问题

OGeek|极客世界-中国程序员成长平台 › 门户 › 编程› Go›Go教程

原作者: [db:作者] 来自: [db:来源] 收藏邀请

A-B

需求是一个中转

A-Proxy-B

读取来源请求A，在proxy读取body作些处理，再转给B，再把返回内容转给A

问题出在proxy这里

如果先把请求给B，再读body

res, err := transport.RoundTrip(req)

bodyBytes, _ := ioutil.ReadAll(req.Body)

则报错 http: invalid Read on closed Body

如果先

bodyBytes, _ := ioutil.ReadAll(req.Body) 
res, err := transport.RoundTrip(req)

则无响应

通用B的后端来看，请求的body为空，没了

req.Body只能一次读取，取完之后就关闭

req.body),只能读一次,代理读了，转发给B就空了；转发给B了，就读不到了。

网上找到解决办法答案

https://medium.com/@xoen/golang-read-from-an-io-readwriter-without-loosing-its-content-2c6911805361

代理先读一次

读出后再写入req

再转发

多了写入req的步骤

代码如下

// 先读
var bodyBytes []byte
if c.Request.Body != nil {
  bodyBytes, _ = ioutil.ReadAll(c.Request.Body)
}
// 写回去
c.Request.Body = ioutil.NopCloser(bytes.NewBuffer(bodyBytes))
// 转发
res,err:=transport.RoundTrip(req)
// 这里就可以随意处理了
bodyString := string(bodyBytes)