当 Flux 为空时返回 404

Question

我试图在 Flux 为空时返回 404，类似于此处：WebFlux functional: How to detect an empty Flux and return 404?

我主要担心的是，当您检查通量是否具有元素时，它会发出该值并且您会丢失它。当我尝试在服务器响应上使用 switch if empty 时，它永远不会被调用（我暗中认为这是因为 Mono 不是空的，只有正文是空的）。

我正在做的一些代码（我的 Router 类上有一个过滤器，用于检查 DataNotFoundException 以返回 notFound）：

Flux<Location> response = this.locationService.searchLocations(searchFields, pageToken);
return ok()
        .contentType(APPLICATION_STREAM_JSON)
        .body(response, Location.class)
        .switchIfEmpty(Mono.error(new DataNotFoundException("The data you seek is not here.")));

^This 从不调用 switchIfEmpty

Flux<Location> response = this.locationService.searchLocations(searchFields, pageToken);

return response.hasElements().flatMap(l ->{
   if(l){
       return ok()
               .contentType(APPLICATION_STREAM_JSON)
               .body(response, Location.class);
   } 
   else{
       return Mono.error(new DataNotFoundException("The data you seek is not here."));
   }
});

^这会丢失 hasElements 上发出的元素。

有没有办法在 hasElements 中恢复发出的元素或让 switchIfEmpty 只检查正文的内容？

Answer 1

您可以将switchIfEmpty 运算符应用于您的Flux<Location> response。

Flux<Location> response = this.locationService
        .searchLocations(searchFields, pageToken)
        .switchIfEmpty(Mono.error(new DataNotFoundException("The data you seek is not here.")));

Answer 2

虽然发布的答案确实正确，但如果您只想返回状态代码（加上原因）并且不想摆弄任何自定义过滤器或定义自己的错误响应异常，则可以使用便利异常类。

另一个好处是您不必将响应包装在任何 ResponseEntity 对象中，虽然在某些情况下很有用（例如，使用位置 URI 创建），但对于简单的状态响应来说有点过头了。

另见https://docs.spring.io/spring-framework/docs/current/javadoc-api/org/springframework/web/server/ResponseStatusException.html

 return this.locationService.searchLocations(searchFields, pageToken)
        .buffer()
        .switchIfEmpty(Mono.error(new ResponseStatusException(HttpStatus.NOT_FOUND, "these are not the droids you are lookig for")));

Answer 3

亚历山大写的是正确的。您在从不为空的对象上调用switchIfEmpty ServerResponse.ok() 根据定义不是一个空的发布者。我喜欢反向处理这种情况，因此调用服务，然后链接所有创建响应的方法。

    this.locationService.searchLocations(searchFields, pageToken)
            .buffer()
            .map(t -> ResponseEntity.ok(t))
            .defaultIfEmpty(ResponseEntity.notFound().build());

更新（不确定是否有效，但试一试）：

 public Mono<ServerResponse> myRestMethod(ServerRequest serverRequest) {
        return serverRequest.bodyToMono(RequestDTO.class)
                .map((request) -> searchLocations(request.searchFields, request.pageToken))
                .flatMap( t -> ServerResponse
                        .ok()
                        .body(t, ResponseDTO.class)
                )
                .switchIfEmpty(ServerResponse.notFound().build())
                ;
    }