java - 如何在 Spring Webflux Controller 中结合 Flux 和 ResponseEntity

Question

我使用 Mono与 ResponseEntity s 在我的 Webflux Controller 中，以便操作标题和其他响应信息。例如:

@GetMapping("/{userId}")
fun getOneUser(@PathVariable userId: UserId): Mono<ResponseEntity<UserDto>> {
    return repository.findById(userId)
        .map(User::asDto)
        .map { ResponseEntity.ok(it) }
        .defaultIfEmpty(ResponseEntity.notFound().build())
}

@GetMapping
fun getAllUsers(): Flux<UserDto> {
    return repository.findAllActive().map(User::asDto)
}

两者都可以正常工作，但在某些情况下需要 ResponseEntity连同Flux以及。响应类型应该是什么？使用 ResponseEntity<Flux<T>> 是否正确？ ？

例如:

@GetMapping("/{userId}/options")
fun getAllUserOptions(@PathVariable userId: UserId): ??? {
    return repository.findById(userId)
        .flatMapIterable{ it.options }
        .map { OptionDto.from(it) }
        // if findById -> empty Mono then:
        //   return ResponseEntity.notFound().build() ?
        // else:
        //   return the result of `.map { OptionDto.from(it) }` ?
}

我想在这里实现的行为是 getAllUserOptions 返回 404如果repository.findById(userId)是空的 Mono , 否则返回 user.options作为Flux .

更新: 这里的存储库是 ReactiveCrudRepository

Answer 1

使用 switchIfEmpty 在用户不存在的情况下抛出异常:

return repository
    .findById(userId)
    .switchIfEmpty(Mono.error(NotFoundException("User not found")))
    .flatMapIterable{ it.options }
    .map { OptionDto.from(it) }

然后用exception handler将其转换为 404 响应。