如何检查和修改 Go 反向代理中的响应正文？

如何在 ReverseProxy 中检查和修改响应正文

在提供的 Go 代码中，设置了反向代理以将请求重定向到 Google。然而，为了获得更深入的见解或自定义响应，访问响应正文是必不可少的。

解决方案在于利用 ReverseProxy 结构中的 ModifyResponse 字段。此字段允许指定在 HTTP 响应到达客户端之前对其进行修改的函数。

实现

以下修改后的代码演示了如何读取和修改响应正文：

package main

import (
    "bytes"
    "fmt"
    "io"
    "io/ioutil"
    "net/http"
    "net/http/httputil"
    "net/url"
    "strconv"
)

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

    // Modify the response body before sending it to the client
    proxy.ModifyResponse = func(resp *http.Response) (err error) {
        b, err := ioutil.ReadAll(resp.Body) // Read the response body
        if err != nil {
            return err
        }
        err = resp.Body.Close() // Close the `Body` and `resp`
        if err != nil {
            return err
        }

        // Modify the response body
        b = bytes.Replace(b, []byte("server"), []byte("schmerver"), -1)

        // Create a new `body` to keep the `Content-Length` and `Body` up-to-date
        body := ioutil.NopCloser(bytes.NewReader(b))
        resp.Body = body
        resp.ContentLength = int64(len(b))
        resp.Header.Set("Content-Length", strconv.Itoa(len(b)))

        fmt.Println("Modified response: ", string(b)) // See the modified response

        return nil
    }

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

工作原理

ModifyResponse 函数使用 ioutil.ReadAll 将原始响应正文读取到缓冲区中。然后它关闭原来的resp.Body并修改body内容。

为了确保响应有效，创建了一个新的body并将其分配给resp.Body。 Content-Length 标头也会更新以反映新的正文长度。

最后，将修改后的响应正文打印到控制台以方便检查，并将修改后的响应发送到客户端。

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