启用AWS Lambda SnapStart后Java代码无法加载AWS凭证问题

核心原因 #

启用SnapStart后，Lambda会在首次执行时创建运行环境的内存快照，后续请求直接恢复快照运行。你的问题本质是AWS SDK v1客户端的凭证生命周期与SnapStart快照机制冲突：

• Lambda执行角色的临时凭证是短期有效（默认15分钟），快照会捕获首次运行时SDK客户端加载的凭证；当后续请求恢复快照时，这些凭证已经过期。
• 即使你在handler内创建客户端，旧版SDK v1的凭证提供者在快照恢复后，可能无法正确读取Lambda运行环境更新后的临时凭证（快照保留了首次运行时的凭证读取状态）。

具体解决方法 #

1. 在afterRestore中重新初始化SQS客户端

将SQS客户端改为类级变量，在快照检查点前关闭客户端释放资源，恢复后重新构建以获取最新凭证：

// 声明为类级变量
private AmazonSQS sqs;

@Override
public void beforeCheckpoint(org.crac.Context<? extends Resource> context) throws Exception {
    System.out.println("Before Checkpoint");
    // 检查点前关闭客户端，避免快照保留无效连接
    if (sqs != null) {
        sqs.shutdown();
    }
}

@Override
public void afterRestore(org.crac.Context<? extends Resource> context) throws Exception {
    System.out.println("After Restore");
    // 恢复后重新初始化客户端，获取最新有效凭证
    sqs = AmazonSQSClientBuilder.standard().withRegion(Regions.EU_WEST_1).build();
}

// 在handler方法中使用该客户端
public void handleRequest(InputStream input, OutputStream output, Context context) throws IOException {
    final Map<String, MessageAttributeValue> messageAttributes = new HashMap<>();
    messageAttributes.put("testID", new MessageAttributeValue().withStringValue("id123").withDataType("String"));
    
    final SendMessageRequest sendMessageStandardQueue = new SendMessageRequest()
            .withQueueUrl("https://sqs.eu-west-1.amazonaws.com/111111/my-queue")
            .withMessageBody("Test message from lambda")
            .withMessageAttributes(messageAttributes);
            
    sqs.sendMessage(sendMessageStandardQueue);
}

2. 每次请求都在handler内创建新客户端

如果不想维护类级变量，可直接在handler方法内每次请求都创建新客户端，确保每次都读取最新凭证（性能略低，但能快速解决问题）：

public void handleRequest(InputStream input, OutputStream output, Context context) throws IOException {
    // 每次请求重新构建客户端
    final AmazonSQS sqs = AmazonSQSClientBuilder.standard().withRegion(Regions.EU_WEST_1).build();
    
    final Map<String, MessageAttributeValue> messageAttributes = new HashMap<>();
    messageAttributes.put("testID", new MessageAttributeValue().withStringValue("id123").withDataType("String"));
    
    final SendMessageRequest sendMessageStandardQueue = new SendMessageRequest()
            .withQueueUrl("https://sqs.eu-west-1.amazonaws.com/111111/my-queue")
            .withMessageBody("Test message from lambda")
            .withMessageAttributes(messageAttributes);
            
    sqs.sendMessage(sendMessageStandardQueue);
    // 请求完成后关闭客户端
    sqs.shutdown();
}

3. 迁移到AWS SDK for Java v2（推荐）

AWS SDK v2原生支持Lambda SnapStart，内置了快照恢复后的凭证自动刷新逻辑，无需手动处理客户端初始化：

import software.amazon.awssdk.regions.Region;
import software.amazon.awssdk.services.sqs.SqsClient;
import software.amazon.awssdk.services.sqs.model.MessageAttributeValue;
import software.amazon.awssdk.services.sqs.model.SendMessageRequest;

import java.util.HashMap;
import java.util.Map;

public class SqsLambdaHandler implements RequestHandler<Object, Object> {
    // SDK v2客户端可安全声明为静态变量，恢复后自动刷新凭证
    private static final SqsClient sqsClient = SqsClient.builder()
            .region(Region.EU_WEST_1)
            .build();

    @Override
    public Object handleRequest(Object input, Context context) {
        Map<String, MessageAttributeValue> messageAttributes = new HashMap<>();
        messageAttributes.put("testID", MessageAttributeValue.builder()
                .stringValue("id123")
                .dataType("String")
                .build());

        SendMessageRequest request = SendMessageRequest.builder()
                .queueUrl("https://sqs.eu-west-1.amazonaws.com/111111/my-queue")
                .messageBody("Test message from lambda")
                .messageAttributes(messageAttributes)
                .build();

        sqsClient.sendMessage(request);
        return null;
    }
}

关键注意点 #

• 禁止在静态初始化块、类构造函数中创建SDK v1客户端，这些代码仅在首次运行时执行，客户端会被快照捕获，恢复后使用过期凭证。
• 若坚持使用SDK v1，必须在快照恢复后重新初始化客户端，确保获取最新临时凭证。

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