42 lines
1.2 KiB
Kotlin
42 lines
1.2 KiB
Kotlin
package dev.inmo.tgbotapi.libraries.cache.media.common
|
|
|
|
import dev.inmo.tgbotapi.types.ChatId
|
|
import dev.inmo.tgbotapi.types.MessageIdentifier
|
|
import dev.inmo.tgbotapi.utils.StorageFile
|
|
import io.ktor.utils.io.core.*
|
|
|
|
interface MessagesFilesCache<K> {
|
|
suspend fun set(k: K, filename: String, inputAllocator: suspend () -> Input)
|
|
suspend fun get(k: K): StorageFile?
|
|
suspend fun remove(k: K)
|
|
suspend fun contains(k: K): Boolean
|
|
}
|
|
|
|
/**
|
|
* It is not recommended to use in production realization of [MessagesFilesCache] which has been created for fast
|
|
* start of application creation with usage of [MessageContentCache] with aim to replace this realization by some
|
|
* disks-oriented one
|
|
*/
|
|
class InMemoryMessagesFilesCache<K> : MessagesFilesCache<K> {
|
|
private val map = mutableMapOf<K, StorageFile>()
|
|
|
|
override suspend fun set(k: K, filename: String, inputAllocator: suspend () -> Input) {
|
|
map[k] = StorageFile(
|
|
filename,
|
|
inputAllocator().readBytes()
|
|
)
|
|
}
|
|
|
|
override suspend fun get(k: K): StorageFile? {
|
|
return map[k]
|
|
}
|
|
|
|
override suspend fun remove(k: K) {
|
|
map.remove(k)
|
|
}
|
|
|
|
override suspend fun contains(k: K): Boolean {
|
|
return map.contains(k)
|
|
}
|
|
}
|