如何用Lambda输出覆盖AWS Step Function单个输入字符串字段

你的问题源于当前配置中ResultSelector和ResultPath的搭配错误：ResultSelector生成了嵌套的{"token": "xxx"}对象，ResultPath又将这个对象赋值给了输入的token字段，最终导致嵌套结构。以下是两种可行的修复方案：

方案一：直接在Task状态中重构输出对象 #

适合需要保留多个原有字段的场景，通过ResultSelector重新构建完整的输出结构，替换指定字段：

修改后的Step Function定义：

{
  "Comment": "A Hello World example demonstrating various state types of the Amazon States Language",
  "StartAt": "ModifyInput",
  "States": {
    "ModifyInput": {
      "Type": "Task",
      "Resource": "arn:aws:states:::lambda:invoke",
      "ResultPath": "$",
      "ResultSelector": {
        "token.$": "$.Payload.token",
        "service_url.$": "$.Input.service_url"
      },
      "End": true,
      "TimeoutSeconds": 180,
      "Parameters": {
        "FunctionName": "arn:aws:lambda:us-west-2:639294609160:function:dummy-output",
        "Payload.$": "$"
      }
    }
  }
}

关键配置说明： #

• ResultPath: "$"：用Task处理后的结果完全替换当前输入
• ResultSelector：
  • token.$: "$.Payload.token"：提取Lambda返回的token字符串
  • service_url.$: "$.Input.service_url"：保留输入中原有的service_url字段

方案二：用Pass状态中转处理 #

适合仅替换单个字段的场景，拆分任务为「调用Lambda获取新值」和「替换字段」两步：

修改后的Step Function定义：

{
  "Comment": "A Hello World example demonstrating various state types of the Amazon States Language",
  "StartAt": "GetNewToken",
  "States": {
    "GetNewToken": {
      "Type": "Task",
      "Resource": "arn:aws:states:::lambda:invoke",
      "ResultPath": "$.new_token",
      "End": false,
      "TimeoutSeconds": 180,
      "Parameters": {
        "FunctionName": "arn:aws:lambda:us-west-2:639294609160:function:dummy-output",
        "Payload.$": "$"
      }
    },
    "ReplaceToken": {
      "Type": "Pass",
      "Parameters": {
        "token.$": "$.new_token.Payload.token",
        "service_url.$": "$.service_url"
      },
      "ResultPath": "$",
      "End": true
    }
  }
}

流程说明： #

1. GetNewToken：调用Lambda，将返回结果暂存到new_token字段
2. ReplaceToken：通过Pass状态重构输出对象，用Lambda返回的token替换原有字段，同时保留service_url

两种方案执行后，最终输出都会符合你的预期：

{
  "output": {
    "token": "1111-1111-1111-1111",
    "service_url": "http://www.test.com"
  },
  "outputDetails": {
    "truncated": false
  }
}

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