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

Question

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


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

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

使用 kotlin 实现此目的的正确方法是什么（假设有办法）？

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

Answer 1

如果阻塞代码因为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 转换为挂起函数。

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

Proper way of dealing with blocking code using Kotling coroutines

kotlin

kotlin-coroutines

blocking

nonblocking