如何检查协程是否使用相同的上下文?

问题描述

我有两个协程,我想检查它们是否在同一上下文/调度程序中运行。这是我的问题的简化版本,但是答案将适用于我正在做的事情:

@Test
fun test() {
   val io = runBlocking(dispatcher.IO) {
       coroutineContext
   }
   val nonIo = runBlocking() {
       coroutineContext
   }
   assertNotEquals(io,nonIo)
}

但是,这是一个糟糕的测试,因为我只是在比较两个不同的对象。我想比较一下他们是否使用dispatcher.IO

解决方法

我认为您的意思是问协程是否使用相同的Dispatcher,对吗?调度程序只是CoroutineContext的一部分。您可以将CoroutineContext的get函数与键ContinuationInterceptor一起使用来检索Dispatcher。

val io = withContext(Dispatchers.IO) {
    coroutineContext[ContinuationInterceptor]
}
val nonIo = withContext(Dispatchers.Default) {
    coroutineContext[ContinuationInterceptor]
}
assertNotEquals(io,nonIo)