Spring 5反应式异常处理

时间:2019-02-23 16:56:04

标签: spring exception-handling spring-webflux

我正在编写一个复杂的应用程序作为Spring 5 Webflux的实验。我打算在此应用程序中使用很多技术。我熟悉“旧样式” @RestController,但是现在,我正在编写功能端点。例如。公司注册服务的后端。我曾经在“旧世界”中使用@ControllerAdvice之类的东西。但是我真的无法找到任何类似的反应式等效物。 (或者对我有用的任何方法。)

我有一个非常基本的设置。路由功能,反应式Cassandra存储库,处理程序和测试类。存储库操作会引发IllegalArgumentException,我想通过将HTTP Status BadRequest返回给客户端来处理它。举例来说,我有能力做到这一点。 :-)异常由处理程序类处理。这是我的代码。

RouterConfig

@Slf4j
@Configuration
@EnableWebFlux
public class RouterConfig {

@Bean
public RouterFunction<ServerResponse> route(@Autowired CompanyHandler handler) {
    return RouterFunctions.route()
            .nest(path("/company"), bc -> bc
                    .GET("/{id}", handler::handleGetCompanyDataRequest)                     
                    .before(request -> {
                        log.info("Request={} has been received.", request.toString());
                        return request;
                    })
            .after((request, response) -> {
                log.info("Response={} has been sent.", response.toString());
                return response;
            }))
            .build();       
    }
}

CompanyHandler

@Slf4j
@Component
public class CompanyHandler {

    @Autowired
    private ReactiveCompanyRepository repository;

    // Handle get single company data request
    public Mono<ServerResponse> handleGetCompanyDataRequest(ServerRequest request) {
        //Some validation ges here

        return repository.findById(Mono.just(uuid))
            .flatMap(this::ok)
            .onErrorResume(IllegalArgumentException.class, e -> ServerResponse.badRequest().build())
            .switchIfEmpty(ServerResponse.notFound().build());
    }

    private Mono<ServerResponse> ok (Company c) {
        return ServerResponse.ok().contentType(MediaType.APPLICATION_JSON)
            .body(BodyInserters.fromPublisher(Mono.just(c), Company.class));
    }
}

ReactiveCompanyRepository

@Component
public interface ReactiveCompanyRepository extends ReactiveCassandraRepository<Company, UUID>{

    Mono<Company> findByName(String name);

    Mono<Company> findByEmail(String email);
}

我的问题是 .onErrorResume(IllegalArgumentException.class,e-> ServerResponse.badRequest()。build())从未调用,并且在测试用例中:

@SuppressWarnings("unchecked")
@Test
public void testGetCompanyExceptionDuringFind() {

    Mockito.when(repository.findById(Mockito.any(Mono.class))).thenThrow(new IllegalArgumentException("Hahaha"));       

    WebTestClient.bindToRouterFunction(routerConfig.route(companyHandler))
        .build()
        .get().uri("/company/2b851f10-356e-11e9-a847-0f89e1aa5554")
        .accept(MediaType.APPLICATION_JSON_UTF8)
        .exchange()
        .expectStatus().isBadRequest()
        .returnResult(Company.class)
        .getResponseBody();
}

我总是得到HttpStatus 500而不是400。所以它失败了。任何帮助将不胜感激!

1 个答案:

答案 0 :(得分:0)

您的测试未表示应有的反应式API的行为。如果这样的存储库直接引发异常,则我认为该错误会报告给维护人员。

当反应式API返回MonoFlux之类的反应式类型时,预计所有错误都不会直接抛出,而是实际上作为消息在反应式管道中发送。

在这种情况下,您的测试用例应该更像是:

Mockito.when(repository.findById(Mockito.any(Mono.class)))
   .thenReturn(Mono.error(new IllegalArgumentException("Hahaha")));

有了这个, Reactor中的onError...个运算符将处理这些错误消息。