github-repos/mirai
1
0
Fork 0
mirror of https://github.com/mamoe/mirai.git synced 2025-01-19 15:59:12 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add some capability for delegation for TypeSafeMap

Browse Source
This commit is contained in:
Him188 2022-05-21 19:54:38 +01:00
parent 9d03260067
commit c1a7dd9843
1 changed files with 34 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

34
mirai-core-utils/src/commonMain/kotlin/TypeSafeMap.kt
View File

@ -15,6 +15,9 @@ import kotlinx.serialization.Serializable
import java.util.concurrent.ConcurrentHashMap
import kotlin.contracts.InvocationKind
import kotlin.contracts.contract
import kotlin.properties.ReadOnlyProperty
import kotlin.properties.ReadWriteProperty
import kotlin.reflect.KProperty
@Serializable
@JvmInline
@ -37,11 +40,24 @@ public sealed interface TypeSafeMap {
public fun toMapBoxed(): Map<TypeKey<*>, Any>
public fun toMap(): Map<String, Any>
public operator fun <T> provideDelegate(thisRef: Any?, property: KProperty<*>): ReadOnlyProperty<Any?, T> {
val typeKey = TypeKey<T>(property.name)
return ReadOnlyProperty { _, _ -> get(typeKey) }
}
public companion object {
public val EMPTY: TypeSafeMap = TypeSafeMapImpl(emptyMap())
}
}
public fun <T> TypeSafeMap.property(name: String): ReadOnlyProperty<Any?, T> {
return property(TypeKey(name))
}
public fun <T> TypeSafeMap.property(typeKey: TypeKey<T>): ReadOnlyProperty<Any?, T> {
return ReadOnlyProperty { _, _ -> get(typeKey) }
}
public operator fun TypeSafeMap.plus(other: TypeSafeMap): TypeSafeMap {
return when {
other.size == 0 -> this
@ -57,6 +73,24 @@ public sealed interface MutableTypeSafeMap : TypeSafeMap {
public operator fun <T> set(key: TypeKey<T>, value: T)
public fun <T> remove(key: TypeKey<T>): T?
public fun setAll(other: TypeSafeMap)
public override operator fun <T> provideDelegate(
thisRef: Any?,
property: KProperty<*>
): ReadWriteProperty<Any?, T> {
val typeKey = TypeKey<T>(property.name)
return object : ReadWriteProperty<Any?, T> {
override fun getValue(thisRef: Any?, property: KProperty<*>): T {
return get(typeKey)
}
override fun setValue(thisRef: Any?, property: KProperty<*>, value: T) {
set(typeKey, value)
}
}
}
}
private val NULL: Any = Symbol("NULL")!!