如何查找触发Lambda函数的SQS队列名称并实现分表写入

没问题，这个场景非常典型，我来帮你实现区分SQS来源并动态切换DynamoDB表的需求。

核心思路 #

当SQS触发Lambda时，Lambda接收的事件对象里已经包含了触发队列的ARN，我们只需要从ARN中提取队列名称，再根据名称来切换对应的DDB表即可。

具体步骤与代码示例 #

假设你的Lambda用Python编写，实现逻辑如下：

1. 从事件中提取队列名称
   SQS触发的Lambda事件结构里，每条记录的eventSourceARN字段就是队列的ARN，格式类似arn:aws:sqs:us-east-1:123456789012:ABC。我们可以通过分割这个字符串来获取最后的队列名称：

import os
import boto3

def lambda_handler(event, context):
    # 从事件中获取触发队列的ARN（批量消息时所有记录来自同一个队列，取第一个即可）
    sqs_arn = event['Records'][0]['eventSourceARN']
    # 分割ARN获取队列名称
    queue_name = sqs_arn.split(':')[-1]
    
    # 初始化DDB客户端
    dynamodb = boto3.resource('dynamodb')
    
    # 根据队列名称选择目标表
    default_table = os.environ['DDB_TABLE_1']
    if queue_name == 'XYZ':
        target_table = os.environ['DDB_TABLE_2']  # 建议把DDB2也设为环境变量，更灵活
    else:
        target_table = default_table
    
    # 处理消息并写入对应表
    table = dynamodb.Table(target_table)
    for record in event['Records']:
        message_body = record['body']
        # 这里写入你的DDB业务逻辑，示例：
        table.put_item(Item={'message_content': message_body, 'source_queue': queue_name})
    
    return {
        'statusCode': 200,
        'body': f'Processed {len(event["Records"])} messages from queue {queue_name}, wrote to table {target_table}'
    }

2. 扩展性优化建议
   如果后续可能新增更多队列和对应表，用字典映射的方式会更易维护：

# 提前定义队列与表的映射关系
queue_table_map = {
    'ABC': os.environ['DDB_TABLE_1'],
    'XYZ': os.environ['DDB_TABLE_2']
}
# 根据队列名获取目标表，默认用DDB1兜底
target_table = queue_table_map.get(queue_name, os.environ['DDB_TABLE_1'])

注意事项 #

• 批量消息场景：SQS批量发送消息到Lambda时，所有记录都来自同一个队列，所以只需要取第一条记录的ARN即可，无需遍历所有记录。
• 权限配置：确保Lambda的执行角色拥有对应DynamoDB表的读写权限，以及SQS的触发权限（如果已经配置过触发规则，这部分通常已经搞定）。

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