如何将Golang Lambda中的Cognito客户端接入AWS X-Ray追踪？

AWS X-Ray SDK for Go针对AWS SDK v1和v2的集成方式不同，直接调用xray.AWS()会因类型不兼容报错——因为xray.AWS()是为v1的*client.Client设计的，而v2的Cognito客户端结构完全不同。

解决方法是通过X-Ray的HTTP客户端中间件来集成v2 SDK，步骤如下：

1. 创建带X-Ray追踪的HTTP客户端
   用xray.Client()包装标准HTTP客户端，X-Ray会自动处理请求的追踪注入：

   httpClient := xray.Client(&http.Client{})
   

2. 基于带追踪的HTTP客户端创建Cognito客户端
   在加载AWS配置时指定这个HTTP客户端，让Cognito的所有请求都被X-Ray追踪：

   import (
       "context"
       "net/http"
   
       "github.com/aws/aws-sdk-go-v2/aws"
       "github.com/aws/aws-sdk-go-v2/config"
       "github.com/aws/aws-sdk-go-v2/service/cognitoidentityprovider"
       "github.com/aws/aws-xray-sdk-go/xray"
   )
   
   func handler(ctx context.Context) {
       cfg, err := config.LoadDefaultConfig(ctx,
           config.WithHTTPClient(xray.Client(&http.Client{})),
       )
       if err != nil {
           panic(err)
       }
   
       // 创建受X-Ray追踪的Cognito客户端
       cognitoClient := cognitoidentityprovider.NewFromConfig(cfg)
   
       // 后续使用cognitoClient发起的请求都会被X-Ray追踪，出现在服务地图中
   }
   

3. 验证前置条件

   • 确保Lambda函数已开启X-Ray追踪：在Lambda控制台的"配置"->"监控和操作工具"中，将X-Ray追踪模式设为"活动"。
   • 确保Lambda执行角色拥有AWSXRayDaemonWriteAccess权限，允许向X-Ray发送追踪数据。

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