在 Golang 中如何使用 Transport 對 HTTP 請求進行自訂？

在 Go 中，可使用 Transport 介面自訂 HTTP 請求。具體步驟如下：實作 RoundTrip 方法，指定自訂行為（新增標頭、連線限制等）；建立自訂傳輸，實作 RoundTrip 方法；使用自訂傳輸建立 HTTP 用戶端；傳送請求，處理回應。

在Golang 中使用Transport 對HTTP 請求進行自訂

介紹

在Go 中，net/http 套件提供了一組標準的HTTP 用戶端和伺服器類型。然而，有時你可能需要對 HTTP 請求進行更精細的控制。這時，你可以使用 Transport 介面來建立自訂的 HTTP 傳輸機制。

Transport 介面

Transport 介面定義了用於執行和傳輸HTTP 請求所需的方法：

type Transport interface {
    RoundTrip(*Request) (*Response, error)
}

其中：

• RoundTrip 方法負責執行HTTP 請求並將回應傳回呼叫者。

自訂 Transport

要建立自訂的 Transport，你需要實作 RoundTrip 方法。在這個方法中，你可以指定對請求的自訂行為，例如：

• 新增自訂標頭
• 設定連線限制
• 使用代理伺服器
• 實作驗證機制

實戰案例：新增自訂標頭

讓我們透過一個實戰案例來說明如何使用Transport 新增自訂標頭：

程式碼

import (
    "context"
    "fmt"
    "net/http"
    "net/http/httputil"
)

type CustomTransport struct {
    http.RoundTripper
}

func (t *CustomTransport) RoundTrip(req *http.Request) (*http.Response, error) {
    // 添加自定义标头
    req.Header.Add("X-Custom-Header", "custom-value")

    // 执行请求
    resp, err := t.RoundTripper.RoundTrip(req)
    if err != nil {
        return nil, err
    }

    // 打印带标头的请求内容
    dump, _ := httputil.DumpRequest(req, true)
    fmt.Println(string(dump))

    return resp, nil
}

func main() {
    // 创建自定义传输
    transport := &CustomTransport{http.DefaultTransport}

    // 创建具有自定义传输的 HTTP 客户端
    client := &http.Client{
        Transport: transport,
    }

    // 发送请求
    resp, err := client.Get("https://example.com")
    if err != nil {
        panic(err)
    }
    _ = resp.Body.Close()  // 关闭响应体
}

#輸出

當你執行此程式碼時，你會在控制台中看到帶有自訂標頭的HTTP 請求內容。

GET https://example.com/ HTTP/1.1
Host: example.com
User-Agent: Go-http-client/1.1
X-Custom-Header: custom-value
...

以上是在 Golang 中如何使用 Transport 對 HTTP 請求進行自訂？的詳細內容。更多資訊請關注PHP中文網其他相關文章！