Spring Cloud Function通过API Gateway Header路由时触发NullPointerException问题求助

我碰到过类似的问题，这个空指针异常是Spring Cloud Function 3.2.1版本在处理API Gateway代理请求时的一个已知bug，根源在于AWSLambdaUtils.generateMessage方法中尝试读取请求header时，没有处理header对象为空的情况。当通过spring.cloud.function.definition header路由时，API Gateway的请求事件结构导致代码无法正确提取headers，进而触发NPE。

问题原因分析 #

在3.2.1版本的AWSLambdaUtils中，处理API Gateway Proxy请求事件时，默认的header提取逻辑存在漏洞：当Lambda收到的APIGatewayProxyRequestEvent中headers为null（或者没有正确被解析）时，直接调用headers.getFirst(FUNCTION_DEFINITION_HEADER)就会抛出空指针异常。而本地运行或环境变量路由时，不需要从请求header中读取函数定义，所以不会触发这个问题。

解决办法 #

方案1：升级Spring Cloud Function版本（推荐）

这个bug在Spring Cloud Function 3.2.2及后续版本中已经被修复。你可以将pom.xml中的Spring Cloud Function依赖版本升级到3.2.2或更高（比如3.2.5，这个版本兼容Java 8，不需要Docker容器部署）。修改后的依赖配置大致如下：

<dependencyManagement>
    <dependencies>
        <dependency>
            <groupId>org.springframework.cloud</groupId>
            <artifactId>spring-cloud-function-dependencies</artifactId>
            <version>3.2.5</version>
            <type>pom</type>
            <scope>import</scope>
        </dependency>
    </dependencies>
</dependencyManagement>

升级后重新打包部署Lambda，header路由应该就能正常工作了。

方案2：自定义FunctionInvoker绕过问题（临时解决方案）

如果暂时无法升级版本，可以自定义一个FunctionInvoker的子类，手动处理API Gateway请求的headers，避免空指针：

import com.amazonaws.services.lambda.runtime.Context;
import com.amazonaws.services.lambda.runtime.events.APIGatewayProxyRequestEvent;
import com.fasterxml.jackson.databind.ObjectMapper;
import org.springframework.cloud.function.adapter.aws.FunctionInvoker;
import org.springframework.http.HttpHeaders;

import java.io.ByteArrayInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;

public class CustomFunctionInvoker extends FunctionInvoker {
    private static final ObjectMapper OBJECT_MAPPER = new ObjectMapper();
    private static final String FUNCTION_DEFINITION_HEADER = "spring.cloud.function.definition";

    @Override
    public Object handleRequest(InputStream input, OutputStream output, Context context) throws IOException {
        // 读取API Gateway请求事件
        APIGatewayProxyRequestEvent event = OBJECT_MAPPER.readValue(input, APIGatewayProxyRequestEvent.class);
        
        // 手动提取并处理headers，确保不为空
        HttpHeaders headers = new HttpHeaders();
        if (event.getHeaders() != null) {
            event.getHeaders().forEach(headers::add);
        }
        
        // 如果header存在，将其设置为系统属性（模拟环境变量方式）
        String functionDefinition = headers.getFirst(FUNCTION_DEFINITION_HEADER);
        if (functionDefinition != null) {
            System.setProperty(FUNCTION_DEFINITION_HEADER, functionDefinition);
        }
        
        // 重新构造输入流并调用父类方法
        ByteArrayInputStream newInput = new ByteArrayInputStream(OBJECT_MAPPER.writeValueAsBytes(event));
        return super.handleRequest(newInput, output, context);
    }
}

然后将Lambda的请求处理器修改为你的自定义类，比如com.yourpackage.CustomFunctionInvoker::handleRequest，重新部署即可。

额外注意事项 #

• 确保API Gateway的集成配置正确：使用Lambda代理集成，并且在集成请求中允许传递spring.cloud.function.definition header（API Gateway默认会保留大部分自定义header，但如果有特殊配置需要检查）。
• 如果升级版本后遇到Java版本问题，3.2.x系列的稳定版本（如3.2.5）是兼容Java 8的，不需要使用Docker容器部署，可以直接在AWS Lambda上运行。

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