98 lines
3.4 KiB
Kotlin
98 lines
3.4 KiB
Kotlin
package dev.inmo.micro_utils.repos
|
|
|
|
import dev.inmo.micro_utils.pagination.Pagination
|
|
import dev.inmo.micro_utils.pagination.PaginationResult
|
|
import dev.inmo.micro_utils.pagination.utils.paginate
|
|
import dev.inmo.micro_utils.pagination.utils.reverse
|
|
import kotlinx.coroutines.flow.Flow
|
|
import kotlinx.coroutines.flow.MutableSharedFlow
|
|
|
|
class ReadMapKeyValueRepo<Key, Value>(
|
|
protected val map: Map<Key, Value> = emptyMap()
|
|
) : ReadKeyValueRepo<Key, Value> {
|
|
override suspend fun get(k: Key): Value? = map[k]
|
|
|
|
override suspend fun values(
|
|
pagination: Pagination,
|
|
reversed: Boolean
|
|
): PaginationResult<Value> {
|
|
val values = map.values
|
|
val actualPagination = if (reversed) pagination.reverse(values.size) else pagination
|
|
return values.paginate(actualPagination).let {
|
|
if (reversed) {
|
|
it.copy(results = it.results.reversed())
|
|
} else {
|
|
it
|
|
}
|
|
}
|
|
}
|
|
|
|
override suspend fun keys(pagination: Pagination, reversed: Boolean): PaginationResult<Key> {
|
|
val keys = map.keys
|
|
val actualPagination = if (reversed) pagination.reverse(keys.size) else pagination
|
|
return keys.paginate(actualPagination).let {
|
|
if (reversed) {
|
|
it.copy(results = it.results.reversed())
|
|
} else {
|
|
it
|
|
}
|
|
}
|
|
}
|
|
|
|
override suspend fun keys(v: Value, pagination: Pagination, reversed: Boolean): PaginationResult<Key> {
|
|
val keys: List<Key> = map.mapNotNull { (k, value) -> if (v == value) k else null }
|
|
val actualPagination = if (reversed) pagination.reverse(keys.size) else pagination
|
|
return keys.paginate(actualPagination).let {
|
|
if (reversed) {
|
|
it.copy(results = it.results.reversed())
|
|
} else {
|
|
it
|
|
}
|
|
}
|
|
}
|
|
|
|
override suspend fun contains(key: Key): Boolean = map.containsKey(key)
|
|
|
|
override suspend fun count(): Long = map.size.toLong()
|
|
}
|
|
|
|
class WriteMapKeyValueRepo<Key, Value>(
|
|
private val map: MutableMap<Key, Value> = mutableMapOf()
|
|
) : WriteKeyValueRepo<Key, Value> {
|
|
private val _onNewValue: MutableSharedFlow<Pair<Key, Value>> = MutableSharedFlow()
|
|
override val onNewValue: Flow<Pair<Key, Value>>
|
|
get() = _onNewValue
|
|
private val _onValueRemoved: MutableSharedFlow<Key> = MutableSharedFlow()
|
|
override val onValueRemoved: Flow<Key>
|
|
get() = _onValueRemoved
|
|
|
|
override suspend fun set(toSet: Map<Key, Value>) {
|
|
map.putAll(toSet)
|
|
toSet.forEach { (k, v) -> _onNewValue.emit(k to v) }
|
|
}
|
|
|
|
override suspend fun unset(toUnset: List<Key>) {
|
|
toUnset.forEach { k ->
|
|
map.remove(k) ?.also { _ -> _onValueRemoved.emit(k) }
|
|
}
|
|
}
|
|
|
|
override suspend fun unsetWithValues(toUnset: List<Value>) {
|
|
map.mapNotNull { (k, v) ->
|
|
k.takeIf { v in toUnset }
|
|
}.forEach {
|
|
map.remove(it)
|
|
_onValueRemoved.emit(it)
|
|
}
|
|
}
|
|
}
|
|
|
|
@Suppress("DELEGATED_MEMBER_HIDES_SUPERTYPE_OVERRIDE")
|
|
class MapKeyValueRepo<Key, Value>(
|
|
private val map: MutableMap<Key, Value> = mutableMapOf()
|
|
) : KeyValueRepo<Key, Value>,
|
|
ReadKeyValueRepo<Key, Value> by ReadMapKeyValueRepo(map),
|
|
WriteKeyValueRepo<Key, Value> by WriteMapKeyValueRepo(map)
|
|
|
|
fun <K, V> MutableMap<K, V>.asKeyValueRepo(): KeyValueRepo<K, V> = MapKeyValueRepo(this)
|