如何使用 AWS X-Ray 跟踪 lambda 调用 lambda？

Question

我有一个由 API 网关事件触发的 AWS Lambda function。 API 网关配置为使用 X-Ray。

由于 Lambda 跟踪配置默认为 PassTrough，因此它也显示在 X-Ray 中（服务 map 等）。

调用的 Lambda 使用 node.js aws-sdk 调用另一个 lambda。 如果我理解正确，则必须将跟踪 ID 传递给下一次调用，以便在 X 射线中也显示此 Lambda。 在SDK 的 API 中，我没有找到任何选项。

const result = await lambda
        .invoke(lambdaParamsCreateUser)
        .promise()

我怎样才能做到这一点？ 如何跟踪原始请求的调用？

在@Balu Vyamajala 的提示下，我将 AWS-SDK 导入更改为以下内容：

import AWS from "aws-sdk";

import AwsXRay from "aws-xray-sdk-core";
const aws = AwsXRay.captureAWS(AWS);
export default aws;

我在为我的第二个 function 开具发票时使用它，如下所示：

import AWS from "aws";
const Lambda = AWS.Lambda;
// ...
const lambda = new Lambda({ region: "eu-central-1" });
const lambdaPromise = lambda
        .invoke({
            FunctionName: AUTH_CREATE_USER_FUNC,
            InvocationType: "RequestResponse",
            Qualifier: AUTH_CREATE_USER_FUNC_VERSION,
            Payload: JSON.stringify({
                eMail: eMail,
                device: device,
                customerId: customerId,
            }),
            LogType: "Tail",
        })
        .promise()

但是在 X-Ray 中没有调用链：-(

https://imgur.com/wDMlNzb

我犯错了吗？

Answer 1

如果我们为两个 Lambda 函数启用 X-Ray，trace-id 会自动传递，并且对于两个 Lambda 都是相同的。

在代码中，我们只需将 X-Ray 包裹在 aws-sdk 周围即可启用

JavaScript：

const AWSXRay = require("aws-xray-sdk-core");
const AWS = AWSXRay.captureAWS(require("aws-sdk"));    

Typescript：

import AWSXRay from 'aws-xray-sdk';
import aws from 'aws-sdk';
const AWS = AWSXRay.captureAWS(aws)

这是一个样本测试来确认。

balu-test >> sample-test

Lambda 1（巴鲁测试）：

const AWSXRay = require("aws-xray-sdk-core");
const AWS = AWSXRay.captureAWS(require("aws-sdk"));    
const lambda = new AWS.Lambda();

exports.handler = async function (event, context) {
  var params = {
    FunctionName: "sample-test",
    InvocationType: "RequestResponse",
    Payload: '{ "name" : "foo" }',
  };
  
  const response = await lambda.invoke(params).promise();
  console.log('response',response);

  return "sucess";
};

Lambda 2（样品测试）：

const AWSXRay = require("aws-xray-sdk-core");
const AWS = AWSXRay.captureAWS(require("aws-sdk"));    
let region = "us-east-1"
let secretName = "SomeSecret"
let secret
let decodedBinarySecret    
var client = new AWS.SecretsManager({
  region: region,
});

exports.handler = (event, context, callback) => {
  client.getSecretValue({ SecretId: secretName }, function (err, data) {
    if (err) {
      callback(err);
    } else {
      if ("SecretString" in data) {
        secret = data.SecretString;
      } else {
        let buff = new Buffer(data.SecretBinary, "base64");
        decodedBinarySecret = buff.toString("ascii");
      }
      callback(null, secret);
    }
  });
};

TraceId 是相同的，并且 X-Ray 对于 Lambda 调用都指向相同的图形。 当从 Api-Gateway 调用第一个 api 时，也会发生同样的事情。 第一次生成跟踪 ID 并作为 http header 传递到下游进程。