Spring 5中的HTTP响应异常处理

Question

我正在使用Spring Boot 2和带有WebFlux反应式启动器的Spring 5开发一些反应式微服务。

我面临以下问题：我想处理调用另一个REST服务时收到的所有HTTP状态，并在收到一些错误的HTTP状态时抛出异常。例如，当我调用一个endpoint并收到404 HTTP状态时，我想抛出一个异常，并在某个ExceptionHandler类中处理该异常，就像在Spring 4中使用@ControllerAdvice时一样。

正确的方法是什么？希望收到一些好的建议。

Answer 1

我认为你正在寻找的是WebFLux响应状态ExceptionHandler检查这个以供参考。

在WebHandler API中，WebExceptionHandler可用于处理来自WebFilter和目标WebHandler链的异常。在使用WebFlux配置时，注册WebExceptionHandler非常简单，只需将其声明为Spring bean，并可以通过bean声明中的@Order或通过实现Order来表示优先级。

这个例子可能会有所帮助，但我自己没有尝试过。

@Component
@Order(-2)
class RestWebExceptionHandler implements WebExceptionHandler{

    @Override
    public Mono<Void> handle(ServerWebExchange exchange, Throwable ex) {
        if (ex instanceof PostNotFoundException) {
            exchange.getResponse().setStatusCode(HttpStatus.NOT_FOUND);

            // marks the response as complete and forbids writing to it
            return exchange.getResponse().setComplete();
        }
        return Mono.error(ex);
    }
}

class PostNotFoundException extends RuntimeException {
    PostNotFoundException(String id) {
        super("Post:" + id + " is not found.");
    }
}

Answer 2

这可以通过两个独立的部分来解决。

使用WebClient时，您可以从远程服务接收HTTP 404响应。默认情况下，所有4xx和5xx客户端响应都将转换为WebClientResponseException。因此，您可以在WebFlux应用程序中直接处理这些异常。

如果只想将404响应转换为自定义异常，可以执行以下操作：

WebClient webClient = //...
webClient.get().uri("/persons/1")
  .retrieve()
  .onStatus(httpStatus -> HttpStatus.NOT_FOUND.equals(httpStatus),
                        clientResponse -> Mono.error(new MyCustomException()))
  .bodyToMono(...);

这显然是在每个客户电话的基础上完成的。

使用ExchangeFilterFunction可以以更可重用的方式实现同样的功能，您可以在WebClient实例上一次性设置该功能，如下所示：

WebClient.builder().filter(myExchangeFilterFunction)...

使用带注释的Spring WebFlux，您可以使用@ExceptionHandler注释的方法处理异常（请参阅Spring框架参考文档）。

注意：使用WebExceptionHandler是可能的，但它的级别非常低，因为您在那里没有高级支持：您需要使用缓冲区手动编写响应，而不支持序列化。