如何在Google API Go库中同时使用自定义HTTP客户端和token source？

核心原理是Google Go SDK的WithTokenSource配置本质是给默认HTTP客户端封装带自动认证的传输层，传入自定义HTTP客户端时SDK不会自动注入该逻辑，因此需要手动将token source的认证能力集成到你的自定义客户端链路中。

实现步骤 #

1. 初始化你带埋点逻辑的自定义HTTP客户端

// 示例：自定义带埋点的传输层
type CustomTransport struct {
    Base http.RoundTripper
    // 你的埋点相关配置字段
}

func (t *CustomTransport) RoundTrip(req *http.Request) (*http.Response, error) {
    // 这里加你的请求埋点逻辑，比如统计请求耗时、上报请求参数等
    start := time.Now()
    resp, err := t.Base.RoundTrip(req)
    // 这里加你的响应埋点逻辑
    metric.RecordRequestCost(time.Since(start))
    return resp, err
}

// 初始化自定义HTTP客户端
customClient := &http.Client{
    Timeout: 30 * time.Second,
    Transport: &CustomTransport{
        Base: http.DefaultTransport,
    },
}

2. 集成token source到自定义客户端的传输链路

import "golang.org/x/oauth2"

// 初始化你需要的token source，可根据你的认证方式选择对应初始化方法，比如服务账号、默认凭据等
ts, err := google.DefaultTokenSource(context.Background(), "你需要的API权限作用域")
if err != nil {
    log.Fatalf("初始化token source失败: %v", err)
}

// 用认证传输层包装你的自定义传输层
authTransport := &oauth2.Transport{
    Source: ts,
    // 底层用你已经实现好的自定义埋点传输层
    Base: customClient.Transport,
}

// 替换自定义客户端的传输层为加了认证能力的版本
customClient.Transport = authTransport

3. 初始化Google API SDK时直接传入配置好的自定义客户端

import "google.golang.org/api/option"

// 不需要再额外传入WithTokenSource配置
svc, err := storage.NewService(context.Background(), option.WithHTTPClient(customClient))
if err != nil {
    log.Fatalf("初始化API服务失败: %v", err)
}

注意事项 #

• 不要同时配置WithTokenSource和WithHTTPClient，认证逻辑已经提前集成到自定义客户端中，重复配置会导致冲突
• 如果你自定义的传输层需要操作请求头，注意不要覆盖oauth2.Transport自动注入的Authorization请求头
• 该实现不会影响原有埋点逻辑的运行，所有请求都会先触发你自定义的埋点逻辑，再完成认证后发出

内容的提问来源于stack exchange，提问作者ThisIsErico