首页 >后端开发 >Golang >如何访问 Go 反向代理中的响应正文?

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

Mary-Kate Olsen
Mary-Kate Olsen原创
2024-11-08 08:20:021080浏览

How to Access the Response Body in a Go Reverse Proxy?

访问反向代理中的响应体

使用反向代理时,您可能会遇到需要访问从后端收到的响应体的情况服务器。本文深入探讨了如何使用 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中文网其他相关文章!

声明:
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn