在Kotlin流中使用ReactiveSecurityContextHolder

Question

我正在使用静态编程语言进行Spring Boot（2.2）项目，CouchDB作为（反应性）数据库，结果是异步DAO（挂起函数或返回Flow的函数）。我试图设置WebFlux，以便也有异步控制器（同样，我想返回流，而不是Flux）。但是我在从ReactiveSecurityContextHolder检索我的安全上下文时遇到了麻烦。

据我所知，SecurityContextHolder使用ThreadLocal来存储它，ReactiveSecurityContextHolder依赖于这样一个事实，Spring在订阅我的反应链时，也将该上下文存储在这个链中，从而允许我调用ReactiveSecurityContextHolder。从链中获取上下文（）。

问题是我得把我的单声道

@GetMapping
fun getArticles(): Flow<String> {
    return ReactiveSecurityContextHolder.getContext().flux().asFlow() // returns nothing
}

请注意，如果我直接返回通量（跳过.asFlow（）），或添加。single（）或。toList（）最后（因此使用了一个suspend fun），然后它就可以正常工作了，并返回了我的安全上下文，但这又不是我想要的。我猜解决方案是将上下文从流量（从ReactiveSecurityContextHolder的初始反应链）转移到流，但它似乎不是默认完成的。

编辑：下面是一个示例项目，展示了这个问题：https://github.com/Simon3/webflux-kotlin-sample




Answer 1

您真正要实现的是从Flow内部访问ReactorContext。

这样做的一种方法是放松对返回Flow的需求，而是返回Flux。这允许您恢复ReactorContext并将其传递给要用于生成数据的Flow。

@ExperimentalCoroutinesApi
@GetMapping("/flow")
fun flow(): Flux<Map<String, String>> = Mono.subscriberContext().flatMapMany { reactorCtx ->
    flow {
        val ctx = coroutineContext[ReactorContext.Key]?.context?.get<Mono<SecurityContext>>(SecurityContext::class.java)?.asFlow()?.single()
        emit(mapOf("user" to ((ctx?.authentication?.principal as? User)?.username ?: "<NONE>")))
    }.flowOn(reactorCtx.asCoroutineContext()).asFlux()
}

在需要从挂起方法访问ReactorContext的情况下，只需从coroutineContext获取它，无需进一步技巧：

@ExperimentalCoroutinesApi
@GetMapping("/suspend")
suspend fun suspend(): Map<String,String> {
    val ctx = coroutineContext[ReactorContext.Key]?.context?.get<Mono<SecurityContext>>(SecurityContext::class.java)?.asFlow()?.single()
    return mapOf("user" to ((ctx?.authentication?.principal as? User)?.username ?: "<NONE>"))
}