Spring 5中的HTTP响应异常处理Reactive

Question

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

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

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

Answer 1

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

如何将WebClient接收的HTTP 404响应转换为自定义异常

使用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)...

如何处理WebFlux应用程序中的自定义异常

随着春天WebFlux与注释，您可以处理与注解的方法例外@ExceptionHandler （见Spring Framework的参考文档 ）。

注意：使用WebExceptionHandler是可能的，但它的级别很低，因为你没有高级支持：你需要手动编写带缓冲区的响应，而不需要任何序列化支持。

Answer 2

我认为你要找的是WebFluxResponseStatusExceptionHandler检查这个以供参考 。

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

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

@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.");
    }
}