87 lines
2.4 KiB
Swift
87 lines
2.4 KiB
Swift
//
|
|
// ImageDataCache.swift
|
|
// Tusker
|
|
//
|
|
// Created by Shadowfacts on 1/16/21.
|
|
// Copyright © 2021 Shadowfacts. All rights reserved.
|
|
//
|
|
|
|
import UIKit
|
|
import Cache
|
|
|
|
class ImageDataCache {
|
|
|
|
private let memory: MemoryStorage<Entry>
|
|
private let disk: DiskStorage<Data>?
|
|
|
|
private let storeOriginalDataInMemory: Bool
|
|
|
|
init(name: String, memoryExpiry: Expiry, diskExpiry: Expiry?, storeOriginalDataInMemory: Bool) {
|
|
let memoryConfig = MemoryConfig(expiry: memoryExpiry)
|
|
self.memory = MemoryStorage(config: memoryConfig)
|
|
|
|
if let diskExpiry = diskExpiry {
|
|
let diskConfig = DiskConfig(name: name, expiry: diskExpiry)
|
|
self.disk = try! DiskStorage(config: diskConfig, transformer: TransformerFactory.forData())
|
|
} else {
|
|
self.disk = nil
|
|
}
|
|
|
|
self.storeOriginalDataInMemory = storeOriginalDataInMemory
|
|
}
|
|
|
|
func has(_ key: String) throws -> Bool {
|
|
if try memory.existsObject(forKey: key) {
|
|
return true
|
|
} else if let disk = self.disk,
|
|
try disk.existsObject(forKey: key) {
|
|
return true
|
|
} else {
|
|
return false
|
|
}
|
|
}
|
|
|
|
func get(_ key: String) throws -> Entry? {
|
|
if try memory.existsObject(forKey: key) {
|
|
return try! memory.object(forKey: key)
|
|
} else if let disk = self.disk,
|
|
try disk.existsObject(forKey: key),
|
|
let data = try? disk.object(forKey: key),
|
|
let image = UIImage(data: data) {
|
|
return Entry(data: data, image: image)
|
|
} else {
|
|
return nil
|
|
}
|
|
}
|
|
|
|
func getImage(_ key: String) throws -> UIImage? {
|
|
return try get(key)?.image
|
|
}
|
|
|
|
func getData(_ key: String) throws -> Data? {
|
|
return try get(key)?.data
|
|
}
|
|
|
|
func set(_ key: String, data: Data, image: UIImage) throws {
|
|
let entry = Entry(data: storeOriginalDataInMemory ? data : nil, image: image)
|
|
memory.setObject(entry, forKey: key)
|
|
|
|
if let disk = self.disk {
|
|
try disk.setObject(data, forKey: key)
|
|
}
|
|
}
|
|
|
|
func removeAll() throws {
|
|
memory.removeAll()
|
|
try? disk?.removeAll()
|
|
}
|
|
|
|
}
|
|
|
|
extension ImageDataCache {
|
|
struct Entry {
|
|
let data: Data?
|
|
let image: UIImage
|
|
}
|
|
}
|