如何将 tlsConfig 与 RoundTripper 一起使用？

Question

我有这段代码实现了一个带有自定义 RoundTripper 的 http 客户端，以在Client用于进行 http 调用时默认添加一些标头。

var Client *http.Client

type customTransport struct {

    underlyingTransport http.RoundTripper

}

func (t *customTransport) RoundTrip(req *http.Request) (*http.Response, error) {

    req.Header.Add("Authorization", "Bearer")

    req.Header.Add("Version", "2017-11-23") 

    return t.underlyingTransport.RoundTrip(req)

}

func CreateClient() error {

    // Setup HTTPS client

    tlsConfig := &tls.Config{

        InsecureSkipVerify: true,

    }

    tlsConfig.BuildNameToCertificate()

    // transport := &http.Transport{TLSClientConfig: tlsConfig}

    // Client = &http.Client{Transport: transport}

    Client = &http.Client{Transport: &customTransport{underlyingTransport: http.DefaultTransport}}

    return nil

}

你可以看到我有一个tlsConfig. 我的问题是如何将其合并tlsConfig到我的客户中？

Answer 1

您可以通过在您的方法中调用原始/默认传输来处理此问题RoundTrip，您已经在执行此操作：return t.underlyingTransport.RoundTrip(req)所以你只需要正确配置它：Client = &http.Client{Transport: &customTransport{underlyingTransport: http.DefaultTransport}}你也已经在做。您唯一需要更改的是，您应该使用 TLS 配置使用传输，而不是http.DefaultTransport：transport := &http.Transport{    TLSClientConfig: tlsConfig,}Client = &http.Client{Transport: &customTransport{underlyingTransport: transport}}