使用 Kotling 协程处理阻塞代码的正确方法

编程问答 2022-05-23

问题描述

假设我由于某些第三方库而具有阻塞功能。一些类似的东西：


fun useTheLibrary(arg: String): String {
   val result = BlockingLibrary.doSomething(arg)
   return result
}

对 BlockingLibrary.doSomething 的调用应在单独的 ThreadPoolExecutor 上运行。

使用 kotlin 实现这一目标的正确方法是什么（假设有一种方法）？

注意：我已阅读 this thread 但似乎已经过时

解决方法

如果阻塞代码因 CPU 使用而阻塞，则应使用 Dispatchers.Default。如果它是网络或磁盘绑定的，请使用 Dispatchers.IO。您可以将其设置为挂起函数并将阻塞调用包装在 withContext 中，以允许该函数在从协程调用时正确挂起：

suspend fun useTheLibrary(arg: String): String = withContext(Dispatchers.Default) {
   BlockingLibrary.doSomething(arg)
}

如果您因为 API 要求需要使用特定的 ThreadPoolExecutor，您可以使用 asCoroutineDispatcher()。

val myDispatcher = myExecutor.asCoroutineDispatcher()

//...

suspend fun useTheLibrary(arg: String): String = withContext(myDispatcher) {
   BlockingLibrary.doSomething(arg)
}

如果您的库包含基于回调的方式来运行阻塞代码，您可以使用 suspendCoroutine() 或 suspendCancellableCoroutine() 将其转换为挂起函数。在这种情况下，您无需担心执行程序或调度程序，因为它是由库自己的线程池处理的。 Here's an example 在 Retrofit 库中，他们将自己的基于回调的 API 转换为挂起函数。

blockingkotlinkotlinkotlin-coroutinesnonblocking