Блог

Главная - Вопросы по программированию - Как установить новое тело запроса в Go Gin при проксировании запроса

Как установить новое тело запроса в Go Gin при проксировании запроса

# #go #go-gin

Вопрос:

Я пытаюсь выполнить прокси-вызов HTTP и пытаюсь изменить тело JSON перед отправкой его в прокси-службу. Однако, если я попытаюсь изменить запрос c.Тело с любыми новыми данными запрос POST заканчивается в плохом формате 400. Но если я снова установлю те же самые предыдущие данные тела в c.Request.Тело, то вызов прокси-сервера работает без проблем.

Функция Джина

 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 --location --request POST 'http://localhost:8020/v1/proxy?url=https://entgkdbyzqm27.x.pipedream.net/' 
--header 'Content-Type: application/json' 
--data-raw '{
    "id": "testString"
}'

Я был бы очень признателен, если бы знал, как правильно задать тело запроса в Gin.

Ответ №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
    }
Метки: