github-repos/mirai
1
0
Fork 0
mirror of https://github.com/mamoe/mirai.git synced 2025-03-06 17:10:48 +08:00
Code Issues Packages Projects Releases Wiki Activity

EventListeners: use atomic lock to avoid redundant object creation

Browse Source
This commit is contained in:
Him188 2020-02-15 11:04:33 +08:00
parent 8ea7027fc9
commit 2e614f65f9
1 changed files with 17 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
mirai-core/src/commonMain/kotlin/net.mamoe.mirai/event/internal/InternalEventListeners.kt
View File

@ -9,6 +9,7 @@
package net.mamoe.mirai.event.internal
import kotlinx.atomicfu.atomic
import kotlinx.coroutines.*
import net.mamoe.mirai.event.Event
import net.mamoe.mirai.event.EventDisabled
@ -87,14 +88,22 @@ internal object EventListenerManager {
private val registries = LockFreeLinkedList<Registry<*>>()
@Suppress("UNCHECKED_CAST")
internal fun <E : Event> get(clazz: KClass<out E>): EventListeners<E> {
return registries.filteringGetOrAdd({ it.clazz == clazz }) {
Registry(
clazz,
EventListeners()
)
}.listeners as EventListeners<E>
private val lock = atomic(false)
@Suppress("UNCHECKED_CAST", "BooleanLiteralArgument")
internal tailrec fun <E : Event> get(clazz: KClass<out E>): EventListeners<E> {
registries.forEach {
if (it.clazz == clazz) {
return it.listeners as EventListeners<E>
}
}
if (lock.compareAndSet(false, true)) {
val registry = Registry(clazz, EventListeners())
registries.addLast(registry)
lock.value = false
return registry.listeners as EventListeners<E>
}
return get(clazz)
}
}