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

Proper way of dealing with blocking code using Kotling coroutines

假设我因为某些第三方库而具有阻止功能。沿着这些线的东西:


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 转换为挂起函数。