如何判断AWS Lambda函数是否正在执行重试？

AWS Lambda本身没有内置的context.retry字段，要实现重试状态检测，需通过以下自定义方案实现：

• 基于请求ID的外部计数法
  Lambda对同一次请求的重试会复用同一个context.requestId。可以用DynamoDB或内存缓存（如ElastiCache）存储每个requestId的执行次数：

  1. 函数启动时，先查询存储中该requestId的计数
  2. 若计数为0，标记为首次执行，执行后将计数加1
  3. 若计数大于0，即可判定为重试执行
     示例代码（DynamoDB）：

  const AWS = require('aws-sdk');
  const dynamodb = new AWS.DynamoDB.DocumentClient();
  
  exports.handler = async (event, context) => {
    const requestId = context.requestId;
    const params = {
      TableName: 'LambdaRetryTracking',
      Key: { requestId },
      UpdateExpression: 'SET executionCount = if_not_exists(executionCount, :zero) + :one',
      ExpressionAttributeValues: { ':zero': 0, ':one': 1 },
      ReturnValues: 'UPDATED_NEW'
    };
  
    const result = await dynamodb.update(params).promise();
    const isRetry = result.Attributes.executionCount > 1;
  
    if (isRetry) {
      console.log('This is a retry execution');
      // 重试逻辑处理
    }
  
    // 业务逻辑
    return { statusCode: 200 };
  };
  

• 触发源层面注入重试标记
  如果Lambda由SQS、SNS、EventBridge等服务触发，可在这些服务的消息中添加自定义重试标记：

  • 比如SQS消息添加retry-count消息属性，每次重试时递增该值
  • Lambda函数中读取消息属性，若retry-count大于0则判定为重试
    示例（SQS触发的Lambda读取消息属性）：

  exports.handler = (event, context) => {
    const record = event.Records[0];
    const retryCount = record.messageAttributes?.['retry-count']?.stringValue;
    const isRetry = retryCount ? parseInt(retryCount) > 0 : false;
  
    if (isRetry) {
      // 处理重试逻辑
    }
  };
  

• 自定义调用时传递重试元数据
  如果是通过AWS SDK手动调用Lambda，可通过clientContext传递重试标记：
  调用时的代码：

  const AWS = require('aws-sdk');
  const lambda = new AWS.Lambda();
  
  lambda.invoke({
    FunctionName: 'your-function-name',
    Payload: JSON.stringify(event),
    ClientContext: Buffer.from(JSON.stringify({ isRetry: true })).toString('base64')
  }).promise();
  

  Lambda handler中解析：

  exports.handler = (event, context) => {
    let isRetry = false;
    if (context.clientContext) {
      const clientContext = JSON.parse(Buffer.from(context.clientContext, 'base64').toString());
      isRetry = clientContext.isRetry;
    }
  
    if (isRetry) {
      // 重试逻辑
    }
  };
  

• 基于业务幂等键的状态判断
  如果业务请求中有唯一幂等键（比如订单ID、交易ID），可在外部存储记录该键的执行状态：

  1. 首次执行时，将幂等键标记为「执行中」
  2. 再次收到相同幂等键的请求时，若状态仍为「执行中」，则判定为重试
     这种方案同时能实现业务幂等，避免重复处理。

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