如何将协程直接分派到 JVM 上的主线程？

Question

我正在为 jvm 建立一个基于 kotlin 协程的网络框架。 Client 和 Server 类实现 CoroutineScope，coroutinecontext 的覆盖是 Dispatchers.IO，因为我很确定这是用于这种情况的正确 Dispatcher。但是，我希望在主线程上处理读取数据包，或者至少提供该选项。在没有阅读文档的情况下，我使用了 Dispatchers.Main，我现在意识到它是用于 android UI 线程的。有没有我可以用来让协程在主线程上运行的调度程序？如果没有，我将如何制作一个？

我查看了有关如何基于单个线程创建调度程序的 kotlin 文档，但除了创建新线程的 newSingleThreadContext 之外我找不到任何东西。我还发现可以从 java Executor 创建调度程序，但我仍然不确定如何将其限制为已经存在的线程。

class AbstractNetworkComponent : CoroutineScope {
    private val packetProcessor = PacketProcessor()
    private val job = Job()
    override val coroutineContext = job + Dispatchers.IO
}

class PacketProcessor : CoroutineScope {

    private val job = Job()
    override val coroutineContext = job + Dispatchers.Main //Android only!
    private val packetHandlers = mutableMapOf<Opcode, PacketHandlerFunc>()

    fun handlePacket(opcode: Opcode, packet: ReceivablePacket, networker: Writable) {
        launch(coroutineContext) {
            packetHandlers[opcode]?.invoke(packet, networker)
        }
    }
}

因此，由于缺少 android 组件，使用 Dispatchers.Main 我得到了 IllegalStateException。有没有办法创建一个调度程序来阻止主线程直到它完成（就像 runBlocking 一样？）谢谢！

Answer 1

runBlocking 正是您所需要的。它创建一个调度程序并将其设置在协程上下文中。您可以使用

访问调度程序

coroutineContext[ContinuationInterceptor] as CoroutineDispatcher

然后你可以将它传递给一个实现CoroutineScope 的对象或者你想用它做的任何其他事情。下面是一些示例代码：

import kotlinx.coroutines.*
import kotlinx.coroutines.Dispatchers.IO
import kotlin.coroutines.ContinuationInterceptor

fun main() {
    println("Top-level: current thread is ${Thread.currentThread().name}")
    runBlocking {
        val dispatcher = coroutineContext[ContinuationInterceptor]
                as CoroutineDispatcher
        ScopedObject(dispatcher).launchMe().join()
    }
}

class ScopedObject(dispatcher: CoroutineDispatcher) : CoroutineScope {
    override val coroutineContext = Job() + dispatcher

    fun launchMe() = launch {
        val result = withContext(IO) {
            "amazing"
        }
        println("Launched coroutine: " +
                "current thread is ${Thread.currentThread().name}, " +
                "result is $result")
    }
}

这将打印出来

Top-level: current thread is main
Launched coroutine: current thread is main, result is amazing

Answer 2

根据Guide to UI programming with coroutines，kotlinx.coroutines 具有三个模块，为不同的 UI 应用程序库提供协程上下文：

• kotlinx-coroutines-android -- Android 的 Dispatchers.Main 上下文 应用程序。
• kotlinx-coroutines-javafx -- JavaFX UI 的 Dispatchers.JavaFx 上下文 应用程序。
• kotlinx-coroutines-swing -- 用于 Swing UI 的 Dispatchers.Swing 上下文 应用程序。

此外，UI 调度程序可通过 kotlinx-coroutines-core 的 Dispatchers.Main 获得，相应的实现（Android、JavaFx 或 Swing）由 ServiceLoader API 发现。例如，如果您正在编写 JavaFx 应用程序，您可以使用 Dispatchers.Main 或 Dispachers.JavaFx 扩展名，这将是同一个对象。