在 Step Functions 中使用 Lambda 函数迭代循环

登录 Lambda 控制台，然后选择创建函数。

在创建函数页面上，选择从头开始创作。

在基本信息部分中，配置您的 Lambda 函数：

1. 对于函数名称，请输入 Iterator。

2. 对于 Runtime (运行时)，选择 Node.js。

3. 在更改默认执行角色中，选择创建具有基本 Lambda 权限的新角色。

4. 选择创建函数。

将 Lambda 函数的以下代码复制到代码源中。

export const handler = function (event, context, callback) {
  let index = event.iterator.index
  let step = event.iterator.step
  let count = event.iterator.count
 
  index = index + step
 
  callback(null, {
    index,
    step,
    count,
    continue: index < count
  })
}

此代码接受 count、index 和 step 的输入值。它将 index 递增 step 的值，并返回这些值及布尔值 continue。如果 index 小于 count，continue 的值为 true。

选择部署。

选择测试。

在 配置测试事件对话框中，在事件名称框中输入 TestIterator。

使用以下内容替换示例数据。

{
  "Comment": "Test my Iterator function",
  "iterator": {
    "count": 10,
    "index": 5,
    "step": 1
  }
}

这些值模拟在迭代期间来自状态机的内容。Lambda 函数将递增索引，当索引小于 count 时，将对于 continue 返回 true。在此测试中，索引已增加到 5。测试将 index 递增到 6，并将 continue 设置为 true。

选择创建。

选择测试来测试您的 Lambda 函数。

测试结果显示在执行结果选项卡中。

要查看执行结果，请选择输出选项卡。

{
  "index": 6,
  "step": 1,
  "count": 10,
  "continue": true
}

打开 Step Functions 控制台，然后选择创建状态机。

选择空白模板。

在代码窗格中，粘贴以下用于定义状态机的 JSON。

有关 Amazon States Language 的更多信息，请参阅状态机结构。

{
    "Comment": "Iterator State Machine Example",
    "StartAt": "ConfigureCount",
    "States": {
        
        "ConfigureCount": {
            "Type": "Pass",
            "Result": {
                "count": 10,
                "index": 0,
                "step": 1
            },
            "ResultPath": "$.iterator",
            "Next": "Iterator"
        },
        "Iterator": {
            "Type": "Task",
            "Resource": "arn:aws:lambda:region:123456789012:function:Iterate",
            "ResultPath": "$.iterator",
            "Next": "IsCountReached"
        },
        "IsCountReached": {
            "Type": "Choice",
            "Choices": [
                {
                    "Variable": "$.iterator.continue",
                    "BooleanEquals": true,
                    "Next": "ExampleWork"
                }
            ],
            "Default": "Done"
        },
        "ExampleWork": {
            "Comment": "Your application logic, to run a specific number of times",
            "Type": "Pass",
            "Result": {
              "success": true
            },
            "ResultPath": "$.result",
            "Next": "Iterator"
        },
        "Done": {
            "Type": "Pass",
            "End": true
          
        }
    }
}

将 Iterator Resource 字段替换为您之前创建的 Iterator Lambda 函数的 ARN。

选择配置，然后为状态机输入名称，例如 IterateCount。

对于类型，接受默认值标准。对于权限，选择创建新角色。

选择创建，然后确认角色创建。

在IterateCount页面上，选择开始执行。

（可选）输入自定义执行名称，以便覆盖生成的默认执行名称。

选择启动执行。

此时将开始状态机的新执行，并显示正在运行的执行。

执行逐步递增，使用您的 Lambda 函数跟踪计数。在每次迭代时，它都会执行在状态机中处于 ExampleWork 状态引用的示例工作。

计数达到您的状态机中 ConfigureCount 状态中指定的数值后，执行停止迭代并结束。