剖析Kotlin 协程挂起的神秘面纱(原理篇)

2023-10-16 23:51:06

前言

协程挂起一直被认为是理解和使用 Kotlin 协程的关键难点之一。在上一篇文章中，我们从拟物的角度阐述了协程挂起/恢复的场景，相信大家对此应该有了一个感性的认识。在更早一篇文章里，我们分析了如何开启一个原始的协程，相信大家也知道协程内部执行原理了。

本文将重点分析协程挂起与恢复的原理，帮助读者更深入地理解协程的底层机制，从而更好地掌握协程的用法。

挂起函数

在 Kotlin 中，一个挂起函数是一个可以在协程中使用的函数，它可以在执行过程中暂停，然后在稍后继续执行。挂起函数可以通过使用 suspend 来声明。

suspend fun mySuspendFunction() {
    // Do something that might take a long time
}

当一个挂起函数被调用时，它会创建一个新的协程，并在协程中执行。如果挂起函数在执行过程中需要等待某个操作完成，它可以调用 suspend 关键字来挂起协程。挂起协程后，当前线程可以去执行其他任务，而挂起的协程会等待操作完成。

suspend fun mySuspendFunction() {
    val result = longRunningOperation()
    // Do something with the result
}

fun longRunningOperation(): Int {
    Thread.sleep(1000) // Simulate a long-running operation
    return 42
}

在上面的例子中，mySuspendFunction() 函数在调用 longRunningOperation() 函数时会挂起协程。在 longRunningOperation() 函数执行期间，协程会被挂起，当前线程可以去执行其他任务。当 longRunningOperation() 函数执行完成后，协程会被恢复，mySuspendFunction() 函数可以继续执行。

恢复协程

当一个协程被挂起后，它可以通过调用 resume() 方法来恢复。resume() 方法可以传递一个参数，该参数将作为挂起函数的返回值。

fun main() {
    val myCoroutine = GlobalScope.launch {
        val result = longRunningOperation()
        println(result)
    }

    myCoroutine.resume(42)
}

在上面的例子中，main() 函数创建了一个协程，并在协程中调用 longRunningOperation() 函数。然后，main() 函数调用 myCoroutine.resume(42) 方法来恢复协程。这将导致 longRunningOperation() 函数返回 42，并将其打印到控制台。