在 Go 中读取缓冲区并将其重写为 http.Response

Question

我想在 golang 中编写一个 HTTP 代理。我将此模块用于代理：https : //github.com/elazarl/goproxy。当有人使用我的代理时，它会调用一个以 http.Response 作为输入的函数。我们称之为“resp”。resp.Body 是一个 io.ReadCloser。我可以使用它的 Read 方法将它读入 []byte 数组。但是随后它的内容从 resp.Body 中消失了。但是我必须返回一个 http.Response 和我读入 []byte 数组的 Body。我怎样才能做到这一点？

你好，

最大限度

我的代码：

proxy.OnResponse().DoFunc(func(resp *http.Response, ctx *goproxy.ProxyCtx) *http.Response {

   body := resp.Body

   var readBody []byte

   nread, readerr := body.Read(readBody)

   //the body is now empty

   //and i have to return a body

   //with the contents i read.

   //how can i do that?

   //doing return resp gives a Response with an empty body

}

Answer 1

您将必须首先阅读所有正文，以便您可以正确关闭它。一旦您读取了整个正文，您就可以简单地将其替换为Response.Body您的缓冲区。readBody, err := ioutil.ReadAll(resp.Body)if err != nil {    // handle error}resp.Body.Close()// use readBodyresp.Body = ioutil.NopCloser(bytes.NewReader(readBody))

Answer 2

那是因为io.Reader它更像是一个缓冲区，当你读取它时，你已经消耗了缓冲区中的数据，并留下一个空的主体。要解决该问题，您只需要关闭响应正文并ReadCloser从正文中创建一个新的正文，该正文现在是一个字符串。import "io/ioutil"readBody, err := ioutil.ReadAll(resp.Body)if err != nil {     // }resp.Body.Close()resp.Body = ioutil.NopCloser(bytes.NewReader(readBody))