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

Add ConcurrentComponentStorage.creationStacktrace

Browse Source
This commit is contained in:
Him188 2021-05-10 13:13:58 +08:00
parent f7f4ccf4f5
commit 1d2b2377cf
2 changed files with 13 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
mirai-core/src/commonMain/kotlin/network/component/ConcurrentComponentStorage.kt
View File

@ -17,7 +17,9 @@ import kotlin.LazyThreadSafetyMode.NONE
* A thread-safe implementation of [MutableComponentStorage] * A thread-safe implementation of [MutableComponentStorage]
*/ */
internal class ConcurrentComponentStorage( internal class ConcurrentComponentStorage(
private val showAllComponents: Boolean = SHOW_ALL_COMPONENTS private val showAllComponents: Boolean = SHOW_ALL_COMPONENTS,
private val creationStacktrace: Exception? =
if (SHOW_COMPONENTS_CREATION_STACKTRACE) Exception("Creation stacktrace") else null,
) : ComponentStorage, MutableComponentStorage { ) : ComponentStorage, MutableComponentStorage {
private val map = ConcurrentHashMap<ComponentKey<*>, Any?>() private val map = ConcurrentHashMap<ComponentKey<*>, Any?>()
@ -25,7 +27,8 @@ internal class ConcurrentComponentStorage(
override val size: Int get() = map.size override val size: Int get() = map.size
override operator fun <T : Any> get(key: ComponentKey<T>): T { override operator fun <T : Any> get(key: ComponentKey<T>): T {
return getOrNull(key) ?: throw NoSuchComponentException(key, this) return getOrNull(key)
?: throw NoSuchComponentException(key, this).apply { creationStacktrace?.let(::initCause) }
} }
override fun <T : Any> getOrNull(key: ComponentKey<T>): T? { override fun <T : Any> getOrNull(key: ComponentKey<T>): T? {
@ -56,4 +59,10 @@ internal class ConcurrentComponentStorage(
} }
} }
private val SHOW_ALL_COMPONENTS: Boolean by lazy(NONE) { systemProp("mirai.debug.network.show.all.components", false) } private val SHOW_ALL_COMPONENTS: Boolean by lazy(NONE) { systemProp("mirai.debug.network.show.all.components", false) }
private val SHOW_COMPONENTS_CREATION_STACKTRACE: Boolean by lazy(NONE) {
systemProp(
"mirai.debug.network.show.components.creation.stacktrace",
false
)
}

1
mirai-core/src/commonTest/kotlin/network/framework/AbstractRealNetworkHandlerTest.kt
View File

@ -44,6 +44,7 @@ internal abstract class AbstractRealNetworkHandlerTest<H : NetworkHandler> : Abs
init { init {
System.setProperty("mirai.debug.network.state.observer.logging", "full") System.setProperty("mirai.debug.network.state.observer.logging", "full")
System.setProperty("mirai.debug.network.show.all.components", "true") System.setProperty("mirai.debug.network.show.all.components", "true")
System.setProperty("mirai.debug.network.show.components.creation.stacktrace", "true")
} }
protected abstract val factory: NetworkHandlerFactory<H> protected abstract val factory: NetworkHandlerFactory<H>