如何在 Android 上使用 Coroutines 取消或不中断,包括根据生命周期自动取消?

问题描述

背景

我在从 Android 上的简单(已弃用)AsyncTaskExecutors 迁移到 Kotlin Coroutines 时遇到问题

问题

我不知道如何使用 AsyncTask 执行我可以在 Executors 甚至 Kotlin Coroutines 上完成的基本操作。

过去,我可以选择在线程中断和不中断的情况下取消任务。现在出于某种原因,给定我在协程上创建的任务,它只是没有中断,这意味着如果我运行一些甚至“睡眠”在其中的代码(并不总是由我),它不会被中断。

我还记得我在某处被告知协程在 Android 上非常好,因为如果您在 Activity 中,它会自动取消所有任务。不过,我找不到有关如何执行此操作的说明。

我的尝试和发现

对于 Coroutines 任务(根据我所见称为 Deferred)我想我在创建它时已经读过,我必须选择它将支持哪个取消,并且出于某种原因我可以'他们两个都有。不确定这是否属实,但我仍然想知道,因为我想同时拥有两者以获得最佳迁移。使用 AsyncTask,我曾经将它们添加一个集合中(并在取消时删除),以便在 Activity 完成后,我可以查看所有内容并将它们全部取消。我什至做了一个很好的课程来为我做这件事。

这是我为了测试这个而创建的:

class MainActivity : AppCompatActivity() {
    val uiScope = Coroutinescope(dispatchers.Main)
    val bgdispatcher: Coroutinedispatcher = dispatchers.IO

    override fun onCreate(savedInstanceState: Bundle?) {
        super.onCreate(savedInstanceState)
        setContentView(R.layout.activity_main)
        loadData()
    }

    private fun loadData(): Job = uiScope.launch {
        Log.d("AppLog","loadData")
        val task = async(bgdispatcher) {
            Log.d("AppLog","bg start")
            try {
                Thread.sleep(5000L) //this Could be any executing of code,including things not editable
            } catch (e: Exception) {
                Log.d("AppLog","$e")
            }
            Log.d("AppLog","bg done this.isActive?${this.isActive}")
            return@async 123
        }
        //simulation of cancellation for any reason,sadly without the ability to cancel with interruption
        Handler(mainLooper).postDelayed({
            task.cancel()
        },2000L)
        val result: Int = task.await()
        Log.d("AppLog","got result:$result") // this is called even if you change orientation,which I might not want when done in Activity
    }
}

构建gradle文件

implementation "org.jetbrains.kotlinx:kotlinx-coroutines-core:1.4.1"
implementation "org.jetbrains.kotlinx:kotlinx-coroutines-android:1.4.1"

问题

  1. 协程是否有可能有一个任务,我可以在线程中断和不中断的情况下取消?
  2. 应该添加什么来使这个工作,以便当 Activity 终止时(例如方向改变),它会被自动取消(选择中断或不中断)?我想我可以使用与 AsyncTask 类似的解决方案,但我记得有人告诉我有一个很好的方法可以为协程做到这一点。

解决方法

协程不是魔法。它们是使用状态机实现的,并且可能有许多暂停点。这在原Coroutines KEEP中都有解释。

取消发生在那些暂停点。除了挂起点(至少在正常执行中)之外,不能在任何其他点取消协同程序。如果您使用 Thread.sleep,则没有暂停点。您应该使用 delay 而不是 sleep,后者引入了暂停点。如果你正在做一个长时间的操作,你可以添加几个 yield() 来添加暂停点并使你的协程可取消。

来自the docs

协程取消是合作的。协程代码必须协作才能取消。 kotlinx.coroutines 中的所有挂起函数都是可以取消的。他们检查协程的取消并在取消时抛出 CancellationException。但是,如果协程在计算中工作并且不检查取消,则无法取消

调用 suspend 函数会自动引入一个暂停点。

正如@CommonsWare 所指出的,很少有理由创建您的协程范围。在 Activity 和 Fragment 或任何与生命周期相关的组件中,您应该使用 lifecycleScope。在 ViewModel 中,有 viewModelScope

编辑:

我尝试在特定条件下将 runInterruptible 的源调整为 not 中断:传递自定义异常类 InterruptionException 的实例作为取消原因将跳过线程中断。我已经用 AtomicInteger 替换了 atomicfu 结构,我假设你的目标只是 JVM。您需要通过添加 -Xopt-in=kotlinx.coroutines.InternalCoroutinesApi 编译器标志来选择加入内部协程 API。

suspend fun <T> runInterruptibleCancellable(
    context: CoroutineContext = EmptyCoroutineContext,block: () -> T
): T = withContext(context) {
    try {
        val threadState = ThreadState(coroutineContext.job)
        threadState.setup()
        try {
            block()
        } finally {
            threadState.clearInterrupt()
        }
    } catch (e: InterruptedException) {
        throw CancellationException("Blocking call was interrupted due to parent cancellation").initCause(e)
    }
}

private const val WORKING = 0
private const val FINISHED = 1
private const val INTERRUPTING = 2
private const val INTERRUPTED = 3

private class ThreadState(private val job: Job) : CompletionHandler {
    /*
       === States ===

       WORKING: running normally
       FINISH: complete normally
       INTERRUPTING: canceled,going to interrupt this thread
       INTERRUPTED: this thread is interrupted

       === Possible Transitions ===

       +----------------+         register job       +-------------------------+
       |    WORKING     |   cancellation listener    |         WORKING         |
       | (thread,null) | -------------------------> | (thread,cancel handle) |
       +----------------+                            +-------------------------+
               |                                                |   |
               | cancel                                  cancel |   | complete
               |                                                |   |
               V                                                |   |
       +---------------+                                        |   |
       | INTERRUPTING  | <--------------------------------------+   |
       +---------------+                                            |
               |                                                    |
               | interrupt                                          |
               |                                                    |
               V                                                    V
       +---------------+                              +-------------------------+
       |  INTERRUPTED  |                              |         FINISHED        |
       +---------------+                              +-------------------------+
    */
    private val _state = AtomicInteger(WORKING)
    private val targetThread = Thread.currentThread()

    // Registered cancellation handler
    private var cancelHandle: DisposableHandle? = null

    fun setup() {
        cancelHandle = job.invokeOnCompletion(onCancelling = true,invokeImmediately = true,handler = this)
        // Either we successfully stored it or it was immediately cancelled
        while (true) {
            when (val state = _state.get()) {
                // Happy-path,move forward
                WORKING -> if (_state.compareAndSet(state,WORKING)) return
                // Immediately cancelled,just continue
                INTERRUPTING,INTERRUPTED -> return
                else -> invalidState(state)
            }
        }
    }

    fun clearInterrupt() {
        /*
         * Do not allow to untriggered interrupt to leak
         */
        while (true) {
            when (val state = _state.get()) {
                WORKING -> if (_state.compareAndSet(state,FINISHED)) {
                    cancelHandle?.dispose()
                    return
                }
                INTERRUPTING -> {
                    /*
                     * Spin,cancellation mechanism is interrupting our thread right now
                     * and we have to wait it and then clear interrupt status
                     */
                }
                INTERRUPTED -> {
                    // Clear it and bail out
                    Thread.interrupted()
                    return
                }
                else -> invalidState(state)
            }
        }
    }

    // Cancellation handler
    override fun invoke(cause: Throwable?) {
        if (cause is InterruptionException) {
            while (true) {
                when (val state = _state.get()) {
                    // Working -> try to transite state and interrupt the thread
                    WORKING -> {
                        if (_state.compareAndSet(state,INTERRUPTING)) {
                            targetThread.interrupt()
                            _state.set(INTERRUPTED)
                            return
                        }
                    }
                    // Finished -- runInterruptible is already complete,INTERRUPTING - ignore
                    FINISHED,INTERRUPTING,INTERRUPTED -> return
                    else -> invalidState(state)
                }
            }
        }
    }

    private fun invalidState(state: Int): Nothing = error("Illegal state $state")
}

class InterruptionException(cause: Throwable?) : CancellationException() {
    init {
        initCause(cause)
    }
}

fun Job.interrupt(cause: Throwable? = null) {
    this.cancel(InterruptionException(cause))
}

suspend fun Job.interruptAndJoin() {
    interrupt()
    return join()
}

可以使用interruptinterruptAndJoin扩展函数触发线程中断,否则使用cancel进行非中断取消。一个例子:

val scope = CoroutineScope(Dispatchers.IO)
val job = scope.launch {
    runInterruptibleCancellable {
        // some blocking code
        Thread.sleep(1000)
        if (!isActive) {
            println("cancelled")
        } else {
            println("completed")
        }
    }
}
job.invokeOnCompletion {
    if (it is InterruptionException) {
        print("interrupted")
    }
}

runBlocking {
//  job.interruptAndJoin()  // prints "interrupted"
//  job.cancelAndJoin()     // prints "cancelled"
    job.join()              // prints "completed"
}

这个例子是我所做的唯一测试。它似乎起作用了。我不知道它是否泄漏,我不知道它是否是线程安全的。老实说,我真的远远超出了我的专业知识。请不要在没有进一步确认它有效的情况下在生产中使用它。

,

默认情况下,协程不执行线程中断 - 根据 Making computation code cancellable documentation,使用 yield() 或检查 isActive 允许协程感知代码参与取消。

但是,当在您确实想要线程中断的地方与阻塞代码交互时,这正是 runInterruptible() 的用例,当协程范围被取消。

这与 lifecycle-aware coroutine scopes 完美配合,当 Lifecycle 被销毁时会自动取消:

class MainActivity : AppCompatActivity() {
    val bgDispatcher: CoroutineDispatcher = Dispatchers.IO

    override fun onCreate(savedInstanceState: Bundle?) {
        super.onCreate(savedInstanceState)
        setContentView(R.layout.activity_main)
        loadData()
    }

    private fun loadData(): Job = lifecycleScope.launch {
        Log.d("AppLog","loadData")
        val result = runInterruptible(bgDispatcher) {
            Log.d("AppLog","bg start")
            try {
                Thread.sleep(5000L) //this could be any executing of code,including things not editable
            } catch (e: Exception) {
                Log.d("AppLog","$e")
            }
            Log.d("AppLog","bg done this.isActive?${this.isActive}")
            return@runInterruptible 123
        }
        Log.d("AppLog","got result:$result")
    }
}