github-repos/mirai
1
0
Fork 0
mirror of https://github.com/mamoe/mirai.git synced 2025-02-10 16:36:56 +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]
*/
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 {
private val map = ConcurrentHashMap<ComponentKey<*>, Any?>()
@ -25,7 +27,8 @@ internal class ConcurrentComponentStorage(
override val size: Int get() = map.size
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? {
@ -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 {
System.setProperty("mirai.debug.network.state.observer.logging", "full")
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>