在kotlin协同例程中,如何将数据范围界定为“请求路径”(例如MDC)?

你说的大部分都是正确的。我们不能轻易使用 ThreadLocal 使用协程,因为它们经常切换线程。线程局部程序的标准替换是将数据存储在伴随协同程序传递的协同程序上下文中。然而,正如您所指出的,如果不在协同程序中运行,我们就无法访问上下文。

不,没有办法访问suspend函数之外的协同程序上下文。上下文未存储在 ThreadLocal -它直接作为参数传递给每个suspend函数。每个suspend函数都有一个“隐藏”的延续参数,这就是上下文的存储位置。非挂起函数不会接收到延续,因此无法从它们访问上下文。

我认为从挂起和非挂起上下文访问数据的最简单方法可能是使用 ThreadLocal ,但是为协程创建一个钩子,以便在恢复时恢复此线程本地值。事实上,协同程序已经提供了这样一个钩子。我们可以为给定的添加一个钩子 ThreadLocal 通过使用 asContextElement() 分机:

private val variable = ThreadLocal<String>()

suspend fun main(): Unit = withContext(Dispatchers.Default) {
    launch(variable.asContextElement("foo")) {
        repeat(5) {
            launch {
                println("foo: ${Thread.currentThread().name}: ${variable.get()}")
            }
        }
    }
    launch(variable.asContextElement("bar")) {
        repeat(5) {
            launch {
                println("bar: ${Thread.currentThread().name}: ${variable.get()}")
            }
        }
    }
}

样本结果:

bar: DefaultDispatcher-worker-1: bar
foo: DefaultDispatcher-worker-3: foo
foo: DefaultDispatcher-worker-2: foo
bar: DefaultDispatcher-worker-1: bar
foo: DefaultDispatcher-worker-2: foo
bar: DefaultDispatcher-worker-1: bar
foo: DefaultDispatcher-worker-2: foo
bar: DefaultDispatcher-worker-4: bar
foo: DefaultDispatcher-worker-1: foo
bar: DefaultDispatcher-worker-3: bar

在本例中,我们运行两个并发的“请求”: foo 和 bar ,对于 variable 。两个请求共享相同的线程,但的值 变量 总是与请求正确匹配。

如果将其专门用于MDC和slf4j,请注意协同程序已经为其提供了支持: MDCContext 。我相信它在下面使用了与上面类似的技术。

使用作用域启动新的协同程序时请小心,例如。 lifecycleScope.launch {} .通过显式提供作用域启动的协同程序不被视为当前协同程序的子程序,它们是独立的执行流,因此,它们不会正确传递您的存储值。您必须重新初始化挂钩,例如:

lifecycleScope.launch(variable.asContextElement()) { ... }

完成 Broot's answer ,要解决使用作用域启动新协同程序的问题,我们可以简单地创建一个新的扩展方法,将threadLocal添加到新作业的上下文中,如下所示:

fun CoroutineScope.launchModified(
    context: CoroutineContext = EmptyCoroutineContext,
    start: CoroutineStart = CoroutineStart.DEFAULT,
    block: suspend CoroutineScope.() -> Unit
): Job {
    return launch(this.coroutineContext + threadLocal.asContextElement() , start = start, block)
}