Go 言語での http.Transport のリクエスト フィルタリングおよびインターセプト技術とアプリケーション

Go 言語における http.Transport のリクエスト フィルタリングおよびインターセプト技術とアプリケーション

1. はじめに
Go 言語の http パッケージでは、http.Transport が基礎となる HTTP トランスポート層として実装され、次の機能を提供します。 HTTP リクエストを開始するために使用できる強力な機能。この記事では、http.Transport を使用してリクエストのフィルタリングとインターセプトの手法とそのアプリケーションを実装する方法に焦点を当てます。

2. リクエストのフィルタリング
2.1 リクエスト メソッドのフィルタリング
http.Transport の RoundTrip メソッドを使用して、HTTP リクエストをカスタマイズできます。 http.RoundTripper インターフェイスを実装する構造を定義し、その中で RoundTrip メソッドをオーバーライドすることで、リクエストをフィルタリングして変更できます。次のコード例は、リクエスト メソッドをフィルタリングする方法を示しています。

type FilterTransport struct {
    Transport http.RoundTripper
}

func (t *FilterTransport) RoundTrip(req *http.Request) (*http.Response, error) {
    // 过滤请求方法
    if req.Method == http.MethodGet {
        return nil, errors.New("Method Not Allowed")
    }

    return t.Transport.RoundTrip(req)
}

func main() {
    // 创建一个Transport
    transport := &FilterTransport{
        Transport: http.DefaultTransport,
    }

    // 创建一个Client
    client := &http.Client{
        Transport: transport,
    }

    // 创建一个GET请求
    req, _ := http.NewRequest(http.MethodGet, "http://example.com", nil)

    // 发起请求
    resp, err := client.Do(req)
    if err != nil {
        fmt.Println("Error:", err)
    } else {
        fmt.Println("Response:", resp.Status)
    }
}

上記のコードでは、http.RoundTripper インターフェイスを実装する FilterTransport 構造体を作成し、RoundTrip メソッドで GET リクエストをフィルタリングしました。リクエストメソッドがGETの場合はエラーを返します。

2.2 リクエスト ヘッダーのフィルタリング
リクエスト メソッドのフィルタリングに加えて、リクエスト ヘッダーのフィルタリングと変更もできます。次のコード例は、リクエスト ヘッダーをフィルターして変更する方法を示しています。

type HeaderFilterTransport struct {
    Transport http.RoundTripper
}

func (t *HeaderFilterTransport) RoundTrip(req *http.Request) (*http.Response, error) {
    // 过滤请求头
    req.Header.Del("User-Agent")
    req.Header.Set("User-Agent", "Custom User-Agent")

    return t.Transport.RoundTrip(req)
}

func main() {
    // 创建一个Transport
    transport := &HeaderFilterTransport{
        Transport: http.DefaultTransport,
    }

    // 创建一个Client
    client := &http.Client{
        Transport: transport,
    }

    // 创建一个GET请求
    req, _ := http.NewRequest(http.MethodGet, "http://example.com", nil)

    // 发起请求
    resp, err := client.Do(req)
    if err != nil {
        fmt.Println("Error:", err)
    } else {
        fmt.Println("Response:", resp.Status)
    }
}

上記のコードでは、http.RoundTripper インターフェイスを実装する HeaderFilterTransport 構造体を作成し、RoundTrip メソッドの User-Agent リクエスト ヘッダーをフィルタリングして変更しました。元の User-Agent リクエスト ヘッダーを削除し、カスタム User-Agent を設定します。

3. リクエストのインターセプト
リクエストのフィルタリングに加えて、リクエストをインターセプトしたり、リクエスト本文を変更したり、リクエストをさらに処理したりすることもできます。次のコード例は、リクエストをインターセプトしてリクエスト本文を変更する方法を示しています。

type BodyInterceptorTransport struct {
    Transport http.RoundTripper
}

func (t *BodyInterceptorTransport) RoundTrip(req *http.Request) (*http.Response, error) {
    // 拦截请求并修改请求体
    if req.Method == http.MethodPost {
        // 读取原始请求体
        body, err := ioutil.ReadAll(req.Body)
        if err != nil {
            return nil, err
        }
        req.Body.Close()

        // 修改请求体
        newBody := bytes.NewReader([]byte("Modified Body"))

        // 替换请求体
        req.Body = ioutil.NopCloser(newBody)

        // 设置Content-Type
        req.Header.Set("Content-Type", "text/plain")
    }

    return t.Transport.RoundTrip(req)
}

func main() {
    // 创建一个Transport
    transport := &BodyInterceptorTransport{
        Transport: http.DefaultTransport,
    }

    // 创建一个Client
    client := &http.Client{
        Transport: transport,
    }

    // 创建一个POST请求
    req, _ := http.NewRequest(http.MethodPost, "http://example.com", strings.NewReader("Original Body"))

    // 发起请求
    resp, err := client.Do(req)
    if err != nil {
        fmt.Println("Error:", err)
    } else {
        fmt.Println("Response:", resp.Status)
    }
}

上記のコードでは、http.RoundTripper インターフェイスを実装する BodyInterceptorTransport 構造体を作成し、RoundTrip メソッドで POST リクエストをインターセプトし、リクエスト本文を変更しました。元のリクエスト本文を読み取り、カスタムで変更されたリクエスト本文に置き換え、Content-Type を text/plain に設定します。

4. 結論
http.Transport のリクエスト フィルタリングおよびインターセプト技術を使用すると、HTTP リクエストのフィルタリング、変更、インターセプトをカスタマイズできます。これは、いくつかの特殊な機能や要件を実装する場合に非常に役立ちます。ただし、通常のリクエスト処理やパフォーマンスへの影響を避けるために、使用中に注意が必要です。この記事が、Go 言語でのリクエストのフィルタリングとインターセプトに http.Transport を使用するのに役立つことを願っています。

以上がGo 言語での http.Transport のリクエスト フィルタリングおよびインターセプト技術とアプリケーションの詳細内容です。詳細については、PHP 中国語 Web サイトの他の関連記事を参照してください。