Golang中Http Client的使用问题求助

Golang中Http Client的使用问题求助

package client

import (
	"context"
	"crypto/tls"
	"crypto/x509"
	"fmt"
	"net/http"
	"net/url"
	"strings"

	"golang.org/x/oauth2"
	"golang.org/x/oauth2/clientcredentials"
)

func makeRequest() {

	credentials := &clientcredentials.Config{
		ClientID:     "ClientID",
		ClientSecret: "Secret",
		TokenURL:     fmt.Sprintf("%s/v1/token", "www.exampleAuth.com"),
		Scopes:       []string{"FullScope"},
		EndpointParams: url.Values{
			"client_id":     {"ClientID"},
			"client_secret": {"Secret"},
		},
	}

	client := credentials.Client(context.TODO())

	caCertPool := x509.NewCertPool()
	_ = caCertPool.AppendCertsFromPEM([]byte("Some RSA cert"))
        
        // 这种方法可行，但不确定是否会搞乱某些东西
	client.Transport.(*oauth2.Transport).Base = &http.Transport{
		TLSClientConfig: &tls.Config{
			RootCAs: caCertPool,
		},
	}
	req, _ := http.NewRequest("POST", "www.example.com", strings.NewReader("hi"))
	client.Do(req)
}

大家好，

我正在尝试编写一个客户端，它基本上使用带有客户端凭证的HTTP客户端，并且还需要使用根CA证书来建立信任。

我实现这一目标的方式如下所述：

我的问题是，这是实现相同目标的好方法吗？覆盖从oauth2/clientcredentials返回的客户端上的Transport会导致任何问题吗？

oauth2/clientcredentials的文档确实指定不要更改返回客户端上的传输。是否有更好的方法/包来实现相同的目标？

任何帮助都将是非常大的帮助。谢谢！

编辑：经过一些代码阅读，发现在oauth2.NewClient()方法中可以传递上下文，其中包含我们在HTTP客户端上传输实现的上下文键。


更多关于Golang中Http Client的使用问题求助的实战教程也可以访问 https://www.itying.com/category-94-b0.html

1 回复

更多关于Golang中Http Client的使用问题求助的实战系列教程也可以访问 https://www.itying.com/category-94-b0.html


在Golang中，你确实可以通过更安全的方式配置自定义TLS配置，而不需要直接修改oauth2.Transport的Base字段。以下是推荐的实现方法：

package client

import (
	"context"
	"crypto/tls"
	"crypto/x509"
	"fmt"
	"net/http"
	"net/url"
	"strings"

	"golang.org/x/oauth2"
	"golang.org/x/oauth2/clientcredentials"
)

func makeRequest() {
	// 创建自定义HTTP客户端
	caCertPool := x509.NewCertPool()
	_ = caCertPool.AppendCertsFromPEM([]byte("Some RSA cert"))
	
	customTransport := &http.Transport{
		TLSClientConfig: &tls.Config{
			RootCAs: caCertPool,
		},
	}
	
	customClient := &http.Client{
		Transport: customTransport,
	}

	// 创建带自定义HTTP客户端的上下文
	ctx := context.WithValue(context.Background(), oauth2.HTTPClient, customClient)

	credentials := &clientcredentials.Config{
		ClientID:     "ClientID",
		ClientSecret: "Secret",
		TokenURL:     fmt.Sprintf("%s/v1/token", "www.exampleAuth.com"),
		Scopes:       []string{"FullScope"},
		EndpointParams: url.Values{
			"client_id":     {"ClientID"},
			"client_secret": {"Secret"},
		},
	}

	// 使用自定义上下文的客户端
	client := credentials.Client(ctx)

	req, _ := http.NewRequest("POST", "www.example.com", strings.NewReader("hi"))
	resp, err := client.Do(req)
	if err != nil {
		fmt.Printf("请求失败: %v\n", err)
		return
	}
	defer resp.Body.Close()
	
	fmt.Printf("响应状态码: %d\n", resp.StatusCode)
}

这种方法利用了oauth2包提供的oauth2.HTTPClient上下文键，这是官方推荐的方式来注入自定义HTTP客户端配置。这样做的好处是：

  1. 不会破坏oauth2包内部的传输机制
  2. 保持了oauth2包对token自动刷新的支持
  3. 代码更加清晰和安全

如果你需要更细粒度的控制，也可以直接使用自定义HTTP客户端进行所有请求：

func makeRequestWithCustomClient() {
	// 配置TLS
	caCertPool := x509.NewCertPool()
	_ = caCertPool.AppendCertsFromPEM([]byte("Some RSA cert"))
	
	transport := &http.Transport{
		TLSClientConfig: &tls.Config{
			RootCAs: caCertPool,
		},
	}
	
	client := &http.Client{
		Transport: transport,
	}

	// 手动处理OAuth2令牌获取
	token, err := getOAuth2Token(client)
	if err != nil {
		fmt.Printf("获取令牌失败: %v\n", err)
		return
	}

	// 使用令牌创建请求
	req, _ := http.NewRequest("POST", "www.example.com", strings.NewReader("hi"))
	req.Header.Set("Authorization", "Bearer "+token.AccessToken)
	
	resp, err := client.Do(req)
	if err != nil {
		fmt.Printf("请求失败: %v\n", err)
		return
	}
	defer resp.Body.Close()
	
	fmt.Printf("响应状态码: %d\n", resp.StatusCode)
}

func getOAuth2Token(client *http.Client) (*oauth2.Token, error) {
	ctx := context.WithValue(context.Background(), oauth2.HTTPClient, client)
	
	config := &clientcredentials.Config{
		ClientID:     "ClientID",
		ClientSecret: "Secret", 
		TokenURL:     fmt.Sprintf("%s/v1/token", "www.exampleAuth.com"),
		Scopes:       []string{"FullScope"},
	}
	
	return config.Token(ctx)
}

第一种方法是最简洁且符合oauth2包设计意图的解决方案。

回到顶部