如何从AWS Lambda函数中调用Spring Boot批处理作业

第一步：适配依赖 #

在pom.xml（Maven）中添加AWS Lambda核心依赖，同时排除不需要的内置web容器减少包体积：

<dependency>
    <groupId>com.amazonaws</groupId>
    <artifactId>aws-lambda-java-core</artifactId>
    <version>1.2.21</version>
</dependency>
<dependency>
    <groupId>org.springframework.boot</groupId>
    <artifactId>spring-boot-starter-batch</artifactId>
    <exclusions>
        <exclusion>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-starter-tomcat</artifactId>
        </exclusion>
    </exclusions>
</dependency>

第二步：在Handler中初始化Spring上下文 #

Lambda冷启动时仅需要初始化一次Spring上下文，使用静态代码块实现上下文复用，避免每次触发都重新加载，你标注了@EnableBatchProcessing的配置类需要作为上下文的注册源：

import com.amazonaws.services.lambda.runtime.RequestHandler;
import com.amazonaws.services.lambda.runtime.Context;
import org.springframework.context.annotation.AnnotationConfigApplicationContext;
import org.springframework.batch.core.launch.JobLauncher;
import org.springframework.batch.core.Job;
import org.springframework.batch.core.JobParameters;
import org.springframework.batch.core.JobParametersBuilder;
import your.package.BatchConfig; // 替换为你自己的@EnableBatchProcessing类的全路径

public class BatchLambdaHandler implements RequestHandler<YourInputType, String> {
    // 静态初始化Spring上下文，同Lambda实例复用无需重复加载
    private static final AnnotationConfigApplicationContext springContext;
    private static final JobLauncher jobLauncher;
    private static final Job batchJob;

    static {
        springContext = new AnnotationConfigApplicationContext(BatchConfig.class);
        jobLauncher = springContext.getBean(JobLauncher.class);
        // 替换为你在BatchConfig中定义的Job Bean名称
        batchJob = (Job) springContext.getBean("yourBatchJobName");
    }

    @Override
    public String handleRequest(YourInputType input, Context lambdaContext) {
        // 后续作业执行逻辑写在这里
    }
}

注：YourInputType替换为你自己定义的Lambda入参类型，纯字符串入参可直接改为String。

第三步：触发Batch作业执行 #

在handleRequest方法中，把Lambda入参转换为Batch作业的运行参数，触发作业执行：

@Override
public String handleRequest(YourInputType input, Context lambdaContext) {
    try {
        // 构造Job参数，可自由传入Lambda请求ID、业务自定义参数
        JobParameters jobParameters = new JobParametersBuilder()
                .addString("requestId", lambdaContext.getAwsRequestId())
                .addString("customParam", input.getCustomParam()) // 从入参取业务参数
                .addLong("timestamp", System.currentTimeMillis())
                .toJobParameters();
        
        // 执行批处理作业
        jobLauncher.run(batchJob, jobParameters);
        return "作业执行成功";
    } catch (Exception e) {
        lambdaContext.getLogger().log("作业执行失败：" + e.getMessage());
        return "作业执行失败：" + e.getMessage();
    }
}

注意事项 #

• 作业时长限制：AWS Lambda最长执行时间为15分钟，若你的批处理作业执行时长超过该阈值，需要拆分作业为多个小任务分批执行
• 元数据存储：如果你的Spring Boot Batch使用JDBC存储作业元数据，需要配置Lambda所属VPC可访问对应的数据库实例，同时配置好数据库账号权限
• 冷启动优化：打包时可使用spring-boot-thin-layout插件减少jar包体积，提升冷启动速度
• 幂等性处理：Lambda存在重试机制，需要确保你的批处理作业支持幂等执行，避免重复处理数据

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