go - 如何在代理请求时在 Go Gin 中设置新的请求主体

Question

我正在尝试代理 HTTP 调用，并尝试在将其发送到代理服务之前修改 JSON 正文。但是，如果我尝试使用任何新数据修改 c.Request.Body，POST 请求将以 400 错误格式结束。但是，如果我再次将相同的先前主体数据设置为 c.Request.Body，则代理调用可以正常工作。

Gin 函数

func Forward(c *gin.Context) {
    remoteURL := c.Query("url")
    remote, _ := url.Parse(remoteURL)

    var bodyBytes []byte
    if c.Request.Body != nil {
        bodyBytes, _ = ioutil.ReadAll(c.Request.Body)
    }

    newBody := string(bodyBytes)
    newBody = strings.Replace(newBody, "testString", "testString1", -1)

    c.Request.Body = ioutil.NopCloser(bytes.NewBuffer([]byte(newBody)))

    proxy := httputil.NewSingleHostReverseProxy(remote)
    proxy.Director = func(req *http.Request) {
        req.Header = c.Request.Header
        req.Host = remote.Host
        req.URL.Scheme = remote.Scheme
        req.URL.Host = remote.Host
        req.URL.Path = remote.Path
    }

    proxy.ServeHTTP(c.Writer, c.Request)
}

curl 命令:

curl --location --request POST 'http://localhost:8020/v1/proxy?url=https://entgkdbyzqm27.x.pipedream.net/' \
--header 'Content-Type: application/json' \
--data-raw '{
    "id": "testString"
}'

如果我能知道如何在 Gin 中正确设置请求主体，我将不胜感激。

Answer 1

您可能有一个 Content-Length 不匹配。替换后，新的主体比以前的主体更长。

编写Director函数如下:

    proxy.Director = func(req *http.Request) {
        // clone the headers
        req.Header = c.Request.Header.Clone()

        // 1. set new header
        req.Header.Set("Content-Length", strconv.Itoa(len(newBody)))

        // 2. also update this field
        req.ContentLength = int64(len(newBody))

        // the rest stays the same
        req.Host = remote.Host
        req.URL.Scheme = remote.Scheme
        req.URL.Host = remote.Host
        req.URL.Path = remote.Path
    }