如何在AWS Lambda自定义运行时调用Spring Boot GraalVM编译的Handler？

我构建了一个Spring Boot 3应用并使用GraalVM编译。使用GraalVM时需在AWS Lambda中选择自定义运行时，我选了arm64架构的Amazon Linux 2023作为运行时，创建了bootstrap文件并和可执行文件打包成zip上传。测试Lambda函数时发现Spring Boot已启动，但无法调用Java代码中的Handler，导致函数报错。

错误日志 #

.   ____          _            __ _ _
/\ / ___'_ __ _ _(_)_ __  __ _ \ \ \ \
( ( )\___ | '_ | '_| | '_ \/ _` | \ \ \ \
\/  ___)| |_)| | | | | || (_| |  ) ) ) )
'  |____| .__|_| |_|_| |_\__, | / / / /
=========|_|==============|___/=/_/_/_/
:: Spring Boot ::                (v3.2.1)
INIT_REPORT Init Duration: 10009.69 ms  Phase: init Status: timeout
INIT_REPORT Init Duration: 3020.29 ms   Phase: invoke   Status: error   Error Type: Runtime.Unknown
START RequestId: d043768d-c4b1-4ad0-af1e-586e383db38e Version: $LATEST
2023-12-28T10:16:52.988Z d043768d-c4b1-4ad0-af1e-586e383db38e Task timed out after 3.04 seconds

END RequestId: d043768d-c4b1-4ad0-af1e-586e383db38e
REPORT RequestId: d043768d-c4b1-4ad0-af1e-586e383db38e  Duration: 3041.01 ms    Billed Duration: 3000 ms    Memory Size: 128 MB Max Memory Used: 38 MB  

注意：自定义运行时配置中没有选择Handler方法的选项，而不使用GraalVM时，在JAVA 17等AWS Lambda支持的环境中可正常运行该应用。

附加文件 #

bootstrap #

#!/bin/sh
cd ${LAMBDA_TASK_ROOT:-.}
set -euo pipefail
./demo_linux_aws

Handler函数 #

public class StreamLambdaHandler implements RequestStreamHandler {
    private static SpringBootLambdaContainerHandler<AwsProxyRequest, AwsProxyResponse> handler;
    static {
        try {
            handler = SpringBootLambdaContainerHandler.getAwsProxyHandler(Application.class);
        } catch (ContainerInitializationException e) {
            // if we fail here. We re-throw the exception to force another cold start
            e.printStackTrace();
            throw new RuntimeException("Could not initialize Spring Boot application", e);
        }
    }

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

1. 补全GraalVM反射配置 #

GraalVM原生镜像默认禁用反射，而Spring Boot Lambda容器依赖反射初始化Handler和Spring上下文。需在编译时添加必要的反射规则：

• 在项目src/main/resources/META-INF/native-image目录下创建reflect-config.json，添加Handler和应用主类的反射配置：

[
  {
    "name": "你的包路径.StreamLambdaHandler",
    "allDeclaredConstructors": true,
    "allPublicConstructors": true,
    "allDeclaredMethods": true,
    "allPublicMethods": true
  },
  {
    "name": "你的包路径.Application",
    "allDeclaredConstructors": true,
    "allPublicConstructors": true,
    "allDeclaredMethods": true,
    "allPublicMethods": true
  }
]

• 或直接使用Spring Boot原生镜像插件自动生成配置，确保pom.xml/build.gradle中启用插件：
  Maven示例：

  <plugin>
    <groupId>org.springframework.boot</groupId>
    <artifactId>spring-boot-maven-plugin</artifactId>
    <configuration>
      <image>
        <builder>paketobuildpacks/builder-jammy-base:latest</builder>
        <env>
          <BP_NATIVE_IMAGE>true</BP_NATIVE_IMAGE>
        </env>
      </image>
    </configuration>
  </plugin>
  

2. 调整Lambda初始化超时 #

日志显示初始化阶段超时（10009.69ms），刚好触发Lambda默认10秒的初始化超时上限。进入Lambda配置页面，将初始化超时调整为30秒或更长，确保Spring上下文能完成初始化。

3. 修改bootstrap文件，适配Lambda事件传递 #

当前bootstrap仅启动可执行文件，但未将Lambda事件传递到Handler。需结合AWS Lambda Runtime Interface Client（RIC）处理事件：

• 先添加RIC依赖：
  Maven示例：

  <dependency>
    <groupId>com.amazonaws</groupId>
    <artifactId>aws-lambda-java-runtime-interface-client</artifactId>
    <version>2.3.1</version>
  </dependency>
  

• 修改bootstrap文件，指定Handler路径：

  #!/bin/sh
  cd ${LAMBDA_TASK_ROOT:-.}
  exec ./demo_linux_aws 你的包路径.StreamLambdaHandler
  

4. 提升Lambda内存配置 #

当前内存设置为128MB，GraalVM原生镜像虽内存占用低，但Spring Boot初始化可能需要更多资源。尝试将内存调整为512MB，降低初始化阶段的内存压力，避免超时。

5. 验证架构与版本兼容性 #

确保使用的GraalVM版本（推荐20/21）与Spring Boot 3.2.1兼容，且编译时使用arm64架构的GraalVM，与Lambda运行时架构完全一致，避免因架构不兼容导致的执行失败。

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