AWS X-Ray Go语言Lambda跨函数追踪及服务地图关联问题咨询

我刚好搞定过一模一样的需求！要让AWS X-Ray服务地图里清晰展示Go写的Lambda1调用Lambda2的关联关系，确实得靠自定义X-Ray段的方式来打通调用链，下面是具体的实现步骤和代码示例：

Lambda之间的调用默认不会自动在X-Ray里建立关联，得手动创建父-子段，把调用Lambda2的操作绑定到子段的上下文里，这样X-Ray就能识别出这是一条完整的调用链路。

首先确保你导入了必要的依赖包：

import (
    "context"
    "github.com/aws/aws-sdk-go/aws"
    "github.com/aws/aws-sdk-go/aws/session"
    "github.com/aws/aws-sdk-go/service/lambda"
    "github.com/aws/aws-xray-sdk-go/xray"
)

然后在Lambda1的处理函数里实现追踪逻辑：

func Lambda1Handler(ctx context.Context) error {
    // 基于Lambda的上下文创建自定义父段
    parentCtx, parentSegment := xray.BeginSegment(ctx, "lambda1-invocation")
    defer parentSegment.Close(nil)

    // 从父段衍生出子段，专门标记调用Lambda2的操作
    childCtx, childSegment := xray.BeginSubsegment(parentCtx, "call-lambda2")
    defer childSegment.Close(nil)

    // 初始化Lambda客户端，记得用X-Ray包装一下，确保追踪生效
    sess := session.Must(session.NewSession())
    client := lambda.New(sess, &aws.Config{Region: aws.String("us-east-1")})
    xray.AWS(client.Client)

    // 准备调用Lambda2的参数，替换成你的实际函数名和负载
    invokeInput := &lambda.InvokeInput{
        FunctionName: aws.String("your-lambda2-function-name"),
        Payload:      []byte(`{"data": "test-payload"}`),
    }

    // 关键：用子段的上下文调用Lambda2，这样X-Ray就能关联两者
    _, err := client.InvokeWithContext(childCtx, invokeInput)
    if err != nil {
        childSegment.Close(err)
        return err
    }

    return nil
}

• 两个Lambda都得在控制台开启Active tracing，不然X-Ray根本收不到追踪数据
• 给Lambda1的IAM角色加上这两个权限：AWSXRayDaemonWriteAccess（允许发送追踪数据）和lambda:InvokeFunction（允许调用Lambda2）
• 段的名称可以自定义，比如把lambda1-invocation改成你的业务标识，call-lambda2改成更贴合实际操作的名字，这样在X-Ray控制台里看链路会更清晰

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