Spring Boot部署AWS Lambda初始化超时及SPRINGBOOT2_COLD_START异常

问题场景 #

尝试将Spring Boot应用部署至AWS Lambda，采用StreamLambdaHandler作为请求处理类，初始实现代码如下：

public class StreamLambdaHandler implements RequestStreamHandler {

    private static SpringBootLambdaContainerHandler<AwsProxyRequest, AwsProxyResponse> handler;

    static {
        try {
            //handler = SpringBootLambdaContainerHandler.getAwsProxyHandler(SituationalFlexibilityApp.class);

            // 启动耗时超过10秒的应用，可使用异步构建器：

             handler = new SpringBootProxyHandlerBuilder<AwsProxyRequest>()
                            .defaultProxy()
                            .asyncInit()
                            .springBootApplication(SituationalFlexibilityApp.class)
                            .buildAndInitialize();

            // 通过handler的onStartup方法注册自定义过滤器
            handler.onStartup(servletContext -> {
                FilterRegistration.Dynamic registration = servletContext.addFilter("CognitoIdentityFilter",CognitoIdentityFilter.class);
                registration.addMappingForUrlPatterns(EnumSet.of(DispatcherType.REQUEST), true, "/*");
            });
        } catch (ContainerInitializationException e) {
            // 若此处初始化失败，将重抛异常以触发新的冷启动
            e.printStackTrace();
            throw new RuntimeException("Could not initialize Spring Boot application", e);
        }
    }

    public StreamLambdaHandler() {
        Timer.enable();
    }
 

    /*
     * public StreamLambdaHandler() throws ContainerInitializationException {
     * 
     * handler = new SpringBootProxyHandlerBuilder() .defaultProxy() .asyncInit()
     * .springBootApplication(SlowApplication.class) .buildAndInitialize(); }
     */

    @Override
    public void handleRequest(InputStream input, OutputStream output, Context context) throws IOException {
        handler.proxyStream(input, output, context);

    }
}

报错演进过程 #

• 首次测试触发初始化超时异常：

  com.amazonaws.serverless.exceptions.ContainerInitializationException: Could not initialize framework within the 20000ms timeout
  

• 临时调整操作：将Lambda函数超时配置修改为5分钟，同时在静态代码块中添加初始化超时配置：

  LambdaContainerHandler.getContainerConfig().setInitializationTimeout(2000000);
  

• 调整后重新测试触发新异常：

  Exception in thread "Thread-0" java.lang.IllegalArgumentException: Could not find timer SPRINGBOOT2_COLD_START
  

根因说明 #

该异常是aws-serverless-java-container组件使用方式错误导致，核心问题有两点：

• Timer.enable()调用时机错误：构造函数中才启用计时器，但使用asyncInit()异步初始化模式时，Spring容器冷启动计时逻辑在静态代码块执行buildAndInitialize()阶段就已经触发，此时计时器未启用，框架找不到对应冷启动计时器实例就会抛出该异常。
• 过滤器注册逻辑位置错误：buildAndInitialize()执行完成后，Servlet上下文初始化流程已经结束，后续再调用handler.onStartup()注册过滤器不仅不会生效，还可能干扰容器正常初始化流程。

修复方案 #

按以下步骤调整代码即可解决问题：

1. 将Timer.enable()调用移到静态代码块最开头，在所有handler初始化逻辑之前执行，保证计时器在冷启动流程启动前就已启用。
2. 将自定义过滤器注册逻辑移到SpringBootProxyHandlerBuilder构建链中，不要在build完成后单独调用onStartup方法，异步初始化模式下builder原生支持传入Servlet上下文初始化配置。
3. 核对依赖版本：确认引入的aws-serverless-java-container组件版本和当前使用的Spring Boot大版本匹配，版本不兼容也会触发计时器枚举找不到的问题。

修正后的核心实现代码参考：

public class StreamLambdaHandler implements RequestStreamHandler {

    private static SpringBootLambdaContainerHandler<AwsProxyRequest, AwsProxyResponse> handler;

    static {
        try {
            // 冷启动流程开始前提前启用计时器
            Timer.enable();
            // 初始化超时设置需大于Lambda配置的函数超时时间，避免框架提前终止初始化
            LambdaContainerHandler.getContainerConfig().setInitializationTimeout(2000000);

            handler = new SpringBootProxyHandlerBuilder<AwsProxyRequest>()
                            .defaultProxy()
                            .asyncInit()
                            .springBootApplication(SituationalFlexibilityApp.class)
                            // 过滤器注册逻辑移入构建链，在Servlet上下文初始化阶段执行
                            .initializationHooks(servletContext -> {
                                FilterRegistration.Dynamic registration = servletContext.addFilter("CognitoIdentityFilter",CognitoIdentityFilter.class);
                                registration.addMappingForUrlPatterns(EnumSet.of(DispatcherType.REQUEST), true, "/*");
                            })
                            .buildAndInitialize();

        } catch (ContainerInitializationException e) {
            e.printStackTrace();
            throw new RuntimeException("Could not initialize Spring Boot application", e);
        }
    }

    public StreamLambdaHandler() {
        // 移除构造函数中的Timer.enable()调用
    }

    @Override
    public void handleRequest(InputStream input, OutputStream output, Context context) throws IOException {
        handler.proxyStream(input, output, context);
    }
}

冷启动优化建议 #

• 排查Spring Boot冗余自动装配：排除不需要的数据源、消息队列、监控等自动装配配置，移除未使用的Starter依赖，可大幅降低冷启动耗时。
• 若使用Java 11及以上运行时，可开启Lambda SnapStart特性，无需修改业务代码即可将冷启动耗时降低90%左右。
• 异步初始化模式下首次请求会等待容器初始化完成属于正常现象，后续热启动请求不会产生该等待耗时。

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