我想在 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}
2 回答
慕田峪9158850
TA貢獻1794條經驗 獲得超8個贊
您將必須首先閱讀所有正文，以便您可以正確關閉它。一旦您讀取了整個正文，您就可以簡單地將其替換為Response.Body您的緩沖區。
readBody, err := ioutil.ReadAll(resp.Body)
if err != nil {
// handle error
}
resp.Body.Close()
// use readBody
resp.Body = ioutil.NopCloser(bytes.NewReader(readBody))
浮云間
TA貢獻1829條經驗 獲得超4個贊
那是因為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))
- 2 回答
- 0 關注
- 543 瀏覽
添加回答
舉報
0/150
提交
取消