Kotlin Coroutines(協程) 完全解析(四),協程的異常處理

Kotlin Coroutines(協程) 完全解析系列:

Kotlin Coroutines(協程) 完全解析(一),協程簡介

Kotlin Coroutines(協程) 完全解析(二),深入理解協程的掛起、恢復與調度

Kotlin Coroutines(協程) 完全解析(三),封裝異步回調、協程間關系及協程的取消

Kotlin Coroutines(協程) 完全解析(四),協程的異常處理

Kotlin Coroutines(協程) 完全解析(五),協程的并發

本文基于 Kotlin v1.3.0-rc-146,Kotlin-Coroutines v1.0.0-RC1

在上一篇文章中提到子協程拋出未捕獲的異常時默認會取消其父協程,而拋出CancellationException卻會當作正常的協程結束不會取消其父協程。本文來詳細解析協程中的異常處理,拋出未捕獲異常后協程結束后運行會不會崩潰,可以攔截協程的未捕獲異常嗎,如何讓子協程的異常不影響父協程。

Kotlin 官網文檔中有關于協程異常處理的文章,里面的內容本文就不再重復,所以讀者們先閱讀官方文檔:

Coroutine Exception handling

協程的異常處理(官方文檔中文版)

看完官方文檔后,可能還是會有一些疑問:

  • launch式協程的未捕獲異常為什么會自動傳播到父協程,為什么對異常只是在控制臺打印而已?

  • async式協程的未捕獲異常為什么需要依賴用戶來最終消耗異常?

  • 自定義的CoroutineExceptionHandler的是如何生效的?

  • 異常的聚合是怎么處理的?

  • SupervisorJobsupervisorScope實現異常單向傳播的原理是什么?

這些疑問在本文逐步解析協程中異常處理的流程時,會一一解答。

1. 協程中異常處理的流程

從拋出異常的地方開始跟蹤協程中異常處理的流程,拋出異常時一般都在協程的運算邏輯中。而在第二篇深入理解協程的掛起、恢復與調度中提到在協程的三層包裝中,運算邏輯在第二層BaseContinuationImplresumeWith()函數中的invokeSuspend運行,所以再來看一次:

internal abstract class BaseContinuationImpl(
    public val completion: Continuation<Any?>?
) : Continuation<Any?>, CoroutineStackFrame, Serializable {
    public final override fun resumeWith(result: Result<Any?>) {
        ...
        var param = result
        while (true) {
            with(current) {
                val completion = completion!!
                val outcome: Result<Any?> =
                    try {
                        // 調用 invokeSuspend 方法執行,執行協程的真正運算邏輯
                        val outcome = invokeSuspend(param)
                        // 協程掛起時 invokeSuspend 才會返回 COROUTINE_SUSPENDED,所以協程掛起時,其實只是協程的 resumeWith 運行邏輯執行完成,再次調用 resumeWith 時,協程掛起點之后的邏輯才能繼續執行
                        if (outcome === COROUTINE_SUSPENDED) return
                        Result.success(outcome)
                    } catch (exception: Throwable) {
                        // 注意這個 catch 語句,其實協程運算中所有異常都會在這里被捕獲,然后作為一種運算結果
                        Result.failure(exception)
                    }
                releaseIntercepted() // this state machine instance is terminating
                if (completion is BaseContinuationImpl) {
                    // unrolling recursion via loop
                    current = completion
                    param = outcome
                } else {
                    // 這里實際調用的是其父類 AbstractCoroutine 的 resumeWith 方法,當捕獲到異常時,調用 resumeWith(Result.failure(exception)) 更新協程狀態
                    completion.resumeWith(outcome)
                    return
                }
            }
        }
    }
}

從上面源碼的try {} catch {}語句來看,首先協程運算過程中所有未捕獲異常其實都會在第二層包裝中被捕獲,然后會通過AbstractCoroutine.resumeWith(Result.failure(exception))進入到第三層包裝中,所以協程的第三層包裝不僅維護協程的狀態,還處理協程運算中的未捕獲異常。這在第三篇分析子協程拋出未捕獲異常,默認情況會取消其父線程時也提到過。

繼續跟蹤 AbstractCoroutine.resumeWith(Result.failure(exception)) -> JobSupport.makeCompletingOnce(CompletedExceptionally(exception), defaultResumeMode) -> JobSupport.tryMakeCompleting(state, CompletedExceptionally(exception), defaultResumeMode),在最后tryMakeCompleting()過程中部分關鍵代碼:

private fun tryMakeCompleting(state: Any?, proposedUpdate: Any?, mode: Int): Int {
    ...
    // process cancelling notification here -- it cancels all the children _before_ we start to to wait them (sic!!!)
// 該情景下,notifyRootCause 的值為 exception
    notifyRootCause?.let { notifyCancelling(list, it) }
    // now wait for children
    val child = firstChild(state)
    if (child != null && tryWaitForChild(finishing, child, proposedUpdate))
        return COMPLETING_WAITING_CHILDREN
    // otherwise -- we have not children left (all were already cancelled?)
// 已取消所有子協程后,更新該協程的最終狀態
    if (tryFinalizeFinishingState(finishing, proposedUpdate, mode))
        return COMPLETING_COMPLETED
    // otherwise retry
    return COMPLETING_RETRY
}

先看notifyCancelling(state.list, exception)函數:

private fun notifyCancelling(list: NodeList, cause: Throwable) {
    // first cancel our own children
    onCancellation(cause)
// 這里會調用 handle 節點的 invoke() 方法取消子協程,具體點就是調用 childJob.parentCancelled(job) 取消子協程
    notifyHandlers<JobCancellingNode<*>>(list, cause)
    // then cancel parent
// 然后可能會取消父協程
    cancelParent(cause) // tentative cancellation -- does not matter if there is no parent
}

private fun cancelParent(cause: Throwable): Boolean {
    // CancellationException is considered "normal" and parent is not cancelled when child produces it.
    // This allow parent to cancel its children (normally) without being cancelled itself, unless
    // child crashes and produce some other exception during its completion.
// CancellationException 是正常的協程結束行為,手動拋出 CancellationException 也不會取消父協程
    if (cause is CancellationException) return true
// cancelsParent 屬性也可以決定出現異常時是否取消父協程,不過一般該屬性都為 true
    if (!cancelsParent) return false
// parentHandle?.childCancelled(cause) 最后會通過調用 parentJob.childCancelled(cause) 取消父協程
    return parentHandle?.childCancelled(cause) == true
}

所以出現未捕獲異常時,首先會取消所有子協程,然后可能會取消父協程。而有些情況下并不會取消父協程,一是當異常屬于 CancellationException 時,而是使用SupervisorJobsupervisorScope時,子協程出現未捕獲異常時也不會影響父協程,它們的原理是重寫 childCancelled() 為override fun childCancelled(cause: Throwable): Boolean = false。

launch式協程和async式協程都會自動向上傳播異常,取消父協程。

接下來再看tryFinalizeFinishingState的實現:

private fun tryFinalizeFinishingState(state: Finishing, proposedUpdate: Any?, mode: Int): Boolean {
    ...
// proposedException 即前面未捕獲的異常
    val proposedException = (proposedUpdate as? CompletedExceptionally)?.cause
    // Create the final exception and seal the state so that no more exceptions can be added
    var suppressed = false
    val finalException = synchronized(state) {
        val exceptions = state.sealLocked(proposedException)
        val finalCause = getFinalRootCause(state, exceptions)
        // Report suppressed exceptions if initial cause doesn't match final cause (due to JCE unwrapping)
// 如果在處理異常過程還有其他異常,這里通過 finalCause.addSuppressedThrowable(exception) 的方式記錄下來
        if (finalCause != null) suppressed = suppressExceptions(finalCause, exceptions) || finalCause !== state.rootCause
        finalCause
    }
    ...
    // Now handle exception if parent can't handle it
// 如果 finalException 不是 CancellationException,而且有父協程且不為 SupervisorJob 和 supervisorScope,cancelParent(finalException) 都返回 true
// 也就是說一般情況下出現未捕獲的異常,一般會傳遞到最根部的協程,由最頂端的協程去處理
    if (finalException != null && !cancelParent(finalException)) {
        handleJobException(finalException)
    }
    ...
    // And process all post-completion actions
    completeStateFinalization(state, finalState, mode, suppressed)
    return true
}

上面代碼中if (finalException != null && !cancelParent(finalException))語句可以看出,除非是 SupervisorJob 和 supervisorScope,一般協程出現未捕獲異常時,不僅會取消父協程,一步步取消到最根部的協程,而且最后還由最根部的協程(Root Coroutine)處理協程。下面繼續看處理異常的handleJobException的實現:

// JobSupport
protected open fun handleJobException(exception: Throwable) {}

// Builders.common.kt
private open class StandaloneCoroutine(
    parentContext: CoroutineContext,
    active: Boolean
) : AbstractCoroutine<Unit>(parentContext, active) {
    override val cancelsParent: Boolean get() = true
    override fun handleJobException(exception: Throwable) = handleExceptionViaHandler(parentContext, exception)
}

// Actor
private open class ActorCoroutine<E>(
    ...
) : ChannelCoroutine<E>(parentContext, channel, active), ActorScope<E> {
    override fun onCancellation(cause: Throwable?) {
        _channel.cancel(cause)
    }

    override val cancelsParent: Boolean get() = true
    override fun handleJobException(exception: Throwable) = handleExceptionViaHandler(parentContext, exception)
}

默認的handleJobException的實現為空,所以如果 Root Coroutine 為async式協程,不會有任何異常打印操作,也不會 crash,但是為launch式協程或者actor式協程的話,會調用handleExceptionViaHandler()處理異常。

下面接著看handleExceptionViaHandler()的實現:

internal fun handleExceptionViaHandler(context: CoroutineContext, exception: Throwable) {
    // Invoke exception handler from the context if present
    try {
        context[CoroutineExceptionHandler]?.let {
            it.handleException(context, exception)
// 如果協程有自定義 CoroutineExceptionHandler,則只調用 handler.handleException() 就返回
            return
        }
    } catch (t: Throwable) {
        handleCoroutineExceptionImpl(context, handlerException(exception, t))
        return
    }

    // If handler is not present in the context or exception was thrown, fallback to the global handler
// 如果沒有自定義 CoroutineExceptionHandler,
    handleCoroutineExceptionImpl(context, exception)
}

internal actual fun handleCoroutineExceptionImpl(context: CoroutineContext, exception: Throwable) {
    // use additional extension handlers
// 在 Android 中,還會有 uncaughtExceptionPreHandler 作為額外的 handlers
    for (handler in handlers) {
        try {
            handler.handleException(context, exception)
        } catch (t: Throwable) {
            // Use thread's handler if custom handler failed to handle exception
            val currentThread = Thread.currentThread()
            currentThread.uncaughtExceptionHandler.uncaughtException(currentThread, handlerException(exception, t))
        }
    }

    // use thread's handler
    val currentThread = Thread.currentThread()
// 調用當前線程的 uncaughtExceptionHandler 處理異常
    currentThread.uncaughtExceptionHandler.uncaughtException(currentThread, exception)
}

// Thread.java
public UncaughtExceptionHandler getUncaughtExceptionHandler() {
// 當前線程沒有定義 uncaughtExceptionHandler,會返回線程組作為 Thread.UncaughtExceptionHandler
    return uncaughtExceptionHandler != null ?
        uncaughtExceptionHandler : group;
}

// ThreadGroup.java
public void uncaughtException(Thread t, Throwable e) {
    if (parent != null) {
        parent.uncaughtException(t, e);
    } else {
        Thread.UncaughtExceptionHandler ueh =
            Thread.getDefaultUncaughtExceptionHandler();
// 優先使用線程通用的 DefaultUncaughtExceptionHandler,如果也沒有的話,則在控制臺打印異常堆棧信息
        if (ueh != null) {
            ueh.uncaughtException(t, e);
        } else if (!(e instanceof ThreadDeath)) {
            System.err.print("Exception in thread \""
                                + t.getName() + "\" ");
            e.printStackTrace(System.err);
        }
    }
}

所以默認情況下,launch式協程對未捕獲的異常只是打印異常堆棧信息,如果在 Android 中還會調用uncaughtExceptionPreHandler處理異常。但是如果使用了 CoroutineExceptionHandler 的話,只會使用自定義的 CoroutineExceptionHandler 處理異常。

到這里協程的異常處理流程就走完了,但是還有一個問題還沒解答,async式協程的未捕獲異常只會導致取消自己和取消父協程,又是如何依賴用戶來最終消耗異常呢?

fun main(args: Array<String>) = runBlocking<Unit> {
    val deferred = GlobalScope.async {
        println("Throwing exception from async")
        throw IndexOutOfBoundsException()
    }
// await() 恢復調用者協程時會重寫拋出異常
    deferred.await()
}

看看反編譯的 class 文件就明白了:

public final Object invokeSuspend(@NotNull Object result) {
    Object coroutine_suspended = IntrinsicsKt.getCOROUTINE_SUSPENDED();
    Deferred deferred;
    switch (this.label) {
        case 0:
            if (result instanceof Failure) {
                throw ((Failure) result).exception;
            }
            CoroutineScope coroutineScope = this.p$;
// 創建并啟動一個新的 async 協程
            deferred = BuildersKt.async$default((CoroutineScope) GlobalScope.INSTANCE, null, null, (Function2) new 1(null), 3, null);
            this.L$0 = deferred;
            this.label = 1;
// await() 掛起函數掛起當前協程,等待 async 協程的結果
            if (deferred.await(this) == coroutine_suspended) {
                return coroutine_suspended;
            }
            break;
        case 1:
            deferred = (Deferred) this.L$0;
// async 協程恢復當前協程時,傳遞進來的結果是 CompletedExceptionally(IndexOutOfBoundsException())
            if (result instanceof Failure) {
// 在當前協程重新拋出 IndexOutOfBoundsException 異常
                throw ((Failure) result).exception;
            }
            break;
        default:
            throw new IllegalStateException("call to 'resume' before 'invoke' with coroutine");
    }
    return Unit.INSTANCE;
}

所以async式協程只有通過await()將異常重新拋出,不過可以可以通過try { deffered.await() } catch () { ... }來捕獲異常。

2. 小結

分析完協程的異常處理流程,其中需要注意的問題有下面這些:

  • 拋出 CancellationException 或者調用cancel()只會取消當前協程和子協程,不會取消父協程,也不會其他例如打印堆棧信息等的異常處理操作。

  • 拋出未捕獲的非 CancellationException 異常會取消子協程和自己,也會取消父協程,一直取消 root 協程,異常也會由 root 協程處理。

  • 如果使用了 SupervisorJob 或 supervisorScope,子協程拋出未捕獲的非 CancellationException 異常不會取消父協程,異常也會由子協程自己處理。

  • launch式協程和actor式協程默認處理異常的方式只是打印堆棧信息,可以自定義 CoroutineExceptionHandler 來處理異常。

  • async式協程本身不會處理異常,自定義 CoroutineExceptionHandler 也無效,但是會在await()恢復調用者協程時重新拋出異常。

最后編輯于
?著作權歸作者所有,轉載或內容合作請聯系作者
平臺聲明:文章內容(如有圖片或視頻亦包括在內)由作者上傳并發布,文章內容僅代表作者本人觀點,簡書系信息發布平臺,僅提供信息存儲服務。
  • 序言:七十年代末,一起剝皮案震驚了整個濱河市,隨后出現的幾起案子,更是在濱河造成了極大的恐慌,老刑警劉巖,帶你破解...
    沈念sama閱讀 228,739評論 6 534
  • 序言:濱河連續發生了三起死亡事件,死亡現場離奇詭異,居然都是意外死亡,警方通過查閱死者的電腦和手機,發現死者居然都...
    沈念sama閱讀 98,634評論 3 419
  • 文/潘曉璐 我一進店門,熙熙樓的掌柜王于貴愁眉苦臉地迎上來,“玉大人,你說我怎么就攤上這事?!?“怎么了?”我有些...
    開封第一講書人閱讀 176,653評論 0 377
  • 文/不壞的土叔 我叫張陵,是天一觀的道長。 經常有香客問我,道長,這世上最難降的妖魔是什么? 我笑而不...
    開封第一講書人閱讀 63,063評論 1 314
  • 正文 為了忘掉前任,我火速辦了婚禮,結果婚禮上,老公的妹妹穿的比我還像新娘。我一直安慰自己,他們只是感情好,可當我...
    茶點故事閱讀 71,835評論 6 410
  • 文/花漫 我一把揭開白布。 她就那樣靜靜地躺著,像睡著了一般。 火紅的嫁衣襯著肌膚如雪。 梳的紋絲不亂的頭發上,一...
    開封第一講書人閱讀 55,235評論 1 324
  • 那天,我揣著相機與錄音,去河邊找鬼。 笑死,一個胖子當著我的面吹牛,可吹牛的內容都是我干的。 我是一名探鬼主播,決...
    沈念sama閱讀 43,315評論 3 442
  • 文/蒼蘭香墨 我猛地睜開眼,長吁一口氣:“原來是場噩夢啊……” “哼!你這毒婦竟也來了?” 一聲冷哼從身側響起,我...
    開封第一講書人閱讀 42,459評論 0 289
  • 序言:老撾萬榮一對情侶失蹤,失蹤者是張志新(化名)和其女友劉穎,沒想到半個月后,有當地人在樹林里發現了一具尸體,經...
    沈念sama閱讀 49,000評論 1 335
  • 正文 獨居荒郊野嶺守林人離奇死亡,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內容為張勛視角 年9月15日...
    茶點故事閱讀 40,819評論 3 355
  • 正文 我和宋清朗相戀三年,在試婚紗的時候發現自己被綠了。 大學時的朋友給我發了我未婚夫和他白月光在一起吃飯的照片。...
    茶點故事閱讀 43,004評論 1 370
  • 序言:一個原本活蹦亂跳的男人離奇死亡,死狀恐怖,靈堂內的尸體忽然破棺而出,到底是詐尸還是另有隱情,我是刑警寧澤,帶...
    沈念sama閱讀 38,560評論 5 362
  • 正文 年R本政府宣布,位于F島的核電站,受9級特大地震影響,放射性物質發生泄漏。R本人自食惡果不足惜,卻給世界環境...
    茶點故事閱讀 44,257評論 3 347
  • 文/蒙蒙 一、第九天 我趴在偏房一處隱蔽的房頂上張望。 院中可真熱鬧,春花似錦、人聲如沸。這莊子的主人今日做“春日...
    開封第一講書人閱讀 34,676評論 0 26
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽。三九已至,卻和暖如春,著一層夾襖步出監牢的瞬間,已是汗流浹背。 一陣腳步聲響...
    開封第一講書人閱讀 35,937評論 1 288
  • 我被黑心中介騙來泰國打工, 沒想到剛下飛機就差點兒被人妖公主榨干…… 1. 我叫王不留,地道東北人。 一個月前我還...
    沈念sama閱讀 51,717評論 3 393
  • 正文 我出身青樓,卻偏偏與公主長得像,于是被迫代替她去往敵國和親。 傳聞我的和親對象是個殘疾皇子,可洞房花燭夜當晚...
    茶點故事閱讀 48,003評論 2 374

推薦閱讀更多精彩內容

  • 生活中很多人樂于預測,但是很少有人想過,預測的準確率要大于1/2才有意義,否則的話,就和拋硬幣沒什么區別。如果預測...
    ziworeborn閱讀 784評論 1 1
  • 有時候總會在夜深人靜的時候想起往事,尤其是部隊的往事,往往寫不下去方案,思路混亂的時候,靠在窗邊,想起了軍校...
    風陵曉渡閱讀 317評論 0 0
  • 圖標元素幾乎無處不在,一個偉大的圖標設計可以是獨特的,并為項目添加技巧和天賦。認識到標志設計趨勢是選擇標志設...
    轟隆隆炸雞閱讀 1,449評論 0 2
  • 初二,串親戚,媽媽家。 幾天前,家庭喜悅街群里發出了一則公告: 《公 告》 為活躍節日氣氛,提高孩子們的表達...
    安心安閱讀 895評論 3 7
  • 今年蘭州的雨水特別多,寺院前面路邊的樹下有許多的蝸牛,有的竟然爬到了兩米多高的樹叉上,有的還在一點一點、堅難地...
    與有緣人共進閱讀 575評論 0 1