69 lines
2.1 KiB
Kotlin
69 lines
2.1 KiB
Kotlin
package org.moire.ultrasonic.cache
|
|
|
|
import com.twitter.serial.serializer.SerializationContext
|
|
import com.twitter.serial.serializer.Serializer
|
|
import com.twitter.serial.stream.Serial
|
|
import com.twitter.serial.stream.bytebuffer.ByteBufferSerial
|
|
import java.io.File
|
|
|
|
internal const val STORAGE_DIR_NAME = "persistent_storage"
|
|
|
|
/**
|
|
* Provides access to permanent file based storage.
|
|
*
|
|
* Look at [org.moire.ultrasonic.cache.serializers] package for available [Serializer]s.
|
|
*/
|
|
class PermanentFileStorage(
|
|
private val directories: Directories,
|
|
private val debug: Boolean = false
|
|
) {
|
|
private val serializationContext = object : SerializationContext {
|
|
override fun isDebug(): Boolean = debug
|
|
override fun isRelease(): Boolean = !debug
|
|
}
|
|
|
|
private val serializer: Serial = ByteBufferSerial(serializationContext)
|
|
|
|
/**
|
|
* Stores given [objectToStore] using [name] as a key and [objectSerializer] as serializer.
|
|
*/
|
|
fun <T> store(
|
|
name: String,
|
|
objectToStore: T,
|
|
objectSerializer: Serializer<T>
|
|
) {
|
|
val storeFile = getFile(name)
|
|
if (!storeFile.exists()) storeFile.createNewFile()
|
|
storeFile.writeBytes(serializer.toByteArray(objectToStore, objectSerializer))
|
|
}
|
|
|
|
/**
|
|
* Loads object with [name] key using [objectDeserializer] deserializer.
|
|
*/
|
|
fun <T> load(
|
|
name: String,
|
|
objectDeserializer: Serializer<T>
|
|
): T? {
|
|
val storeFile = getFile(name)
|
|
if (!storeFile.exists()) return null
|
|
|
|
return serializer.fromByteArray(storeFile.readBytes(), objectDeserializer)
|
|
}
|
|
|
|
/**
|
|
* Clear all files in storage.
|
|
*/
|
|
fun clearAll() {
|
|
val storageDir = getStorageDir()
|
|
if (storageDir.exists()) {
|
|
storageDir.listFiles().forEach { if (it.isFile) it.delete() }
|
|
}
|
|
}
|
|
|
|
private fun getFile(name: String) = File(getStorageDir(), "$name.ser")
|
|
|
|
private fun getStorageDir() = File(directories.getInternalDataDir(), STORAGE_DIR_NAME).apply {
|
|
if (!exists()) mkdirs()
|
|
}
|
|
}
|