如何在带ItemSelector的Map状态中为lambda:invoke.waitForTaskToken传入TaskToken？

问题 #

我正在使用AWS Step Functions遍历输入文档中的列表，每次迭代需要调用外部服务。希望遍历每个条目并执行使用lambda:invoke.waitForTaskToken的步骤，同时将TaskToken传入每次迭代的执行中。

核心问题是：如何同时在Map状态层级使用ItemSelector（需要保留Map状态输入的信息），又能在内部步骤中注入TaskToken。

AWS文档有两条关键说明：

ItemSelector字段替代了Map状态内的Parameters字段。如果您在Map状态定义中使用Parameters字段创建自定义输入，强烈建议替换为ItemSelector。

在执行期间，上下文对象会为访问它的Parameters字段填充相关数据。如果Parameters字段在任务状态之外，Task字段的值为null。

这两条说明似乎暗示想要实现的功能无法完成。

我期望的配置如下：

{
  "StartAt": "ExampleMapState",
  "States": {
    "ExampleMapState": {
      "Type": "Map",
      "ItemsPath": "$.items",
      "ItemSelector": {
        "dynamic.$": "$.dynamic",
        "ContextIndex.$": "$$.Map.Item.Index",
        "ContextValue.$": "$$.Map.Item.Value"
      },
      "ItemProcessor": {
        "ProcessorConfig": {
          "Mode": "INLINE"
        },
        "StartAt": "TestPass",
        "States": {
          "TestPass": {
            "Type": "Task",
            "Parameters": {
              "FunctionName": "arn:aws:lambda:us-west-2:123456789012:function:echo-lambda",
              "Payload": {
                "item.$": "$",
                "token.$": "$$.Task.Token"
              }
            },
            "Resource": "arn:aws:states:::lambda:invoke.waitForTaskToken",
            "End": true
          }
        }
      },
      "End": true
    }
  }
}

但此配置无法生效，因为ItemSelector会覆盖内部TestPass状态的Payload。

补充：尝试过将$$.Task.Token放入ItemSelector中，但出现错误，因为该层级的上下文对象中不存在$$.Task。无效配置示例：

{
    "StartAt": "ExampleMapState",
    "States": {
      "ExampleMapState": {
        "Type": "Map",
        "ItemsPath": "$.items",
        "ItemSelector": {
          "dynamic.$": "$.dynamic",
          "ContextIndex.$": "$$.Map.Item.Index",
          "ContextValue.$": "$$.Map.Item.Value",
          "token.$": "$$.Task.Token"
        },
        "ItemProcessor": {
          "ProcessorConfig": {
            "Mode": "INLINE"
          },
          "StartAt": "TestPass",
          "States": {
            "TestPass": {
              "Type": "Task",
              "Parameters": {
                "FunctionName": "arn:aws:lambda:us-west-2:123456789012:function:echo-lambda"
              },
              "Resource": "arn:aws:states:::lambda:invoke.waitForTaskToken",
              "End": true
            }
          }
        },
        "End": true
      }
    }
  }

解决方法 #

核心思路是在Task状态的Parameters里直接构造完整的Payload，同时引用ItemSelector处理后的迭代输入，以及当前Task的Token。不需要让ItemSelector处理Token，而是在Task内部的Parameters中完成拼接。

可行的配置如下：

{
  "StartAt": "ExampleMapState",
  "States": {
    "ExampleMapState": {
      "Type": "Map",
      "ItemsPath": "$.items",
      "ItemSelector": {
        "dynamic.$": "$.dynamic",
        "index.$": "$$.Map.Item.Index",
        "value.$": "$$.Map.Item.Value"
      },
      "ItemProcessor": {
        "ProcessorConfig": {
          "Mode": "INLINE"
        },
        "StartAt": "InvokeLambdaWithToken",
        "States": {
          "InvokeLambdaWithToken": {
            "Type": "Task",
            "Resource": "arn:aws:states:::lambda:invoke.waitForTaskToken",
            "Parameters": {
              "FunctionName": "arn:aws:lambda:us-west-2:123456789012:function:echo-lambda",
              "Payload": {
                "dynamic.$": "$.dynamic",
                "index.$": "$.index",
                "value.$": "$.value",
                "taskToken.$": "$$.Task.Token"
              }
            },
            "End": true
          }
        }
      },
      "End": true
    }
  }
}

配置说明 #

• ItemSelector保留原有功能：负责将Map层级的共享数据（dynamic）和迭代上下文（索引、条目值）整理成每个迭代的输入。
• Task状态内拼接Payload：在Task的Parameters里，直接从ItemSelector生成的迭代输入中引用字段，同时注入$$.Task.Token。这样既保留了ItemSelector的优势，又能正确获取到当前Task的Token（因为Task层级的上下文才包含$$.Task对象）。
• 避免覆盖问题：这种方式没有让ItemSelector干预Task的Payload构造，而是在Task内部完成所有参数的组装，完全符合AWS Step Functions的上下文规则。

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