92 lines
3.8 KiB
Kotlin
92 lines
3.8 KiB
Kotlin
package dev.inmo.tgbotapi.libraries.cache.admins.micro_utils
|
|
|
|
import com.soywiz.klock.DateTime
|
|
import dev.inmo.micro_utils.coroutines.*
|
|
import dev.inmo.micro_utils.repos.*
|
|
import dev.inmo.tgbotapi.libraries.cache.admins.DefaultAdminsCacheAPIRepo
|
|
import dev.inmo.tgbotapi.types.*
|
|
import dev.inmo.tgbotapi.types.chat.member.AdministratorChatMember
|
|
import kotlinx.coroutines.CompletableDeferred
|
|
import kotlinx.coroutines.CoroutineScope
|
|
import kotlinx.coroutines.channels.*
|
|
import kotlin.coroutines.*
|
|
|
|
private sealed class RepoActions<T> {
|
|
abstract val deferred: CompletableDeferred<T>
|
|
}
|
|
private class GetUpdateDateTimeRepoAction(
|
|
val chatId: IdChatIdentifier,
|
|
override val deferred: CompletableDeferred<DateTime?>
|
|
) : RepoActions<DateTime?>()
|
|
private class GetChatAdminsRepoAction(
|
|
val chatId: IdChatIdentifier,
|
|
override val deferred: CompletableDeferred<List<AdministratorChatMember>?>
|
|
) : RepoActions<List<AdministratorChatMember>?>()
|
|
private class SetChatAdminsRepoAction(
|
|
val chatId: IdChatIdentifier,
|
|
val newValue: List<AdministratorChatMember>,
|
|
override val deferred: CompletableDeferred<Unit>
|
|
) : RepoActions<Unit>()
|
|
|
|
class DefaultAdminsCacheAPIRepoImpl(
|
|
private val adminsRepo: KeyValuesRepo<IdChatIdentifier, AdministratorChatMember>,
|
|
private val updatesRepo: KeyValueRepo<IdChatIdentifier, MilliSeconds>,
|
|
private val scope: CoroutineScope
|
|
) : DefaultAdminsCacheAPIRepo {
|
|
private val actor = scope.actorAsync<RepoActions<*>>(Channel.UNLIMITED) {
|
|
safelyWithoutExceptions(
|
|
{ e ->
|
|
it.deferred.completeExceptionally(e)
|
|
}
|
|
) {
|
|
when (it) {
|
|
is GetUpdateDateTimeRepoAction -> it.deferred.complete(
|
|
updatesRepo.get(it.chatId) ?.let { DateTime(it.toDouble()) }
|
|
)
|
|
is GetChatAdminsRepoAction -> it.deferred.complete(adminsRepo.getAll(it.chatId))
|
|
is SetChatAdminsRepoAction -> {
|
|
adminsRepo.clear(it.chatId)
|
|
adminsRepo.set(it.chatId, it.newValue)
|
|
updatesRepo.set(it.chatId, DateTime.now().unixMillisLong)
|
|
it.deferred.complete(Unit)
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
override suspend fun getChatAdmins(chatId: IdChatIdentifier): List<AdministratorChatMember>? {
|
|
val deferred = CompletableDeferred<List<AdministratorChatMember>?>()
|
|
actor.trySend(
|
|
GetChatAdminsRepoAction(chatId, deferred)
|
|
).onFailure {
|
|
deferred.completeExceptionally(it ?: IllegalStateException("Something went wrong when tried to add getChatAdmins action"))
|
|
}
|
|
return deferred.await()
|
|
}
|
|
|
|
override suspend fun setChatAdmins(chatId: IdChatIdentifier, chatMembers: List<AdministratorChatMember>) {
|
|
val deferred = CompletableDeferred<Unit>()
|
|
actor.trySend(
|
|
SetChatAdminsRepoAction(chatId, chatMembers, deferred)
|
|
).onFailure {
|
|
deferred.completeExceptionally(it ?: IllegalStateException("Something went wrong when tried to add setChatAdmins action"))
|
|
}
|
|
return deferred.await()
|
|
}
|
|
override suspend fun lastUpdate(chatId: IdChatIdentifier): DateTime? {
|
|
val deferred = CompletableDeferred<DateTime?>()
|
|
actor.trySend(
|
|
GetUpdateDateTimeRepoAction(chatId, deferred)
|
|
).onFailure {
|
|
deferred.completeExceptionally(it ?: IllegalStateException("Something went wrong when tried to add lastUpdate action"))
|
|
}
|
|
return deferred.await()
|
|
}
|
|
}
|
|
|
|
fun DefaultAdminsCacheAPIRepo(
|
|
adminsRepo: KeyValuesRepo<IdChatIdentifier, AdministratorChatMember>,
|
|
updatesRepo: KeyValueRepo<IdChatIdentifier, MilliSeconds>,
|
|
scope: CoroutineScope
|
|
) = DefaultAdminsCacheAPIRepoImpl(adminsRepo, updatesRepo, scope)
|