如何访问 Go 反向代理中的响应正文？

访问反向代理中的响应体

使用反向代理时，您可能会遇到需要访问从后端收到的响应体的情况服务器。本文深入探讨了如何使用 httputil 包在 Go 中检索响应正文。

在提供的代码片段中，您有一个简单的反向代理，可以将请求转发到 Google：

target := &url.URL{Scheme: "http", Host: "www.google.com"}
proxy := httputil.NewSingleHostReverseProxy(target)

http.Handle("/google", proxy)
http.ListenAndServe(":8099", nil)

访问响应体的关键在于ReverseProxy类型的ModifyResponse字段。根据官方文档，ModifyResponse 是一个可选函数，允许您修改后端的响应。

下面是一个在将响应正文发送到客户端之前读取和修改响应正文的示例：

import (
    "bytes"
    "fmt"
    "io/ioutil"
)

func rewriteBody(resp *http.Response) (err error) {
    b, err := ioutil.ReadAll(resp.Body) //Read html
    if err != nil {
        return err
    }
    err = resp.Body.Close()
    if err != nil {
        return err
    }
    b = bytes.Replace(b, []byte("server"), []byte("schmerver"), -1) // replace html
    body := ioutil.NopCloser(bytes.NewReader(b))
    resp.Body = body
    resp.ContentLength = int64(len(b))
    resp.Header.Set("Content-Length", strconv.Itoa(len(b)))
    return nil
}

func main() {
    target, _ := url.Parse("http://example.com")
    proxy := httputil.NewSingleHostReverseProxy(target)
    proxy.ModifyResponse = rewriteBody

    http.Handle("/google", proxy)
    http.ListenAndServe(":8099", nil)
}

在此代码中，rewriteBody 函数读取响应正文，对其进行修改（在本例中，将单词“server”替换为“schmerver”），然后将修改后的正文设置为新的响应正文。

以上是如何访问 Go 反向代理中的响应正文？的详细内容。更多信息请关注PHP中文网其他相关文章！