通过AWS Step Functions调用Redshift存储过程/查询未等待完成即触发后续任务

场景1：未启用Wait for Callback时，任务提前触发后续步骤 #

默认情况下，Step Functions调用Redshift的InvokeStatement API时，只会确认语句提交成功就返回，不会等待Redshift侧执行完成，这就是后续任务被提前触发的核心原因。

解决办法：改用Step Functions轮询机制跟踪语句执行状态：

• 调用InvokeStatement提交查询/存储过程，保存返回的Id（语句唯一标识）
• 添加Wait步骤，设置合理等待间隔（比如30秒，根据任务实际耗时调整）
• 添加Task步骤调用DescribeStatement API，传入之前保存的语句ID
• 通过Choice步骤判断返回的Status字段：
  • 若为FINISHED，进入后续业务步骤
  • 若为FAILED/ABORTED，跳转至错误处理分支
  • 若为PICKED/STARTED，回到Wait步骤继续轮询

场景2：启用Wait for Callback后，任务持续挂起需手动终止 #

启用Wait for Callback时，Step Functions会等待外部系统发送SendTaskSuccess/SendTaskFailure回调来结束当前步骤，但Redshift本身不会主动触发该回调，导致步骤一直处于等待状态。

解决办法：结合Lambda函数实现回调触发逻辑：

• 在Step Functions的Redshift任务中启用Wait for Callback，记录生成的taskToken
• 提交Redshift语句后，将taskToken和语句ID传入Lambda函数
• Lambda函数内部实现轮询逻辑（同场景1），当检测到Redshift语句执行完成（成功或失败）时，调用SendTaskSuccess或SendTaskFailure API，传入对应的taskToken来结束Step Functions的等待步骤

关键注意事项 #

• 合理设置轮询间隔和任务超时时间，避免无效API调用或无限等待
• 必须处理Redshift语句执行失败的场景，确保Step Functions能正确进入错误分支
• 对于长时间运行的存储过程，需同步设置Redshift语句超时和Step Functions任务超时，避免资源占用

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