WebFlux funktional: Wie erkennt man einen leeren Flux und kehrt 404 zurück?

Question

Ich habe die folgende vereinfachte Handler-Funktion (Spring WebFlux und die funktionale API mit Kotlin). Ich brauche jedoch einen Hinweis, wie man ein leeres Flux erkennt und dann noContent() für 404 verwendet, wenn das Flux leer ist.

fun findByLastname(request: ServerRequest): Mono<ServerResponse> { 
    val lastnameOpt = request.queryParam("lastname") 
    val customerFlux = if (lastnameOpt.isPresent) { 
     service.findByLastname(lastnameOpt.get()) 
    } else { 
     service.findAll() 
    } 
    // How can I detect an empty Flux and then invoke noContent() ? 
    return ok().body(customerFlux, Customer::class.java) 
} 

Answer 1

Von einem Mono:

return customerMono 
      .flatMap(c -> ok().body(BodyInserters.fromObject(c))) 
      .switchIfEmpty(notFound().build()); 

Aus Flux:

return customerFlux 
      .collectList() 
      .flatMap(l -> { 
       if(l.isEmpty()) { 
       return notFound().build(); 

       } 
       else { 
       return ok().body(BodyInserters.fromObject(l))); 
       } 
      }); 

Beachten Sie, dass collectList Puffer Daten im Speicher, so dass dies nicht die beste Wahl für große Listen sein könnten. Es könnte einen besseren Weg geben, dies zu lösen.

fun <R> Flux<R>.collectListOrEmpty(): Mono<List<R>> = this.collectList().flatMap { 
     val result = if (it.isEmpty()) { 
      Mono.empty() 
     } else { 
      Mono.just(it) 
     } 
     result 
    } 

es

Und nennen, wie Sie es tun:

Answer 2

Neben der Lösung von Brian, wenn Sie nicht eine leere Überprüfung der Liste die ganze Zeit tun wollen, können Sie eine Erweiterungsfunktion erstellen für das Mono:

return customerFlux().collectListOrEmpty() 
        .switchIfEmpty(notFound().build()) 
        .flatMap(c -> ok().body(BodyInserters.fromObject(c)))