73 lines
2.9 KiB
Kotlin
73 lines
2.9 KiB
Kotlin
package dev.inmo.micro_utils.repos.exposed.onetomany
|
|
|
|
import dev.inmo.micro_utils.repos.KeyValuesRepo
|
|
import kotlinx.coroutines.channels.BufferOverflow
|
|
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,
|
|
flowsExtraBufferCapacity: Int = Int.MAX_VALUE,
|
|
flowsBufferOverflow: BufferOverflow = BufferOverflow.SUSPEND
|
|
) : KeyValuesRepo<Key, Value>, AbstractExposedReadKeyValuesRepo<Key, Value>(
|
|
database,
|
|
tableName
|
|
) {
|
|
protected val _onNewValue: MutableSharedFlow<Pair<Key, Value>> = MutableSharedFlow(extraBufferCapacity = flowsExtraBufferCapacity, onBufferOverflow = flowsBufferOverflow)
|
|
override val onNewValue: Flow<Pair<Key, Value>>
|
|
get() = _onNewValue.asSharedFlow()
|
|
protected val _onValueRemoved: MutableSharedFlow<Pair<Key, Value>> = MutableSharedFlow(extraBufferCapacity = flowsExtraBufferCapacity, onBufferOverflow = flowsBufferOverflow)
|
|
override val onValueRemoved: Flow<Pair<Key, Value>>
|
|
get() = _onValueRemoved.asSharedFlow()
|
|
protected val _onDataCleared: MutableSharedFlow<Key> = MutableSharedFlow(extraBufferCapacity = flowsExtraBufferCapacity, onBufferOverflow = flowsBufferOverflow)
|
|
override val onDataCleared: Flow<Key>
|
|
get() = _onDataCleared.asSharedFlow()
|
|
|
|
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 (selectAll().where { 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(it, k).and(SqlExpressionBuilder.selectByValue(v)) } > 0 ) {
|
|
k to v
|
|
} else {
|
|
null
|
|
}
|
|
} ?: emptyList()
|
|
}
|
|
}.forEach {
|
|
_onValueRemoved.emit(it)
|
|
}
|
|
}
|
|
|
|
override suspend fun clear(k: Key) {
|
|
transaction(database) {
|
|
deleteWhere { selectById(it, k) }
|
|
}.also { _onDataCleared.emit(k) }
|
|
}
|
|
}
|