github-repos/mirai
1
0
Fork 0
mirror of https://github.com/mamoe/mirai.git synced 2025-03-13 14:50:43 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix nextEvent cancellation

Browse Source
This commit is contained in:
Him188 2021-01-29 11:21:56 +08:00
parent 22ac89207a
commit a82c68c7ec
1 changed files with 32 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
mirai-core-api/src/commonMain/kotlin/event/nextEvent.kt
View File

@ -73,7 +73,9 @@ internal suspend inline fun <E : Event> nextEventImpl(
priority: EventPriority,
crossinline filter: (E) -> Boolean
): E = suspendCancellableCoroutine { cont ->
coroutineScope.globalEventChannel().subscribe(eventClass, priority = priority) {
val listener = coroutineScope.globalEventChannel()
.parentJob(coroutineScope.coroutineContext[Job])
.subscribe(eventClass, priority = priority) {
if (!filter(this)) return@subscribe ListeningStatus.LISTENING
try {
@ -82,8 +84,18 @@ internal suspend inline fun <E : Event> nextEventImpl(
}
return@subscribe ListeningStatus.STOPPED
}
cont.invokeOnCancellation {
runCatching { listener.cancel("nextEvent outer scope cancelled", it) }
}
}
@Suppress("INVISIBLE_MEMBER", "INVISIBLE_REFERENCE", "NOTHING_TO_INLINE")
@kotlin.internal.HidesMembers
@PublishedApi
internal inline fun <BaseEvent : Event> EventChannel<BaseEvent>.parentJob(job: Job?): EventChannel<BaseEvent> =
if (job != null) parentJob(job) else this
@JvmSynthetic
@PublishedApi
internal suspend inline fun <E : BotEvent> nextBotEventImpl(
@ -92,25 +104,33 @@ internal suspend inline fun <E : BotEvent> nextBotEventImpl(
coroutineScope: CoroutineScope,
priority: EventPriority
): E = suspendCancellableCoroutine { cont ->
coroutineScope.globalEventChannel().subscribe(eventClass, priority = priority) {
val listener = coroutineScope.globalEventChannel()
.parentJob(coroutineScope.coroutineContext[Job])
.subscribe(eventClass, priority = priority) {
try {
if (this.bot == bot) cont.resume(this)
} catch (e: Exception) {
}
return@subscribe ListeningStatus.STOPPED
}
cont.invokeOnCancellation {
runCatching { listener.cancel("nextEvent outer scope cancelled", it) }
}
}
@JvmSynthetic
@PublishedApi
internal suspend inline fun <R> withTimeoutOrCoroutineScope(
timeoutMillis: Long,
useCoroutineScope: CoroutineScope? = null,
noinline block: suspend CoroutineScope.() -> R
): R {
require(timeoutMillis == -1L || timeoutMillis > 0) { "timeoutMillis must be -1 or > 0 " }
return if (timeoutMillis == -1L) {
coroutineScope(block)
if (useCoroutineScope == null) coroutineScope(block)
else block(useCoroutineScope)
} else {
withTimeout(timeoutMillis, block)
}