commonMain.network.component.ConcurrentComponentStorage.kt Maven / Gradle / Ivy

Go to download
/*
 * Copyright 2019-2021 Mamoe Technologies and contributors.
 *
 *  此源代码的使用受 GNU AFFERO GENERAL PUBLIC LICENSE version 3 许可证的约束, 可以在以下链接找到该许可证.
 *  Use of this source code is governed by the GNU AGPLv3 license that can be found through the following link.
 *
 *  https://github.com/mamoe/mirai/blob/master/LICENSE
 */

package net.mamoe.mirai.internal.network.component

import net.mamoe.mirai.internal.network.component.ComponentKey.Companion.componentName
import net.mamoe.mirai.utils.systemProp
import java.util.concurrent.ConcurrentHashMap
import kotlin.LazyThreadSafetyMode.NONE

/**
 * A thread-safe implementation of [MutableComponentStorage]
 */
internal class ConcurrentComponentStorage(
    private val showAllComponents: Boolean = SHOW_ALL_COMPONENTS,
    private val creationStacktrace: Exception? =
        if (SHOW_COMPONENTS_CREATION_STACKTRACE) Exception("ConcurrentComponentStorage Creation stacktrace") else null,
) : ComponentStorage, MutableComponentStorage {
    private val map = ConcurrentHashMap, Any?>()

    override val keys: Set> get() = map.keys
    override val size: Int get() = map.size

    override operator fun  get(key: ComponentKey): T {
        return getOrNull(key)
            ?: throw NoSuchComponentException(key, this).apply { creationStacktrace?.let(this::initCause) }
    }

    override fun  getOrNull(key: ComponentKey): T? {
        @Suppress("UNCHECKED_CAST")
        return map[key] as T?
    }

    override operator fun  set(key: ComponentKey, value: T) {
        map[key] = value
    }

    override fun  remove(key: ComponentKey): T? {
        @Suppress("UNCHECKED_CAST")
        return map.remove(key) as T?
    }

    override fun toString(): String {
        if (showAllComponents) {
            return buildString {
                append("ConcurrentComponentStorage(size=").append(map.size).append(") {").appendLine()
                for ((key, value) in map) {
                    append("  ").append(key.componentName(qualified = false)).append(": ").append(value).appendLine()
                }
                append('}')
            }
        }
        return "ConcurrentComponentStorage(size=${map.size})"
    }
}

internal inline fun ConcurrentComponentStorage(builderAction: ConcurrentComponentStorage.() -> Unit): ConcurrentComponentStorage {
    return ConcurrentComponentStorage().apply(builderAction)
}

private val SHOW_ALL_COMPONENTS: Boolean by lazy(NONE) { systemProp("mirai.network.show.all.components", false) }
private val SHOW_COMPONENTS_CREATION_STACKTRACE: Boolean by lazy(NONE) {
    systemProp(
        "mirai.network.show.components.creation.stacktrace",
        false
    )
}