线程 API 转换成挂起函数:suspendCoroutine
在实际项目中即使已经使用协程了,可是要完全避免跟传统的线程 API 交互并不容易,大型项目一般都会有比较多的老代码或外部库没有用协程,使用的还是回调的写法。那么就很有必要知道怎么让协程和线程 API 的回调交互。
协程有一个专用的函数 suspendCoroutine,它是一个挂起函数,在它里面调用传统的回调式函数,就能把回调式的函数转换成挂起函数:
lifecycleScope.launch {
val contributors = callbackToSuspend()
showContributors(contributors)
}
private suspend fun callbackToSuspend() = suspendCoroutine {
gitHub.contributorsCall("square", "retrofit")
.enqueue(object : Callback<List<Contributor>> {
override fun onResponse(
call: Call<List<Contributor>>, response: Response<List<Contributor>> {
// 将结果返回
it.resume(response.body()!!)
}
override fun onFailure(call: Call<List<Contributor>>) {
// 发生异常时让 suspendCoroutine 立即结束并抛出这个异常
it.resumeWithException(t)
}
})
}
}
使用 suspendCoroutine 包裹的回调式代码需要调用 suspendCoroutine 提供的 continuation.resume 和 continuation.resumeWithException 分别处理正常返回结果和异常的情况。
支持取消的 suspendCoroutine:suspendCancellableCoroutine
协程还提供了一个类似的函数 suspendCancellableCoroutine,和 suspendCoroutine 的区别是它支持取消。
private suspend fun callbackToCancellableSuspend() = suspendCancellableCoroutine {
it.invokeOnCancellation {
// 协程被取消,处理协程被取消时要做的一些收尾工作清理现场
}
gitHub.contributorsCall("square", "retrofit")
.enqueue(object : Callback<List<Contributor>> {
override fun onResponse(
call: Call<List<Contributor>>, response: Response<List<Contributor>> {
// 将结果返回
it.resume(response.body()!!)
}
override fun onFailure(call: Call<List<Contributor>>) {
// 发生异常时让 suspendCoroutine 立即结束并抛出这个异常
it.resumeWithException(t)
}
})
}
}
使用 suspendCancellableCoroutine 还可以注册取消的回调,使用 cancellableContinuation.invokeOnCancellation 处理协程被取消时的收尾清理工作。
我们用一个例子来说明 suspendCoroutine 和 suspendCancellableCoroutine 的区别:
val job = lifecycleScope.launch {
// 假设 callbackToSuspend 会在延时 2s 后继续执行
try {
val contributors = callbackToSuspend()
// val contributors = callbackToCancellableSuspend()
showContributors(contributors)
} catch (e: Exception) {
textView.text = e.message
}
}
lifecycleScope.launch {
delay(200)
job.cancel() // 200ms 后取消协程
}
假设 callbackToSuspend 函数是使用 suspendCoroutine 包起来的回调代码,会在 2s 后返回结果;协程 200ms 后被取消了,但是里面的代码是不配合的,因为协程的取消本身就是一个状态标记,2s 后还是会继续执行代码。
而如果用 suspendCancellableCoroutine 在 200ms 后会正常取消协程,会在 try-catch 抛出 CancellableException 异常,不会在继续执行后续代码。
我们一般在项目中都使用能支持取消的 suspendCancellableCoroutine,除非特殊需求需要启动后协程取消了也得继续执行才用 suspendCoroutine。
总结
将线程 API 的回调式代码用 suspendCoroutine 或 suspendCancellableCoroutine 包住,就能实现将回调式代码转换为挂起函数在协程执行,需要调用提供的 continuation.resume 和 continuation.resumeWithException 分别处理正常返回结果和异常的情况
suspendCancellableCoroutine 和 suspendCoroutine 的区别是它支持取消和注册协程取消回调;我们一般在项目中都使用能支持取消的 suspendCancellableCoroutine,除非特殊需求需要启动后协程取消了也得继续执行才用 suspendCoroutine