如何使用AWS Java SDK v2更新Lambda响应S3事件的代码片段？

问题分析 #

原代码存在以下核心问题：

• 依赖版本混乱：混用了AWS SDK v1的aws-sdk-java包和v2的S3客户端API，Lambda事件模型也用了v1的com.amazonaws类
• S3客户端未复用：每次请求新建S3Client实例，会导致不必要的资源开销和连接泄漏
• 错误处理粗糙：直接抛出RuntimeException，缺乏针对性的日志和异常处理
• 日志写法不规范：字符串拼接不符合SLF4J最佳实践
• 未做空值校验：直接取第一个事件记录，未处理空列表场景

修复步骤 #

1. 更新Maven依赖 #

替换原依赖为AWS SDK v2的官方依赖，适配JDK 21：

<dependencies>
    <!-- AWS Lambda运行时依赖 -->
    <dependency>
        <groupId>software.amazon.awssdk</groupId>
        <artifactId>lambda</artifactId>
        <version>2.23.18</version>
        <scope>provided</scope>
    </dependency>
    <!-- S3客户端依赖 -->
    <dependency>
        <groupId>software.amazon.awssdk</groupId>
        <artifactId>s3</artifactId>
        <version>2.23.18</version>
    </dependency>
    <!-- Lambda S3事件模型依赖 -->
    <dependency>
        <groupId>software.amazon.awssdk</groupId>
        <artifactId>lambda-events</artifactId>
        <version>3.1.0</version>
    </dependency>
    <!-- SLF4J日志API -->
    <dependency>
        <groupId>org.slf4j</groupId>
        <artifactId>slf4j-api</artifactId>
        <version>2.0.9</version>
    </dependency>
    <!-- Lambda兼容的日志实现 -->
    <dependency>
        <groupId>com.amazonaws</groupId>
        <artifactId>aws-lambda-java-log4j2</artifactId>
        <version>1.5.1</version>
    </dependency>
</dependencies>

2. 修复后的代码 #

import software.amazon.awssdk.services.s3.S3Client;
import software.amazon.awssdk.services.s3.model.HeadObjectRequest;
import software.amazon.awssdk.services.s3.model.HeadObjectResponse;
import software.amazon.awssdk.services.s3.model.S3Exception;
import software.amazon.awssdk.services.lambda.runtime.Context;
import software.amazon.awssdk.services.lambda.runtime.RequestHandler;
import software.amazon.awssdk.services.lambda.runtime.events.S3Event;
import software.amazon.awssdk.services.lambda.runtime.events.S3Event.S3EventRecord;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

public class Handler implements RequestHandler<S3Event, String> {
    // 静态初始化S3Client，复用实例
    private static final S3Client S3_CLIENT = S3Client.create();
    private static final Logger logger = LoggerFactory.getLogger(Handler.class);

    @Override
    public String handleRequest(S3Event s3Event, Context context) {
        // 校验事件记录是否为空
        if (s3Event.records() == null || s3Event.records().isEmpty()) {
            logger.warn("Received empty S3 event records");
            return "No records processed";
        }

        try {
            S3EventRecord record = s3Event.records().get(0);
            String srcBucket = record.s3().bucket().name();
            String srcKey = record.s3().object().urlDecodedKey();

            HeadObjectResponse headObject = getHeadObject(srcBucket, srcKey);
            // 使用SLF4J参数化日志，避免字符串拼接
            logger.info("Successfully retrieved object {} of type {}", 
                        srcBucket + "/" + srcKey, headObject.contentType());

            return "Ok";
        } catch (S3Exception e) {
            logger.error("S3 operation failed: {}", e.awsErrorDetails().errorMessage(), e);
            throw new RuntimeException("Failed to process S3 object", e);
        } catch (Exception e) {
            logger.error("Unexpected error processing S3 event", e);
            throw new RuntimeException("Unexpected error", e);
        }
    }

    private HeadObjectResponse getHeadObject(String bucket, String key) {
        HeadObjectRequest request = HeadObjectRequest.builder()
                .bucket(bucket)
                .key(key)
                .build();
        return S3_CLIENT.headObject(request);
    }
}

3. 关键修复点说明 #

• 客户端复用：将S3Client改为静态实例，避免每次请求新建连接，符合Lambda的执行环境最佳实践
• 依赖统一：全部使用AWS SDK v2的包，移除v1的com.amazonaws相关导入，避免版本冲突
• 空值校验：增加事件记录为空的判断，避免IndexOutOfBoundsException
• 针对性异常处理：捕获S3Exception并打印AWS错误详情，便于排查权限、对象不存在等问题
• 规范日志：使用SLF4J参数化日志，提升性能并避免潜在的字符串拼接问题
• Lambda事件模型适配：使用v2的software.amazon.awssdk.services.lambda.runtime.events.S3Event类，替换v1的旧模型

内容的提问来源于stack exchange，提问作者Đỗ Như Vỹ