go proxy转发工作中碰到的问题

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)