mirror of
https://github.com/InsanusMokrassar/TelegramBotApiLibraries.git
synced 2024-11-24 19:18:47 +00:00
Compare commits
No commits in common. "d5ae880e77b203305687e20d86e0a03989f5eabf" and "6327366720dc205178831cc0370a53e83778775d" have entirely different histories.
d5ae880e77
...
6327366720
@ -1,7 +1,6 @@
|
|||||||
package dev.inmo.tgbotapi.libraries.cache.media.common
|
package dev.inmo.tgbotapi.libraries.cache.media.common
|
||||||
|
|
||||||
import dev.inmo.tgbotapi.bot.TelegramBot
|
import dev.inmo.tgbotapi.bot.TelegramBot
|
||||||
import dev.inmo.tgbotapi.requests.DeleteMessage
|
|
||||||
import dev.inmo.tgbotapi.requests.DownloadFileStream
|
import dev.inmo.tgbotapi.requests.DownloadFileStream
|
||||||
import dev.inmo.tgbotapi.requests.abstracts.MultipartFile
|
import dev.inmo.tgbotapi.requests.abstracts.MultipartFile
|
||||||
import dev.inmo.tgbotapi.requests.get.GetFile
|
import dev.inmo.tgbotapi.requests.get.GetFile
|
||||||
@ -17,9 +16,6 @@ class DefaultMessageContentCache<K>(
|
|||||||
private val bot: TelegramBot,
|
private val bot: TelegramBot,
|
||||||
private val filesRefreshingChatId: ChatId,
|
private val filesRefreshingChatId: ChatId,
|
||||||
private val simpleMessageContentCache: MessagesSimpleCache<K>,
|
private val simpleMessageContentCache: MessagesSimpleCache<K>,
|
||||||
private val mediaFileActualityChecker: MediaFileActualityChecker = MediaFileActualityChecker.WithDelay(
|
|
||||||
MediaFileActualityChecker.Default(filesRefreshingChatId)
|
|
||||||
),
|
|
||||||
private val messagesFilesCache: MessagesFilesCache<K> = InMemoryMessagesFilesCache()
|
private val messagesFilesCache: MessagesFilesCache<K> = InMemoryMessagesFilesCache()
|
||||||
) : MessageContentCache<K> {
|
) : MessageContentCache<K> {
|
||||||
override suspend fun save(content: MessageContent): K {
|
override suspend fun save(content: MessageContent): K {
|
||||||
@ -52,10 +48,6 @@ class DefaultMessageContentCache<K>(
|
|||||||
messagesFilesCache.set(key, filename, inputAllocator)
|
messagesFilesCache.set(key, filename, inputAllocator)
|
||||||
}.onFailure {
|
}.onFailure {
|
||||||
simpleMessageContentCache.remove(key)
|
simpleMessageContentCache.remove(key)
|
||||||
}.onSuccess {
|
|
||||||
with(mediaFileActualityChecker) {
|
|
||||||
bot.saved(content)
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
return key
|
return key
|
||||||
@ -64,7 +56,10 @@ class DefaultMessageContentCache<K>(
|
|||||||
override suspend fun get(k: K): MessageContent? {
|
override suspend fun get(k: K): MessageContent? {
|
||||||
val savedSimpleContent = simpleMessageContentCache.get(k) ?: return null
|
val savedSimpleContent = simpleMessageContentCache.get(k) ?: return null
|
||||||
|
|
||||||
if (savedSimpleContent is MediaContent && !with(mediaFileActualityChecker) { bot.isActual(savedSimpleContent) }) {
|
if (savedSimpleContent is MediaContent) {
|
||||||
|
runCatching {
|
||||||
|
bot.execute(GetFile(savedSimpleContent.media.fileId))
|
||||||
|
}.onFailure {
|
||||||
val savedFileContentAllocator = messagesFilesCache.get(k) ?: error("Unexpected absence of $k file for content ($simpleMessageContentCache)")
|
val savedFileContentAllocator = messagesFilesCache.get(k) ?: error("Unexpected absence of $k file for content ($simpleMessageContentCache)")
|
||||||
val newContent = bot.execute(
|
val newContent = bot.execute(
|
||||||
when (savedSimpleContent.asInputMedia()) {
|
when (savedSimpleContent.asInputMedia()) {
|
||||||
@ -109,6 +104,7 @@ class DefaultMessageContentCache<K>(
|
|||||||
simpleMessageContentCache.update(k, newContent.content)
|
simpleMessageContentCache.update(k, newContent.content)
|
||||||
return newContent.content
|
return newContent.content
|
||||||
}
|
}
|
||||||
|
}
|
||||||
return savedSimpleContent
|
return savedSimpleContent
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -126,10 +122,7 @@ class DefaultMessageContentCache<K>(
|
|||||||
bot: TelegramBot,
|
bot: TelegramBot,
|
||||||
filesRefreshingChatId: ChatId,
|
filesRefreshingChatId: ChatId,
|
||||||
simpleMessageContentCache: MessagesSimpleCache<String> = InMemoryMessagesSimpleCache(),
|
simpleMessageContentCache: MessagesSimpleCache<String> = InMemoryMessagesSimpleCache(),
|
||||||
mediaFileActualityChecker: MediaFileActualityChecker = MediaFileActualityChecker.WithDelay(
|
|
||||||
MediaFileActualityChecker.Default(filesRefreshingChatId)
|
|
||||||
),
|
|
||||||
messagesFilesCache: MessagesFilesCache<String> = InMemoryMessagesFilesCache()
|
messagesFilesCache: MessagesFilesCache<String> = InMemoryMessagesFilesCache()
|
||||||
) = DefaultMessageContentCache(bot, filesRefreshingChatId, simpleMessageContentCache, mediaFileActualityChecker, messagesFilesCache)
|
) = DefaultMessageContentCache(bot, filesRefreshingChatId, simpleMessageContentCache, messagesFilesCache)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
@ -1,55 +0,0 @@
|
|||||||
package dev.inmo.tgbotapi.libraries.cache.media.common
|
|
||||||
|
|
||||||
import com.soywiz.klock.DateTime
|
|
||||||
import com.soywiz.klock.milliseconds
|
|
||||||
import dev.inmo.tgbotapi.bot.TelegramBot
|
|
||||||
import dev.inmo.tgbotapi.requests.DeleteMessage
|
|
||||||
import dev.inmo.tgbotapi.requests.abstracts.FileId
|
|
||||||
import dev.inmo.tgbotapi.types.ChatId
|
|
||||||
import dev.inmo.tgbotapi.types.MilliSeconds
|
|
||||||
import dev.inmo.tgbotapi.types.message.content.abstracts.MediaContent
|
|
||||||
|
|
||||||
fun interface MediaFileActualityChecker {
|
|
||||||
suspend fun TelegramBot.isActual(mediaContent: MediaContent): Boolean
|
|
||||||
suspend fun TelegramBot.saved(mediaContent: MediaContent) {}
|
|
||||||
|
|
||||||
class Default(
|
|
||||||
private val checkingChatId: ChatId
|
|
||||||
) : MediaFileActualityChecker {
|
|
||||||
override suspend fun TelegramBot.isActual(mediaContent: MediaContent): Boolean {
|
|
||||||
return runCatching {
|
|
||||||
execute(mediaContent.createResend(checkingChatId)).also { sentMessage ->
|
|
||||||
execute(DeleteMessage(sentMessage.chat.id, sentMessage.messageId))
|
|
||||||
}
|
|
||||||
}.isSuccess
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
class WithDelay(
|
|
||||||
private val underhoodChecker: MediaFileActualityChecker,
|
|
||||||
private val checkingDelay: MilliSeconds = 24 * 60 * 60 * 1000L // one day
|
|
||||||
) : MediaFileActualityChecker {
|
|
||||||
private val fileIdChecksMap = mutableMapOf<FileId, DateTime>()
|
|
||||||
private val checkingDelayTimeSpan = checkingDelay.milliseconds
|
|
||||||
|
|
||||||
override suspend fun TelegramBot.isActual(mediaContent: MediaContent): Boolean {
|
|
||||||
val now = DateTime.now()
|
|
||||||
val lastCheck = fileIdChecksMap[mediaContent.media.fileId]
|
|
||||||
return if (lastCheck == null || now - lastCheck > checkingDelayTimeSpan) {
|
|
||||||
with(underhoodChecker) {
|
|
||||||
isActual(mediaContent)
|
|
||||||
}.also {
|
|
||||||
if (it) {
|
|
||||||
fileIdChecksMap[mediaContent.media.fileId] = now
|
|
||||||
}
|
|
||||||
}
|
|
||||||
} else {
|
|
||||||
true
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
override suspend fun TelegramBot.saved(mediaContent: MediaContent) {
|
|
||||||
fileIdChecksMap[mediaContent.media.fileId] = DateTime.now()
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
Loading…
Reference in New Issue
Block a user