mirror of
https://github.com/InsanusMokrassar/MicroUtils.git
synced 2024-06-02 07:55:13 +00:00
70 lines
2.4 KiB
Kotlin
70 lines
2.4 KiB
Kotlin
package dev.inmo.micro_utils.repos.exposed.onetomany
|
|
|
|
import dev.inmo.micro_utils.repos.KeyValuesRepo
|
|
import kotlinx.coroutines.flow.*
|
|
import org.jetbrains.exposed.sql.*
|
|
import org.jetbrains.exposed.sql.statements.InsertStatement
|
|
import org.jetbrains.exposed.sql.transactions.transaction
|
|
|
|
abstract class AbstractExposedKeyValuesRepo<Key, Value>(
|
|
override val database: Database,
|
|
tableName: String? = null
|
|
) : KeyValuesRepo<Key, Value>, AbstractExposedReadKeyValuesRepo<Key, Value>(
|
|
database,
|
|
tableName
|
|
) {
|
|
protected val _onNewValue: MutableSharedFlow<Pair<Key, Value>> = MutableSharedFlow()
|
|
override val onNewValue: Flow<Pair<Key, Value>>
|
|
get() = _onNewValue
|
|
protected val _onValueRemoved: MutableSharedFlow<Pair<Key, Value>> = MutableSharedFlow()
|
|
override val onValueRemoved: Flow<Pair<Key, Value>>
|
|
get() = _onValueRemoved
|
|
protected val _onDataCleared: MutableSharedFlow<Key> = MutableSharedFlow()
|
|
override val onDataCleared: Flow<Key>
|
|
get() = _onDataCleared
|
|
|
|
protected abstract fun insert(k: Key, v: Value, it: InsertStatement<Number>)
|
|
|
|
override suspend fun add(toAdd: Map<Key, List<Value>>) {
|
|
transaction(database) {
|
|
toAdd.keys.flatMap { k ->
|
|
toAdd[k] ?.mapNotNull { v ->
|
|
if (select { selectById(k).and(selectByValue(v)) }.limit(1).any()) {
|
|
return@mapNotNull null
|
|
}
|
|
val insertResult = insert {
|
|
insert(k, v, it)
|
|
}
|
|
if (insertResult.insertedCount > 0) {
|
|
k to v
|
|
} else {
|
|
null
|
|
}
|
|
} ?: emptyList()
|
|
}
|
|
}.forEach { _onNewValue.emit(it) }
|
|
}
|
|
|
|
override suspend fun remove(toRemove: Map<Key, List<Value>>) {
|
|
transaction(database) {
|
|
toRemove.keys.flatMap { k ->
|
|
toRemove[k] ?.mapNotNull { v ->
|
|
if (deleteWhere { selectById(k).and(selectByValue(v)) } > 0 ) {
|
|
k to v
|
|
} else {
|
|
null
|
|
}
|
|
} ?: emptyList()
|
|
}
|
|
}.forEach {
|
|
_onValueRemoved.emit(it)
|
|
}
|
|
}
|
|
|
|
override suspend fun clear(k: Key) {
|
|
transaction(database) {
|
|
deleteWhere { selectById(k) }
|
|
}.also { _onDataCleared.emit(k) }
|
|
}
|
|
}
|