如何在CDK中为Step Functions ASL关联指定栈内的Lambda函数

你不需要直接获取Lambda的物理ID，有几种更可靠的方式来完成关联：

方法1：在ASL文件中使用占位符，CDK部署时替换为Lambda ARN #

Step Functions的Task状态支持直接使用Lambda的ARN作为FunctionName的值，比物理ID更稳定。具体步骤：

1. 在本地ASL文件中用占位符标记Lambda的位置，比如：

   {
     "States": {
       "InvokeLambda": {
         "Type": "Task",
         "Resource": "arn:aws:states:::lambda:invoke",
         "Parameters": {
           "FunctionName": "${LAMBDA_ARN}",
           "Payload.$": "$"
         },
         "End": true
       }
     }
   }
   

2. 在CDK代码中读取ASL文件，替换占位符为Lambda的ARN，再创建状态机：

   import * as fs from 'fs';
   import * as cdk from 'aws-cdk-lib';
   import { NodejsFunction } from 'aws-cdk-lib/aws-lambda-nodejs';
   import { StateMachine } from 'aws-cdk-lib/aws-stepfunctions';
   
   const stack = new cdk.Stack(app, 'MyStack');
   
   const myLambda = new NodejsFunction(stack, 'MyLambda', {
     entry: './lambda/index.ts',
     runtime: cdk.aws_lambda.Runtime.NODEJS_20_X
   });
   
   // 读取并替换ASL内容
   const rawAsl = fs.readFileSync('./asl/state-machine-def.json', 'utf8');
   const resolvedAsl = rawAsl.replace('${LAMBDA_ARN}', myLambda.functionArn);
   
   // 创建状态机
   new StateMachine(stack, 'MyStateMachine', {
     definitionBody: cdk.aws_stepfunctions.DefinitionBody.fromString(resolvedAsl),
     stateMachineType: cdk.aws_stepfunctions.StateMachineType.STANDARD
   });
   
   // 手动添加Step Functions调用Lambda的权限
   myLambda.grantInvoke(new cdk.aws_iam.ServicePrincipal('states.amazonaws.com'));
   

方法2：用CDK Step Functions构造直接构建状态机定义 #

完全避免手动编写ASL文件，用CDK的原生构造来组装状态机，这样可以直接关联Lambda，CDK会自动处理权限和资源引用：

import * as cdk from 'aws-cdk-lib';
import { NodejsFunction } from 'aws-cdk-lib/aws-lambda-nodejs';
import { StateMachine } from 'aws-cdk-lib/aws-stepfunctions';
import { LambdaInvoke } from 'aws-cdk-lib/aws-stepfunctions-tasks';

const stack = new cdk.Stack(app, 'MyStack');

const myLambda = new NodejsFunction(stack, 'MyLambda', {
  entry: './lambda/index.ts',
  runtime: cdk.aws_lambda.Runtime.NODEJS_20_X
});

// 创建Lambda调用任务
const invokeLambdaTask = new LambdaInvoke(stack, 'InvokeMyLambda', {
  lambdaFunction: myLambda,
  payloadResponseOnly: true // 根据需求配置
});

// 构建并创建状态机
new StateMachine(stack, 'MyStateMachine', {
  definition: invokeLambdaTask,
  stateMachineType: cdk.aws_stepfunctions.StateMachineType.STANDARD
});

这种方式下，CDK会自动生成正确的ASL定义，并添加必要的IAM权限，无需手动处理资源ID。

方法3：如果必须使用物理ID（不推荐） #

若业务场景确实需要物理ID，可以通过CDK的Fn.ref获取Lambda的逻辑ID对应的物理资源ID，替换ASL中的占位符：

const resolvedAsl = rawAsl.replace('${LAMBDA_PHYSICAL_ID}', cdk.Fn.ref(myLambda.node.id));

不过这种方式不如ARN可靠，因为物理ID会随栈的重建或重命名变化，而ARN在资源生命周期内保持稳定。

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