|
|
|
@ -15,45 +15,45 @@ class ImageCache {
|
|
|
|
|
static let headers = ImageCache(name: "Headers", memoryExpiry: .seconds(60 * 60), diskExpiry: .seconds(60 * 60 * 24))
|
|
|
|
|
static let attachments = ImageCache(name: "Attachments", memoryExpiry: .seconds(60 * 2))
|
|
|
|
|
|
|
|
|
|
let cache: Cache<UIImage>
|
|
|
|
|
let cache: Cache<Data>
|
|
|
|
|
|
|
|
|
|
var requests = [URL: Request]()
|
|
|
|
|
|
|
|
|
|
init(name: String, memoryExpiry expiry: Expiry) {
|
|
|
|
|
let storage = MemoryStorage<UIImage>(config: MemoryConfig(expiry: expiry))
|
|
|
|
|
let storage = MemoryStorage<Data>(config: MemoryConfig(expiry: expiry))
|
|
|
|
|
self.cache = .memory(storage)
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
init(name: String, diskExpiry expiry: Expiry) {
|
|
|
|
|
let storage = try! DiskStorage<UIImage>(config: DiskConfig(name: name, expiry: expiry), transformer: TransformerFactory.forImage())
|
|
|
|
|
let storage = try! DiskStorage<Data>(config: DiskConfig(name: name, expiry: expiry), transformer: TransformerFactory.forData())
|
|
|
|
|
self.cache = .disk(storage)
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
init(name: String, memoryExpiry: Expiry, diskExpiry: Expiry) {
|
|
|
|
|
let memory = MemoryStorage<UIImage>(config: MemoryConfig(expiry: memoryExpiry))
|
|
|
|
|
let disk = try! DiskStorage<UIImage>(config: DiskConfig(name: name, expiry: diskExpiry), transformer: TransformerFactory.forImage())
|
|
|
|
|
let memory = MemoryStorage<Data>(config: MemoryConfig(expiry: memoryExpiry))
|
|
|
|
|
let disk = try! DiskStorage<Data>(config: DiskConfig(name: name, expiry: diskExpiry), transformer: TransformerFactory.forData())
|
|
|
|
|
self.cache = .hybrid(HybridStorage(memoryStorage: memory, diskStorage: disk))
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
func get(_ url: URL, completion: ((UIImage?) -> Void)?) {
|
|
|
|
|
func get(_ url: URL, completion: ((Data?) -> Void)?) {
|
|
|
|
|
let key = url.absoluteString
|
|
|
|
|
if (try? cache.existsObject(forKey: key)) ?? false,
|
|
|
|
|
let image = try? cache.object(forKey: key) {
|
|
|
|
|
completion?(image)
|
|
|
|
|
let data = try? cache.object(forKey: key) {
|
|
|
|
|
completion?(data)
|
|
|
|
|
} else {
|
|
|
|
|
if let completion = completion, let request = requests[url] {
|
|
|
|
|
request.callbacks.append(completion)
|
|
|
|
|
} else {
|
|
|
|
|
let request = Request(url: url, completion: completion)
|
|
|
|
|
requests[url] = request
|
|
|
|
|
request.run { (image) in
|
|
|
|
|
try? self.cache.setObject(image, forKey: key)
|
|
|
|
|
request.run { (data) in
|
|
|
|
|
try? self.cache.setObject(data, forKey: key)
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
func get(_ url: URL) -> UIImage? {
|
|
|
|
|
func get(_ url: URL) -> Data? {
|
|
|
|
|
return try? cache.object(forKey: url.absoluteString)
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
@ -64,9 +64,9 @@ class ImageCache {
|
|
|
|
|
class Request {
|
|
|
|
|
let url: URL
|
|
|
|
|
var task: URLSessionDataTask?
|
|
|
|
|
var callbacks: [(UIImage?) -> Void]
|
|
|
|
|
var callbacks: [(Data?) -> Void]
|
|
|
|
|
|
|
|
|
|
init(url: URL, completion: ((UIImage?) -> Void)?) {
|
|
|
|
|
init(url: URL, completion: ((Data?) -> Void)?) {
|
|
|
|
|
if let completion = completion {
|
|
|
|
|
self.callbacks = [completion]
|
|
|
|
|
} else {
|
|
|
|
@ -75,14 +75,14 @@ class ImageCache {
|
|
|
|
|
self.url = url
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
func run(cache: @escaping (UIImage) -> Void) {
|
|
|
|
|
func run(cache: @escaping (Data) -> Void) {
|
|
|
|
|
task = URLSession.shared.dataTask(with: url, completionHandler: { (data, response, error) in
|
|
|
|
|
guard error == nil, let data = data, let image = UIImage(data: data) else {
|
|
|
|
|
guard error == nil, let data = data else {
|
|
|
|
|
self.complete(with: nil)
|
|
|
|
|
return
|
|
|
|
|
}
|
|
|
|
|
cache(image)
|
|
|
|
|
self.complete(with: image)
|
|
|
|
|
cache(data)
|
|
|
|
|
self.complete(with: data)
|
|
|
|
|
})
|
|
|
|
|
task!.resume()
|
|
|
|
|
}
|
|
|
|
@ -92,8 +92,8 @@ class ImageCache {
|
|
|
|
|
complete(with: nil)
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
func complete(with image: UIImage?) {
|
|
|
|
|
callbacks.forEach { $0(image) }
|
|
|
|
|
func complete(with data: Data?) {
|
|
|
|
|
callbacks.forEach { $0(data) }
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|