如何在Java版AWS Lambda中返回自定义HTTP状态码与消息

你的代码看起来是基于Spring Cloud Function构建的HTTP服务，默认情况下函数返回值会被包装为200响应，导致ResponseEntity无法直接生效。以下几种方法可以实现自定义HTTP状态码和消息：

方案1：使用Message<>包装返回结果 #

通过Spring Messaging的Message对象，在消息头中指定HTTP状态码，Spring Cloud Function的HTTP适配器会识别该头并设置响应状态：

package process;

import org.springframework.boot.autoconfigure.SpringBootApplication;
import org.springframework.http.HttpHeaders;
import org.springframework.http.HttpStatus;
import org.springframework.messaging.Message;
import org.springframework.messaging.support.MessageBuilder;
import org.springframework.web.bind.annotation.RequestBody;
import process.dto.Request;

@SpringBootApplication
public class FunctionConfiguration {

    public static void main(String[] args) {
        // 自定义运行时场景保留该行；直接HTTP请求建议启动Spring应用
        // SpringApplication.run(FunctionConfiguration.class, args);
    }

    public Message<String> process(@RequestBody Request request) {
        // 模拟错误场景：邮箱为空
        if (request.getEmail() == null || request.getEmail().isBlank()) {
            String errorMsg = "邮箱不能为空";
            return MessageBuilder.withPayload(errorMsg)
                    .setHeader(HttpHeaders.STATUS_CODE, HttpStatus.BAD_REQUEST.value())
                    .build();
        }

        // 正常处理逻辑
        String response = "Hello, World " + request.getEmail();
        return MessageBuilder.withPayload(response)
                .setHeader(HttpHeaders.STATUS_CODE, HttpStatus.OK.value())
                .build();
    }
}

方案2：自定义异常+全局异常处理器 #

通过抛出自定义异常，结合全局异常处理器捕获并返回指定状态码与消息：

步骤1：定义自定义异常类 #

package process.exception;

import org.springframework.http.HttpStatus;

public class CustomProcessingException extends RuntimeException {
    private final HttpStatus status;

    public CustomProcessingException(String message, HttpStatus status) {
        super(message);
        this.status = status;
    }

    public HttpStatus getStatus() {
        return status;
    }
}

步骤2：编写全局异常处理器 #

package process.handler;

import org.springframework.http.ResponseEntity;
import org.springframework.web.bind.annotation.ControllerAdvice;
import org.springframework.web.bind.annotation.ExceptionHandler;
import process.exception.CustomProcessingException;

@ControllerAdvice
public class GlobalExceptionHandler {

    @ExceptionHandler(CustomProcessingException.class)
    public ResponseEntity<String> handleCustomException(CustomProcessingException ex) {
        return new ResponseEntity<>(ex.getMessage(), ex.getStatus());
    }
}

步骤3：在业务方法中抛出异常 #

package process;

import org.springframework.boot.autoconfigure.SpringBootApplication;
import org.springframework.http.HttpStatus;
import org.springframework.web.bind.annotation.RequestBody;
import process.dto.Request;
import process.exception.CustomProcessingException;

@SpringBootApplication
public class FunctionConfiguration {

    public static void main(String[] args) {
        // 根据运行环境选择是否启动Spring应用
        // SpringApplication.run(FunctionConfiguration.class, args);
    }

    public String process(@RequestBody Request request) {
        if (request.getEmail() == null || request.getEmail().isBlank()) {
            throw new CustomProcessingException("邮箱为必填项", HttpStatus.BAD_REQUEST);
        }
        return "Hello, World " + request.getEmail();
    }
}

方案3：改用标准Spring Web端点（推荐直接HTTP请求场景） #

如果不需要Spring Cloud Function的函数式特性，可以将类改为@RestController，使用标准的Spring Web注解，此时ResponseEntity会完全生效：

package process;

import org.springframework.boot.SpringApplication;
import org.springframework.boot.autoconfigure.SpringBootApplication;
import org.springframework.http.ResponseEntity;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestBody;
import org.springframework.web.bind.annotation.RestController;
import process.dto.Request;
import org.springframework.http.HttpStatus;

@SpringBootApplication
@RestController
public class FunctionConfiguration {

    public static void main(String[] args) {
        // 必须启动Spring Boot应用才能接收HTTP请求
        SpringApplication.run(FunctionConfiguration.class, args);
    }

    @PostMapping("/process")
    public ResponseEntity<String> process(@RequestBody Request request) {
        if (request.getEmail() == null || request.getEmail().isBlank()) {
            return ResponseEntity.status(HttpStatus.BAD_REQUEST).body("邮箱不能为空");
        }
        return ResponseEntity.ok("Hello, World " + request.getEmail());
    }
}

注意事项 #

• 若为AWS Lambda自定义运行时场景，需确保main方法包含SpringApplication.run调用，否则Spring上下文无法初始化。
• 方案1、2适用于保留Spring Cloud Function函数式特性的场景，方案3适用于纯HTTP API场景。

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