时间戳

首页 > 解决方案 > 如何将 tlsConfig 与 RoundTripper 一起使用？

go - 如何将 tlsConfig 与 RoundTripper 一起使用？

问题描述

我有这段代码实现了一个带有自定义 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到我的客户中？

标签： go

解决方案

您可以通过在您的方法中调用原始/默认传输来处理此问题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}}

推荐阅读