AWS SQS死信队列重驱动后ApproximateReceiveCount未延续计数问题

为什么ApproximateReceiveCount会重置为1？ #

SQS的ApproximateReceiveCount是队列级别的属性，仅统计消息在当前队列内被接收的次数。当你把死信队列中的消息移回原队列时，原队列会将该消息视为全新的入队消息，因此会重置ApproximateReceiveCount为1，这是SQS的默认行为，和是否新增重驱动Lambda无关。

解决办法 #

1. 通过自定义消息属性传递原重试次数 #

在重驱动死信队列的Lambda中，读取消息在死信队列中的ApproximateReceiveCount（这个值等于消息在原队列触发死信时的接收次数，即3），将其作为自定义属性附加到发送回原队列的消息中。原队列的处理Lambda则结合当前队列的接收次数，计算总重试次数，判断是否需要直接移入死信。

重驱动Lambda示例（Python）：

import boto3

sqs_client = boto3.client('sqs')
SOURCE_DLQ_URL = "你的死信队列URL"
TARGET_QUEUE_URL = "你的原队列URL"

def lambda_handler(event, context):
    for record in event['Records']:
        # 获取消息在原队列的累计接收次数
        original_retry_count = int(record['attributes']['ApproximateReceiveCount'])
        message_body = record['body']
        # 构造包含原重试次数的自定义属性
        custom_attrs = {
            "OriginalRetryCount": {
                "DataType": "Number",
                "StringValue": str(original_retry_count)
            }
        }
        # 合并原消息的属性（如果有）
        if record.get('messageAttributes'):
            custom_attrs.update(record['messageAttributes'])
        
        # 发送消息回原队列
        sqs_client.send_message(
            QueueUrl=TARGET_QUEUE_URL,
            MessageBody=message_body,
            MessageAttributes=custom_attrs
        )
        # 删除死信队列中的已处理消息
        sqs_client.delete_message(
            QueueUrl=SOURCE_DLQ_URL,
            ReceiptHandle=record['receiptHandle']
        )

原队列处理Lambda修改：

def lambda_handler(event, context):
    for record in event['Records']:
        current_receive_count = int(record['attributes']['ApproximateReceiveCount'])
        # 获取自定义属性中的原重试次数，默认0
        original_retry_count = int(
            record['messageAttributes'].get('OriginalRetryCount', {}).get('StringValue', '0')
        )
        # 计算总重试次数：原队列累计次数 + 当前队列接收次数
        total_retry_count = original_retry_count + current_receive_count
        
        # 超过原重试限制（3次）则直接抛出异常，触发死信
        if total_retry_count > 3:
            raise Exception(f"Total retry count {total_retry_count} exceeds threshold")
        
        # 正常执行数据处理与上传逻辑
        process_data_and_upload(record['body'])

2. 临时调整原队列的重试策略（快速临时方案） #

如果只是一次性重驱动死信消息，可以临时修改原队列的MaximumReceives参数为1：

• 这样消息回到原队列后，若处理失败，仅重试1次就会移入死信队列，避免累计过多重试次数。
• 待所有死信消息处理完成后，再将MaximumReceives改回3。

3. 直接在死信队列处理消息（跳过移回步骤） #

如果业务允许，也可以直接修改重驱动Lambda的逻辑，在死信队列中完成数据处理与上传，无需将消息移回原队列，从根源上避免计数重置问题。

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