Kotlin协程在队列化弹框中的应用和原理剖析

2023-12-11 06:17:05

利用 Kotlin 协程实现优雅高效的"连续依次弹窗"

前言

在 Android 开发中，我们经常需要处理连续依次显示弹窗的需求。传统的方法通常涉及复杂的状态管理和回调处理，十分繁琐。本文将深入探讨如何使用 Kotlin 协程的强大功能，让"连续依次弹窗"变得简单而优雅。

代码预览

// 定义协程挂起函数显示弹窗
suspend fun showDialog(title: String, message: String) {
    // 创建协程作用域
    val scope = CoroutineScope(Dispatchers.Main)

    // 创建协程挂起函数等待用户点击弹窗按钮
    val deferred = scope.async {
        val dialog = AlertDialog.Builder(this@MainActivity)
            .setTitle(title)
            .setMessage(message)
            .setPositiveButton("确定") { dialog, which -> }
            .setNegativeButton("取消") { dialog, which -> }
            .show()

        // 等待用户点击弹窗按钮
        dialog.awaitButton()
    }

    // 显示弹窗
    deferred.await()
}

// 定义扩展函数等待弹窗按钮点击
fun AlertDialog.awaitButton() {
    // 创建 CompletableFuture 接收按钮点击事件
    val future = CompletableFuture<Unit>()

    // 设置按钮点击事件监听器
    setButtonOnClickListener(DialogInterface.BUTTON_POSITIVE) { _, _ ->
        // 点击“确定”按钮，将 CompletableFuture 置为完成状态
        future.complete(Unit)
    }

    setButtonOnClickListener(DialogInterface.BUTTON_NEGATIVE) { _, _ ->
        // 点击“取消”按钮，将 CompletableFuture 置为取消状态
        future.cancel(false)
    }

    // 等待 CompletableFuture 完成或取消
    future.get()
}

代码解析

在上述代码中，我们定义了一个协程挂起函数showDialog()来显示弹窗。该函数接收两个参数：title和message，分别表示弹窗标题和内容。

我们使用async函数创建协程，以便在显示弹窗的同时继续执行后续代码。在协程内部，我们使用AlertDialog.Builder构建弹窗，然后调用show()方法显示它。

为了等待用户点击弹窗按钮，我们定义了一个扩展函数awaitButton()。该函数创建一个CompletableFuture对象来接收按钮点击事件，并设置按钮点击事件监听器。当用户点击按钮时，CompletableFuture会置为完成或取消状态。

最后，我们使用await()方法等待CompletableFuture完成或取消。这会挂起协程，直到用户做出选择。

使用 suspendCancellableCoroutine 恢复协程

除了使用CompletableFuture，我们还可以使用suspendCancellableCoroutine函数将基于回调的方法包装成协程挂起函数。这可以让我们在弹窗消失时恢复协程。

suspend fun showDialog(title: String, message: String): Boolean {
    // 创建协程作用域
    val scope = CoroutineScope(Dispatchers.Main)

    // 创建协程挂起函数等待弹窗消失
    val deferred = scope.async {
        // 创建 CompletableFuture 接收弹窗消失事件
        val future = CompletableFuture<Boolean>()

        // 设置弹窗消失事件监听器
        val dialog = AlertDialog.Builder(this@MainActivity)
            .setTitle(title)
            .setMessage(message)
            .setPositiveButton("确定") { dialog, which ->
                // 点击“确定”按钮，将 CompletableFuture 置为完成状态，并传递 true
                future.complete(true)
            }
            .setNegativeButton("取消") { dialog, which ->
                // 点击“取消”按钮，将 CompletableFuture 置为完成状态，并传递 false
                future.complete(false)
            }
            .setOnDismissListener {
                // 弹窗消失时，将 CompletableFuture 置为完成状态，并传递 false
                future.complete(false)
            }
            .show()

        // 等待弹窗消失
        future.get()
    }

    // 显示弹窗
    deferred.await()

    // 返回用户点击的按钮
    return deferred.getCompleted()
}