如何在Spring Webflux控制器中结合Flux和ResponseEntity

我在我的Webflux控制器中使用Monos和ResponseEntitys来操纵标头和其他响应信息.例如：

@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) }` ?
}

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

更新：
这里的存储库是ReactiveCrudRepository

解决方法:

如果用户不存在,请使用switchIfEmpty抛出异常：

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

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

标签：java,spring,kotlin,spring-webflux,project-reactor
来源： https://codeday.me/bug/20190622/1261806.html