github-repos/mirai
1
0
Fork 0
mirror of https://github.com/mamoe/mirai.git synced 2025-01-21 07:56:56 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix cancellation check in AbstractKeepAliveNetworkHandlerSelector

Browse Source
This commit is contained in:
Him188 2021-07-31 16:11:34 +08:00
parent 978228afc1
commit 4a2b8ac92c
1 changed files with 5 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
mirai-core/src/commonMain/kotlin/network/handler/selector/AbstractKeepAliveNetworkHandlerSelector.kt
View File

@ -10,7 +10,10 @@
package net.mamoe.mirai.internal.network.handler.selector
import kotlinx.atomicfu.atomic
import kotlinx.coroutines.*
import kotlinx.coroutines.currentCoroutineContext
import kotlinx.coroutines.delay
import kotlinx.coroutines.isActive
import kotlinx.coroutines.yield
import net.mamoe.mirai.internal.network.handler.NetworkHandler
import net.mamoe.mirai.internal.network.handler.NetworkHandlerFactory
import net.mamoe.mirai.internal.network.handler.logger
@ -87,9 +90,7 @@ internal abstract class AbstractKeepAliveNetworkHandlerSelector<H : NetworkHandl
// ?: MaxAttemptsReachedException(null)
}
if (!currentCoroutineContext().isActive) {
logIfEnabled { "Cancellation detected." }
yield() // throw canonical CancellationException if cancelled
throw CancellationException("Cancellation detected.")
yield() // check cancellation
}
val current = getCurrentInstanceOrNull()
lastNetwork = current