如何让AWS Step Functions等待RDS删除完成后返回taskToken？

核心思路是利用AWS的事件通知机制，监听RDS实例删除完成的事件，触发Lambda调用Step Functions的SendTaskSuccess API来返回taskToken。以下是具体实现步骤：

方法一：通过EventBridge监听RDS删除完成事件 #

1. 初始Lambda逻辑
   这个Lambda负责发起RDS删除请求，同时把Step Functions传入的taskToken和目标RDS实例ID存储到临时存储（比如DynamoDB），方便后续事件触发的Lambda获取。示例代码：

   import boto3
   import uuid
   
   rds = boto3.client('rds')
   dynamodb = boto3.resource('dynamodb')
   table = dynamodb.Table('RDSDeletionTasks')
   
   def lambda_handler(event, context):
       # 获取Step Functions传入的taskToken
       task_token = event['TaskToken']
       rds_instance_id = 'qa-rds-instance'  # 替换为你的QA RDS实例ID
   
       # 发起异步RDS删除请求
       rds.delete_db_instance(
           DBInstanceIdentifier=rds_instance_id,
           SkipFinalSnapshot=True  # 根据需求决定是否保留最终快照
       )
   
       # 将taskToken与实例ID关联存入DynamoDB
       table.put_item(
           Item={
               'InstanceId': rds_instance_id,
               'TaskToken': task_token,
               'TaskId': str(uuid.uuid4())
           }
       )
   
       return None
   

2. 配置EventBridge规则
   创建EventBridge规则，精准监听RDS实例删除完成的事件：

   • 事件源选择AWS RDS
   • 事件模式设置为：

     {
       "source": ["aws.rds"],
       "detail-type": ["RDS DB Instance Event"],
       "detail": {
         "EventCategories": ["deletion"],
         "EventID": ["RDS-EVENT-0076"],  # 对应实例删除完成的固定事件ID
         "DBInstanceIdentifier": ["qa-rds-instance"]  # 你的QA RDS实例ID
       }
     }
     

   • 目标选择一个新的Lambda函数，用于发送taskToken给Step Functions。

3. 触发Lambda发送taskToken
   这个Lambda从DynamoDB中根据RDS实例ID取出对应的taskToken，调用Step Functions的SendTaskSuccess API完成信号返回：

   import boto3
   
   sfn = boto3.client('stepfunctions')
   dynamodb = boto3.resource('dynamodb')
   table = dynamodb.Table('RDSDeletionTasks')
   
   def lambda_handler(event, context):
       rds_instance_id = event['detail']['DBInstanceIdentifier']
   
       # 从DynamoDB获取关联的taskToken
       response = table.get_item(Key={'InstanceId': rds_instance_id})
       task_token = response['Item']['TaskToken']
   
       # 向Step Functions发送删除完成信号
       sfn.send_task_success(
           taskToken=task_token,
           output='{"status": "RDS deleted successfully"}'
       )
   
       # 清理DynamoDB中的临时记录
       table.delete_item(Key={'InstanceId': rds_instance_id})
   
       return {'statusCode': 200}
   

方法二：利用RDS事件通知直接触发Lambda #

除了EventBridge，也可以直接配置RDS的事件通知链路：

• 进入RDS控制台，找到目标QA实例，配置事件通知，选择deletion类事件，发送到指定SNS主题。
• 给SNS主题添加Lambda订阅，该Lambda的逻辑与方法一中的触发Lambda一致：从存储中取出taskToken，调用SendTaskSuccess。

关键注意事项 #

• 权限配置：确保初始Lambda拥有RDS删除权限和DynamoDB写入权限；触发Lambda拥有DynamoDB读取权限和Step Functions的SendTaskSuccess权限。
• 错误处理：需额外监听RDS删除失败事件，调用SendTaskFailure通知Step Functions，避免工作流无限等待。
• taskToken存储：除DynamoDB外，也可使用S3存储，但不建议将taskToken作为RDS标签存储（删除过程中标签可能无法读取）。

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