AWS X-Ray与Lambda、Kinesis集成：全请求追踪配置咨询

问题根源 #

Lambda消费Kinesis流时，X-Ray默认复用函数启动时的采样决策，不会为每条Kinesis事件重新生成追踪上下文。若初始采样标记为sampled=false，转发到第二个Lambda的请求会继承该标记，导致多数请求无法被追踪。

解决方法 #

1. 手动生成强制采样的追踪上下文 #

在第一个Lambda（Kinesis消费者）中，为每个待转发的请求创建新的X-Ray子段，强制设置采样为true，并将新的追踪头传递给第二个Lambda。Node.js示例代码：

const AWS = require('aws-sdk');
const AWSXRay = require('aws-xray-sdk');
const lambda = AWSXRay.captureAWSClient(new AWS.Lambda());

exports.handler = async (event) => {
  for (const record of event.Records) {
    const parentSegment = AWSXRay.getSegment();
    // 创建新子段并强制采样
    const invokeSubsegment = parentSegment.addNewSubsegment('call-second-lambda');
    invokeSubsegment.setSampling(true);

    try {
      // 构造新的X-Ray追踪头
      const traceHeader = `Root=${invokeSubsegment.trace_id};Parent=${invokeSubsegment.id};Sampled=1`;
      await lambda.invoke({
        FunctionName: 'your-second-lambda-function-name',
        Payload: JSON.stringify(record),
        // 传递追踪头
        Headers: {
          'X-Amzn-Trace-Id': traceHeader
        }
      }).promise();
    } catch (err) {
      invokeSubsegment.addError(err);
      throw err;
    } finally {
      invokeSubsegment.close();
    }
  }
};

2. 配置X-Ray自定义采样规则 #

在AWS X-Ray控制台创建自定义采样规则：

• 目标服务指定第二个Lambda的ARN
• 设置固定采样率为100%（或业务所需的高比例）
• 将规则优先级设为最高，确保覆盖默认采样规则

注意：采样率100%会增加X-Ray的使用成本，需根据实际业务场景调整。

3. 确保Lambda启用主动追踪 #

在两个Lambda的配置页面（控制台→Lambda→配置→监控和操作工具→X-Ray）都启用主动追踪，保证追踪链的完整性。第一个Lambda启用X-Ray后，才能正确生成和传递追踪上下文。

已知问题参考（原GitHub Issue内容翻译） #

Node.js版AWS X-Ray SDK存在已知行为：当Lambda处理Kinesis等流式事件时，SDK会复用函数初始化阶段的采样决策，不会为每条事件重新评估采样规则。即使第二个Lambda单独启用了X-Ray，也会继承上游传递的Sampled=false标记，导致无法追踪。社区给出的临时解决方案正是上述手动生成追踪上下文或调整采样规则的方式。

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