ImageCache.swift 33 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752
  1. //
  2. // ImageCache.swift
  3. // Kingfisher
  4. //
  5. // Created by Wei Wang on 15/4/6.
  6. //
  7. // Copyright (c) 2018 Wei Wang <onevcat@gmail.com>
  8. //
  9. // Permission is hereby granted, free of charge, to any person obtaining a copy
  10. // of this software and associated documentation files (the "Software"), to deal
  11. // in the Software without restriction, including without limitation the rights
  12. // to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
  13. // copies of the Software, and to permit persons to whom the Software is
  14. // furnished to do so, subject to the following conditions:
  15. //
  16. // The above copyright notice and this permission notice shall be included in
  17. // all copies or substantial portions of the Software.
  18. //
  19. // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  20. // IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  21. // FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
  22. // AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  23. // LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
  24. // OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
  25. // THE SOFTWARE.
  26. #if os(macOS)
  27. import AppKit
  28. #else
  29. import UIKit
  30. #endif
  31. extension Notification.Name {
  32. /// This notification will be sent when the disk cache got cleaned either there are cached files expired or the
  33. /// total size exceeding the max allowed size. The manually invoking of `clearDiskCache` method will not trigger
  34. /// this notification.
  35. ///
  36. /// The `object` of this notification is the `ImageCache` object which sends the notification.
  37. /// A list of removed hashes (files) could be retrieved by accessing the array under
  38. /// `KingfisherDiskCacheCleanedHashKey` key in `userInfo` of the notification object you received.
  39. /// By checking the array, you could know the hash codes of files are removed.
  40. public static let KingfisherDidCleanDiskCache =
  41. Notification.Name("com.onevcat.Kingfisher.KingfisherDidCleanDiskCache")
  42. }
  43. /// Key for array of cleaned hashes in `userInfo` of `KingfisherDidCleanDiskCacheNotification`.
  44. public let KingfisherDiskCacheCleanedHashKey = "com.onevcat.Kingfisher.cleanedHash"
  45. /// Cache type of a cached image.
  46. /// - none: The image is not cached yet when retrieving it.
  47. /// - memory: The image is cached in memory.
  48. /// - disk: The image is cached in disk.
  49. public enum CacheType {
  50. /// The image is not cached yet when retrieving it.
  51. case none
  52. /// The image is cached in memory.
  53. case memory
  54. /// The image is cached in disk.
  55. case disk
  56. /// Whether the cache type represents the image is already cached or not.
  57. public var cached: Bool {
  58. switch self {
  59. case .memory, .disk: return true
  60. case .none: return false
  61. }
  62. }
  63. }
  64. /// Represents the caching operation result.
  65. public struct CacheStoreResult {
  66. /// The cache result for memory cache. Caching an image to memory will never fail.
  67. public let memoryCacheResult: Result<(), Never>
  68. /// The cache result for disk cache. If an error happens during caching operation,
  69. /// you can get it from `.failure` case of this `diskCacheResult`.
  70. public let diskCacheResult: Result<(), KingfisherError>
  71. }
  72. extension Image: CacheCostCalculatable {
  73. /// Cost of an image
  74. public var cacheCost: Int { return kf.cost }
  75. }
  76. extension Data: DataTransformable {
  77. public func toData() throws -> Data {
  78. return self
  79. }
  80. public static func fromData(_ data: Data) throws -> Data {
  81. return data
  82. }
  83. public static let empty = Data()
  84. }
  85. /// Represents the getting image operation from the cache.
  86. ///
  87. /// - disk: The image can be retrieved from disk cache.
  88. /// - memory: The image can be retrieved memory cache.
  89. /// - none: The image does not exist in the cache.
  90. public enum ImageCacheResult {
  91. /// The image can be retrieved from disk cache.
  92. case disk(Image)
  93. /// The image can be retrieved memory cache.
  94. case memory(Image)
  95. /// The image does not exist in the cache.
  96. case none
  97. /// Extracts the image from cache result. It returns the associated `Image` value for
  98. /// `.disk` and `.memory` case. For `.none` case, `nil` is returned.
  99. public var image: Image? {
  100. switch self {
  101. case .disk(let image): return image
  102. case .memory(let image): return image
  103. case .none: return nil
  104. }
  105. }
  106. /// Returns the corresponding `CacheType` value based on the result type of `self`.
  107. public var cacheType: CacheType {
  108. switch self {
  109. case .disk: return .disk
  110. case .memory: return .memory
  111. case .none: return .none
  112. }
  113. }
  114. }
  115. /// Represents a hybrid caching system which is composed by a `MemoryStorage.Backend` and a `DiskStorage.Backend`.
  116. /// `ImageCache` is a high level abstract for storing an image as well as its data to disk memory and disk, and
  117. /// retrieving them back.
  118. ///
  119. /// While a default image cache object will be used if you prefer the extension methods of Kingfisher, you can create
  120. /// your own cache object and configure its storages as your need. This class also provide an interface for you to set
  121. /// the memory and disk storage config.
  122. open class ImageCache {
  123. /// The `MemoryStorage.Backend` object used in this cache. This storage holds loaded images in memory with a
  124. /// reasonable expire duration and a maximum memory usage. To modify the configuration of a storage, just set
  125. /// the storage `config` and its properties.
  126. public let memoryStorage: MemoryStorage.Backend<Image>
  127. /// The `DiskStorage.Backend` object used in this cache. This storage stores loaded images in disk with a
  128. /// reasonable expire duration and a maximum disk usage. To modify the configuration of a storage, just set
  129. /// the storage `config` and its properties.
  130. public let diskStorage: DiskStorage.Backend<Data>
  131. private let ioQueue: DispatchQueue
  132. /// The default `ImageCache` object. Kingfisher will use this cache for its related methods if there is no
  133. /// other cache specified. The `name` of this default cache is "default", and you should not use this name
  134. /// for any of your customize cache.
  135. public static let `default` = ImageCache(name: "default")
  136. /// Closure that defines the disk cache path from a given path and cacheName.
  137. public typealias DiskCachePathClosure = (URL, String) -> URL
  138. /// Creates an `ImageCache` from a customized `MemoryStorage` and `DiskStorage`.
  139. ///
  140. /// - Parameters:
  141. /// - memoryStorage: The `MemoryStorage` object to use in the image cache.
  142. /// - diskStorage: The `DiskStorage.Backend` object to use in the image cache.
  143. /// - name: A name used as a part of the bound IO queue.
  144. public init(
  145. memoryStorage: MemoryStorage.Backend<Image>,
  146. diskStorage: DiskStorage.Backend<Data>,
  147. name: String = "")
  148. {
  149. self.memoryStorage = memoryStorage
  150. self.diskStorage = diskStorage
  151. var ioQueueName = "com.onevcat.Kingfisher.ImageCache.ioQueue"
  152. if !name.isEmpty {
  153. ioQueueName.append(".\(name)")
  154. }
  155. ioQueue = DispatchQueue(label: ioQueueName)
  156. #if !os(macOS) && !os(watchOS)
  157. let notifications: [(Notification.Name, Selector)] = [
  158. (UIApplication.didReceiveMemoryWarningNotification, #selector(clearMemoryCache)),
  159. (UIApplication.willTerminateNotification, #selector(cleanExpiredDiskCache)),
  160. (UIApplication.didEnterBackgroundNotification, #selector(backgroundCleanExpiredDiskCache))
  161. ]
  162. notifications.forEach {
  163. NotificationCenter.default.addObserver(self, selector: $0.1, name: $0.0, object: nil)
  164. }
  165. #endif
  166. }
  167. /// Creates an `ImageCache` with a given `name`. Both `MemoryStorage` and `DiskStorage` will be created
  168. /// with a default config based on the `name`.
  169. ///
  170. /// - Parameter name: The name of cache object. It is used to setup disk cache directories and IO queue.
  171. /// You should not use the same `name` for different caches, otherwise, the disk storage would
  172. /// be conflicting to each other. The `name` should not be an empty string.
  173. public convenience init(name: String) {
  174. try! self.init(name: name, path: nil, diskCachePathClosure: nil)
  175. }
  176. /// Creates an `ImageCache` with a given `name`, cache directory `path`
  177. /// and a closure to modify the cache directory.
  178. ///
  179. /// - Parameters:
  180. /// - name: The name of cache object. It is used to setup disk cache directories and IO queue.
  181. /// You should not use the same `name` for different caches, otherwise, the disk storage would
  182. /// be conflicting to each other.
  183. /// - path: Location of cache path on disk. It will be internally pass to the initializer of `DiskStorage` as the
  184. /// disk cache directory.
  185. /// - diskCachePathClosure: Closure that takes in an optional initial path string and generates
  186. /// the final disk cache path. You could use it to fully customize your cache path.
  187. /// - Throws: An error that happens during image cache creating, such as unable to create a directory at the given
  188. /// path.
  189. public convenience init(
  190. name: String,
  191. path: String?,
  192. diskCachePathClosure: DiskCachePathClosure? = nil) throws
  193. {
  194. if name.isEmpty {
  195. fatalError("[Kingfisher] You should specify a name for the cache. A cache with empty name is not permitted.")
  196. }
  197. let totalMemory = ProcessInfo.processInfo.physicalMemory
  198. let costLimit = totalMemory / 4
  199. let memoryStorage = MemoryStorage.Backend<Image>(config:
  200. .init(totalCostLimit: (costLimit > Int.max) ? Int.max : Int(costLimit)))
  201. var diskConfig = DiskStorage.Config(
  202. name: name,
  203. sizeLimit: 0,
  204. directory: path.flatMap { URL(string: $0) }
  205. )
  206. if let closure = diskCachePathClosure {
  207. diskConfig.cachePathBlock = closure
  208. }
  209. let diskStorage = try DiskStorage.Backend<Data>(config: diskConfig)
  210. diskConfig.cachePathBlock = nil
  211. self.init(memoryStorage: memoryStorage, diskStorage: diskStorage, name: name)
  212. }
  213. deinit {
  214. NotificationCenter.default.removeObserver(self)
  215. }
  216. // MARK: - Store & Remove
  217. /// Stores an image to the cache.
  218. ///
  219. /// - Parameters:
  220. /// - image: The image to be stored.
  221. /// - original: The original data of the image. This value will be forwarded to the provided `serializer` for
  222. /// further use. By default, Kingfisher uses a `DefaultCacheSerializer` to serialize the image to
  223. /// data for caching in disk, it checks the image format based on `original` data to determine in
  224. /// which image format should be used. For other types of `serializer`, it depends on thier
  225. /// implemetation detail on how to use this original data.
  226. /// - key: The key used for caching the image.
  227. /// - identifier: The identifier of processor being used for caching. If you are using a processor for the
  228. /// image, pass the identifier of processor to this parameter.
  229. /// - serializer: The `CacheSerializer`
  230. /// - toDisk: Whether this image should be cached to disk or not. If `false`, the image is only cached in memory.
  231. /// Otherwise, it is cached in both memory storage and disk storage. Default is `true`.
  232. /// - callbackQueue: The callback queue on which `completionHandler` is invoked. Default is `.untouch`. For case
  233. /// that `toDisk` is `false`, a `.untouch` queue means `callbackQueue` will be invoked from the
  234. /// caller queue of this method. If `toDisk` is `true`, the `completionHandler` will be called
  235. /// from an internal file IO queue. To change this behavior, specify another `CallbackQueue`
  236. /// value.
  237. /// - completionHandler: A closure which is invoked when the cache operation finishes.
  238. open func store(_ image: Image,
  239. original: Data? = nil,
  240. forKey key: String,
  241. processorIdentifier identifier: String = "",
  242. cacheSerializer serializer: CacheSerializer = DefaultCacheSerializer.default,
  243. toDisk: Bool = true,
  244. callbackQueue: CallbackQueue = .untouch,
  245. completionHandler: ((CacheStoreResult) -> Void)? = nil)
  246. {
  247. let computedKey = key.computedKey(with: identifier)
  248. // Memory storage should not throw.
  249. memoryStorage.storeNoThrow(value: image, forKey: computedKey)
  250. guard toDisk else {
  251. if let completionHandler = completionHandler {
  252. let result = CacheStoreResult(memoryCacheResult: .success(()), diskCacheResult: .success(()))
  253. callbackQueue.execute { completionHandler(result) }
  254. }
  255. return
  256. }
  257. ioQueue.async {
  258. if let data = serializer.data(with: image, original: original) {
  259. self.syncStoreData(
  260. data,
  261. forKey: key,
  262. processorIdentifier: identifier,
  263. callbackQueue: callbackQueue,
  264. completionHandler: completionHandler)
  265. } else {
  266. guard let completionHandler = completionHandler else { return }
  267. let diskError = KingfisherError.cacheError(
  268. reason: .cannotSerializeImage(image: image, original: original, serializer: serializer))
  269. let result = CacheStoreResult(
  270. memoryCacheResult: .success(()),
  271. diskCacheResult: .failure(diskError))
  272. callbackQueue.execute { completionHandler(result) }
  273. }
  274. }
  275. }
  276. open func storeToDisk(
  277. _ data: Data,
  278. forKey key: String,
  279. processorIdentifier identifier: String = "",
  280. callbackQueue: CallbackQueue = .untouch,
  281. completionHandler: ((CacheStoreResult) -> Void)? = nil)
  282. {
  283. ioQueue.async {
  284. self.syncStoreData(
  285. data,
  286. forKey: key,
  287. processorIdentifier: identifier,
  288. callbackQueue: callbackQueue,
  289. completionHandler: completionHandler)
  290. }
  291. }
  292. private func syncStoreData(
  293. _ data: Data,
  294. forKey key: String,
  295. processorIdentifier identifier: String = "",
  296. callbackQueue: CallbackQueue = .untouch,
  297. completionHandler: ((CacheStoreResult) -> Void)? = nil)
  298. {
  299. let computedKey = key.computedKey(with: identifier)
  300. let result: CacheStoreResult
  301. do {
  302. try self.diskStorage.store(value: data, forKey: computedKey)
  303. result = CacheStoreResult(memoryCacheResult: .success(()), diskCacheResult: .success(()))
  304. } catch {
  305. let diskError: KingfisherError
  306. if let error = error as? KingfisherError {
  307. diskError = error
  308. } else {
  309. diskError = .cacheError(reason: .cannotConvertToData(object: data, error: error))
  310. }
  311. result = CacheStoreResult(
  312. memoryCacheResult: .success(()),
  313. diskCacheResult: .failure(diskError)
  314. )
  315. }
  316. if let completionHandler = completionHandler {
  317. callbackQueue.execute { completionHandler(result) }
  318. }
  319. }
  320. /// Removes the image for the given key from the cache.
  321. ///
  322. /// - Parameters:
  323. /// - key: The key used for caching the image.
  324. /// - identifier: The identifier of processor being used for caching. If you are using a processor for the
  325. /// image, pass the identifier of processor to this parameter.
  326. /// - fromMemory: Whether this image should be removed from memory storage or not.
  327. /// If `false`, the image won't be removed from the memory storage. Default is `true`.
  328. /// - fromDisk: Whether this image should be removed from disk storage or not.
  329. /// If `false`, the image won't be removed from the disk storage. Default is `true`.
  330. /// - callbackQueue: The callback queue on which `completionHandler` is invoked. Default is `.untouch`.
  331. /// - completionHandler: A closure which is invoked when the cache removing operation finishes.
  332. open func removeImage(forKey key: String,
  333. processorIdentifier identifier: String = "",
  334. fromMemory: Bool = true,
  335. fromDisk: Bool = true,
  336. callbackQueue: CallbackQueue = .untouch,
  337. completionHandler: (() -> Void)? = nil)
  338. {
  339. let computedKey = key.computedKey(with: identifier)
  340. if fromMemory {
  341. try? memoryStorage.remove(forKey: computedKey)
  342. }
  343. if fromDisk {
  344. ioQueue.async{
  345. try? self.diskStorage.remove(forKey: computedKey)
  346. if let completionHandler = completionHandler {
  347. callbackQueue.execute { completionHandler() }
  348. }
  349. }
  350. } else {
  351. if let completionHandler = completionHandler {
  352. callbackQueue.execute { completionHandler() }
  353. }
  354. }
  355. }
  356. func retrieveImage(forKey key: String,
  357. options: KingfisherParsedOptionsInfo,
  358. callbackQueue: CallbackQueue = .untouch,
  359. completionHandler: ((Result<ImageCacheResult, KingfisherError>) -> Void)?)
  360. {
  361. // No completion handler. No need to start working and early return.
  362. guard let completionHandler = completionHandler else { return }
  363. let imageModifier = options.imageModifier
  364. // Try to check the image from memory cache first.
  365. if let image = retrieveImageInMemoryCache(forKey: key, options: options) {
  366. let image = imageModifier.modify(image)
  367. callbackQueue.execute { completionHandler(.success(.memory(image))) }
  368. } else if options.fromMemoryCacheOrRefresh {
  369. callbackQueue.execute { completionHandler(.success(.none)) }
  370. } else {
  371. // Begin to disk search.
  372. self.retrieveImageInDiskCache(forKey: key, options: options, callbackQueue: callbackQueue) {
  373. result in
  374. // The callback queue is already correct in this closure.
  375. switch result {
  376. case .success(let image):
  377. guard let image = imageModifier.modify(image) else {
  378. // No image found in disk storage.
  379. completionHandler(.success(.none))
  380. return
  381. }
  382. // Cache the disk image to memory.
  383. // We are passing `false` to `toDisk`, the memory cache does not change
  384. // callback queue, we can call `completionHandler` without another dispatch.
  385. self.store(
  386. image,
  387. forKey: key,
  388. processorIdentifier: options.processor.identifier,
  389. cacheSerializer: options.cacheSerializer,
  390. toDisk: false)
  391. {
  392. _ in
  393. completionHandler(.success(.disk(image)))
  394. }
  395. case .failure(let error):
  396. completionHandler(.failure(error))
  397. }
  398. }
  399. }
  400. }
  401. /// Gets an image for a given key from the cache, either from memory storage or disk storage.
  402. ///
  403. /// - Parameters:
  404. /// - key: The key used for caching the image.
  405. /// - options: The `KingfisherOptionsInfo` options setting used for retrieving the image.
  406. /// - callbackQueue: The callback queue on which `completionHandler` is invoked. Default is `.untouch`.
  407. /// - completionHandler: A closure which is invoked when the image getting operation finishes. If the
  408. /// image retrieving operation finishes without problem, an `ImageCacheResult` value
  409. /// will be sent to this closuer as result. Otherwise, a `KingfisherError` result
  410. /// with detail failing reason will be sent.
  411. open func retrieveImage(forKey key: String,
  412. options: KingfisherOptionsInfo? = nil,
  413. callbackQueue: CallbackQueue = .untouch,
  414. completionHandler: ((Result<ImageCacheResult, KingfisherError>) -> Void)?)
  415. {
  416. retrieveImage(
  417. forKey: key,
  418. options: KingfisherParsedOptionsInfo(options),
  419. callbackQueue: callbackQueue,
  420. completionHandler: completionHandler)
  421. }
  422. func retrieveImageInMemoryCache(
  423. forKey key: String,
  424. options: KingfisherParsedOptionsInfo) -> Image?
  425. {
  426. let computedKey = key.computedKey(with: options.processor.identifier)
  427. do {
  428. return try memoryStorage.value(forKey: computedKey)
  429. } catch {
  430. return nil
  431. }
  432. }
  433. /// Gets an image for a given key from the memory storage.
  434. ///
  435. /// - Parameters:
  436. /// - key: The key used for caching the image.
  437. /// - options: The `KingfisherOptionsInfo` options setting used for retrieving the image.
  438. /// - Returns: The image stored in memory cache, if exists and valid. Otherwise, if the image does not exist or
  439. /// has already expired, `nil` is returned.
  440. open func retrieveImageInMemoryCache(
  441. forKey key: String,
  442. options: KingfisherOptionsInfo? = nil) -> Image?
  443. {
  444. return retrieveImageInMemoryCache(forKey: key, options: KingfisherParsedOptionsInfo(options))
  445. }
  446. func retrieveImageInDiskCache(
  447. forKey key: String,
  448. options: KingfisherParsedOptionsInfo,
  449. callbackQueue: CallbackQueue = .untouch,
  450. completionHandler: @escaping (Result<Image?, KingfisherError>) -> Void)
  451. {
  452. let computedKey = key.computedKey(with: options.processor.identifier)
  453. let loadingQueue: CallbackQueue = options.loadDiskFileSynchronously ? .untouch : .dispatch(ioQueue)
  454. loadingQueue.execute {
  455. do {
  456. var image: Image? = nil
  457. if let data = try self.diskStorage.value(forKey: computedKey) {
  458. image = options.cacheSerializer.image(with: data, options: options)
  459. }
  460. callbackQueue.execute { completionHandler(.success(image)) }
  461. } catch {
  462. if let error = error as? KingfisherError {
  463. callbackQueue.execute { completionHandler(.failure(error)) }
  464. } else {
  465. assertionFailure("The internal thrown error should be a `KingfisherError`.")
  466. }
  467. }
  468. }
  469. }
  470. /// Gets an image for a given key from the disk storage.
  471. ///
  472. /// - Parameters:
  473. /// - key: The key used for caching the image.
  474. /// - options: The `KingfisherOptionsInfo` options setting used for retrieving the image.
  475. /// - callbackQueue: The callback queue on which `completionHandler` is invoked. Default is `.untouch`.
  476. /// - completionHandler: A closure which is invoked when the operation finishes.
  477. open func retrieveImageInDiskCache(
  478. forKey key: String,
  479. options: KingfisherOptionsInfo? = nil,
  480. callbackQueue: CallbackQueue = .untouch,
  481. completionHandler: @escaping (Result<Image?, KingfisherError>) -> Void)
  482. {
  483. retrieveImageInDiskCache(
  484. forKey: key,
  485. options: KingfisherParsedOptionsInfo(options),
  486. callbackQueue: callbackQueue,
  487. completionHandler: completionHandler)
  488. }
  489. // MARK: - Clear & Clean
  490. /// Clears the memory storage of this cache.
  491. @objc public func clearMemoryCache() {
  492. try? memoryStorage.removeAll()
  493. }
  494. /// Clears the disk storage of this cache. This is an async operation.
  495. ///
  496. /// - Parameter handler: A closure which is invoked when the cache clearing operation finishes.
  497. /// This `handler` will be called from the main queue.
  498. open func clearDiskCache(completion handler: (()->())? = nil) {
  499. ioQueue.async {
  500. do {
  501. try self.diskStorage.removeAll()
  502. } catch _ { }
  503. if let handler = handler {
  504. DispatchQueue.main.async { handler() }
  505. }
  506. }
  507. }
  508. /// Clears the expired images from disk storage. This is an async operation.
  509. @objc func cleanExpiredDiskCache() {
  510. cleanExpiredDiskCache(completion: nil)
  511. }
  512. /// Clears the expired images from disk storage. This is an async operation.
  513. ///
  514. /// - Parameter handler: A closure which is invoked when the cache clearing operation finishes.
  515. /// This `handler` will be called from the main queue.
  516. open func cleanExpiredDiskCache(completion handler: (() -> Void)? = nil) {
  517. ioQueue.async {
  518. do {
  519. var removed: [URL] = []
  520. let removedExpired = try self.diskStorage.removeExpiredValues()
  521. removed.append(contentsOf: removedExpired)
  522. let removedSizeExceeded = try self.diskStorage.removeSizeExceededValues()
  523. removed.append(contentsOf: removedSizeExceeded)
  524. if !removed.isEmpty {
  525. DispatchQueue.main.async {
  526. let cleanedHashes = removed.map { $0.lastPathComponent }
  527. NotificationCenter.default.post(
  528. name: .KingfisherDidCleanDiskCache,
  529. object: self,
  530. userInfo: [KingfisherDiskCacheCleanedHashKey: cleanedHashes])
  531. }
  532. }
  533. if let handler = handler {
  534. DispatchQueue.main.async { handler() }
  535. }
  536. } catch {}
  537. }
  538. }
  539. #if !os(macOS) && !os(watchOS)
  540. /// Clears the expired images from disk storage when app is in background. This is an async operation.
  541. /// In most cases, you should not call this method explicitly.
  542. /// It will be called automatically when `UIApplicationDidEnterBackgroundNotification` received.
  543. @objc public func backgroundCleanExpiredDiskCache() {
  544. // if 'sharedApplication()' is unavailable, then return
  545. guard let sharedApplication = KingfisherWrapper<UIApplication>.shared else { return }
  546. func endBackgroundTask(_ task: inout UIBackgroundTaskIdentifier) {
  547. sharedApplication.endBackgroundTask(task)
  548. task = UIBackgroundTaskIdentifier.invalid
  549. }
  550. var backgroundTask: UIBackgroundTaskIdentifier!
  551. backgroundTask = sharedApplication.beginBackgroundTask {
  552. endBackgroundTask(&backgroundTask!)
  553. }
  554. cleanExpiredDiskCache {
  555. endBackgroundTask(&backgroundTask!)
  556. }
  557. }
  558. #endif
  559. /// Returns the cache type for a given `key` and `identifier` combination.
  560. /// This method is used for checking whether an image is cached in current cache.
  561. /// It also provides information on which kind of cache can it be found in the return value.
  562. ///
  563. /// - Parameters:
  564. /// - key: The key used for caching the image.
  565. /// - identifier: Processor identifier which used for this image. Default is the `identifier` of
  566. /// `DefaultImageProcessor.default`.
  567. /// - Returns: A `CacheType` instance which indicates the cache status.
  568. /// `.none` means the image is not in cache or it is already expired.
  569. open func imageCachedType(
  570. forKey key: String,
  571. processorIdentifier identifier: String = DefaultImageProcessor.default.identifier) -> CacheType
  572. {
  573. let computedKey = key.computedKey(with: identifier)
  574. if memoryStorage.isCached(forKey: computedKey) { return .memory }
  575. if diskStorage.isCached(forKey: computedKey) { return .disk }
  576. return .none
  577. }
  578. /// Returns whether the file exists in cache for a given `key` and `identifier` combination.
  579. ///
  580. /// - Parameters:
  581. /// - key: The key used for caching the image.
  582. /// - identifier: Processor identifier which used for this image. Default is the `identifier` of
  583. /// `DefaultImageProcessor.default`.
  584. /// - Returns: A `Bool` which indicates whether a cache could match the given `key` and `identifier` combination.
  585. ///
  586. /// - Note:
  587. /// The return value does not contain information about from which kind of storage the cache matches.
  588. /// To get the information about cache type according `CacheType`,
  589. /// use `imageCachedType(forKey:processorIdentifier:)` instead.
  590. public func isCached(
  591. forKey key: String,
  592. processorIdentifier identifier: String = DefaultImageProcessor.default.identifier) -> Bool
  593. {
  594. return imageCachedType(forKey: key, processorIdentifier: identifier).cached
  595. }
  596. /// Gets the hash used as cache file name for the key.
  597. ///
  598. /// - Parameters:
  599. /// - key: The key used for caching the image.
  600. /// - identifier: Processor identifier which used for this image. Default is the `identifier` of
  601. /// `DefaultImageProcessor.default`.
  602. /// - Returns: The hash which is used as the cache file name.
  603. ///
  604. /// - Note:
  605. /// By default, for a given combination of `key` and `identifier`, `ImageCache` will use the value
  606. /// returned by this method as the cache file name. You can use this value to check and match cache file
  607. /// if you need.
  608. open func hash(
  609. forKey key: String,
  610. processorIdentifier identifier: String = DefaultImageProcessor.default.identifier) -> String
  611. {
  612. let computedKey = key.computedKey(with: identifier)
  613. return diskStorage.cacheFileName(forKey: computedKey)
  614. }
  615. /// Calculates the size taken by the disk storage.
  616. /// It is the total file size of all cached files in the `diskStorage` on disk in bytes.
  617. ///
  618. /// - Parameter handler: Called with the size calculating finishes. This closure is invoked from the main queue.
  619. open func calculateDiskStorageSize(completion handler: @escaping ((Result<UInt, KingfisherError>) -> Void)) {
  620. ioQueue.async {
  621. do {
  622. let size = try self.diskStorage.totalSize()
  623. DispatchQueue.main.async { handler(.success(size)) }
  624. } catch {
  625. if let error = error as? KingfisherError {
  626. DispatchQueue.main.async { handler(.failure(error)) }
  627. } else {
  628. assertionFailure("The internal thrown error should be a `KingfisherError`.")
  629. }
  630. }
  631. }
  632. }
  633. /// Gets the cache path for the key.
  634. /// It is useful for projects with web view or anyone that needs access to the local file path.
  635. ///
  636. /// i.e. Replacing the `<img src='path_for_key'>` tag in your HTML.
  637. ///
  638. /// - Parameters:
  639. /// - key: The key used for caching the image.
  640. /// - identifier: Processor identifier which used for this image. Default is the `identifier` of
  641. /// `DefaultImageProcessor.default`.
  642. /// - Returns: The disk path of cached image under the given `key` and `identifier`.
  643. ///
  644. /// - Note:
  645. /// This method does not guarantee there is an image already cached in the returned path. It just gives your
  646. /// the path that the image should be, if it exists in disk storage.
  647. ///
  648. /// You could use `isImageCached(forKey:)` method to check whether the image is cached under that key in disk.
  649. open func cachePath(
  650. forKey key: String,
  651. processorIdentifier identifier: String = DefaultImageProcessor.default.identifier) -> String
  652. {
  653. let computedKey = key.computedKey(with: identifier)
  654. return diskStorage.cacheFileURL(forKey: computedKey).path
  655. }
  656. }
  657. extension Dictionary {
  658. func keysSortedByValue(_ isOrderedBefore: (Value, Value) -> Bool) -> [Key] {
  659. return Array(self).sorted{ isOrderedBefore($0.1, $1.1) }.map{ $0.0 }
  660. }
  661. }
  662. #if !os(macOS) && !os(watchOS)
  663. // MARK: - For App Extensions
  664. extension UIApplication: KingfisherCompatible { }
  665. extension KingfisherWrapper where Base: UIApplication {
  666. public static var shared: UIApplication? {
  667. let selector = NSSelectorFromString("sharedApplication")
  668. guard Base.responds(to: selector) else { return nil }
  669. return Base.perform(selector).takeUnretainedValue() as? UIApplication
  670. }
  671. }
  672. #endif
  673. extension String {
  674. func computedKey(with identifier: String) -> String {
  675. if identifier.isEmpty {
  676. return self
  677. } else {
  678. return appending("@\(identifier)")
  679. }
  680. }
  681. }