ImageCache.swift 31 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685
  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. if toDisk {
  251. ioQueue.async {
  252. let result: CacheStoreResult
  253. if let data = serializer.data(with: image, original: original) {
  254. do {
  255. try self.diskStorage.store(value: data, forKey: computedKey)
  256. result = CacheStoreResult(memoryCacheResult: .success(()), diskCacheResult: .success(()))
  257. } catch {
  258. let diskError: KingfisherError
  259. if let error = error as? KingfisherError {
  260. diskError = error
  261. } else {
  262. diskError = .cacheError(reason: .cannotConvertToData(object: data, error: error))
  263. }
  264. result = CacheStoreResult(
  265. memoryCacheResult: .success(()),
  266. diskCacheResult: .failure(diskError)
  267. )
  268. }
  269. } else {
  270. let diskError = KingfisherError.cacheError(
  271. reason: .cannotSerializeImage(image: image, original: original, serializer: serializer))
  272. result = CacheStoreResult(memoryCacheResult: .success(()), diskCacheResult: .failure(diskError))
  273. }
  274. if let completionHandler = completionHandler {
  275. callbackQueue.execute { completionHandler(result) }
  276. }
  277. }
  278. } else {
  279. if let completionHandler = completionHandler {
  280. let result = CacheStoreResult(memoryCacheResult: .success(()), diskCacheResult: .success(()))
  281. callbackQueue.execute { completionHandler(result) }
  282. }
  283. }
  284. }
  285. /// Removes the image for the given key from the cache.
  286. ///
  287. /// - Parameters:
  288. /// - key: The key used for caching the image.
  289. /// - identifier: The identifier of processor being used for caching. If you are using a processor for the
  290. /// image, pass the identifier of processor to this parameter.
  291. /// - fromMemory: Whether this image should be removed from memory storage or not.
  292. /// If `false`, the image won't be removed from the memory storage. Default is `true`.
  293. /// - fromDisk: Whether this image should be removed from disk storage or not.
  294. /// If `false`, the image won't be removed from the disk storage. Default is `true`.
  295. /// - callbackQueue: The callback queue on which `completionHandler` is invoked. Default is `.untouch`.
  296. /// - completionHandler: A closure which is invoked when the cache removing operation finishes.
  297. open func removeImage(forKey key: String,
  298. processorIdentifier identifier: String = "",
  299. fromMemory: Bool = true,
  300. fromDisk: Bool = true,
  301. callbackQueue: CallbackQueue = .untouch,
  302. completionHandler: (() -> Void)? = nil)
  303. {
  304. let computedKey = key.computedKey(with: identifier)
  305. if fromMemory {
  306. try? memoryStorage.remove(forKey: computedKey)
  307. }
  308. if fromDisk {
  309. ioQueue.async{
  310. try? self.diskStorage.remove(forKey: computedKey)
  311. if let completionHandler = completionHandler {
  312. callbackQueue.execute { completionHandler() }
  313. }
  314. }
  315. } else {
  316. if let completionHandler = completionHandler {
  317. callbackQueue.execute { completionHandler() }
  318. }
  319. }
  320. }
  321. /// Gets an image for a given key from the cache, either from memory storage or disk storage.
  322. ///
  323. /// - Parameters:
  324. /// - key: The key used for caching the image.
  325. /// - options: The `KingfisherOptionsInfo` options setting used for retrieving the image.
  326. /// - callbackQueue: The callback queue on which `completionHandler` is invoked. Default is `.untouch`.
  327. /// - completionHandler: A closure which is invoked when the image getting operation finishes. If the
  328. /// image retrieving operation finishes without problem, an `ImageCacheResult` value
  329. /// will be sent to this closuer as result. Otherwise, a `KingfisherError` result
  330. /// with detail failing reason will be sent.
  331. open func retrieveImage(forKey key: String,
  332. options: KingfisherOptionsInfo? = nil,
  333. callbackQueue: CallbackQueue = .untouch,
  334. completionHandler: ((Result<ImageCacheResult, KingfisherError>) -> Void)?)
  335. {
  336. // No completion handler. No need to start working and early return.
  337. guard let completionHandler = completionHandler else { return }
  338. let options = options ?? .empty
  339. let imageModifier = options.imageModifier
  340. // Try to check the image from memory cache first.
  341. if let image = retrieveImageInMemoryCache(forKey: key, options: options) {
  342. let image = imageModifier.modify(image)
  343. callbackQueue.execute { completionHandler(.success(.memory(image))) }
  344. } else if options.fromMemoryCacheOrRefresh {
  345. callbackQueue.execute { completionHandler(.success(.none)) }
  346. } else {
  347. // Begin to disk search.
  348. self.retrieveImageInDiskCache(forKey: key, options: options, callbackQueue: callbackQueue) {
  349. result in
  350. // The callback queue is already correct in this closure.
  351. switch result {
  352. case .success(let image):
  353. guard let image = imageModifier.modify(image) else {
  354. // No image found in disk storage.
  355. completionHandler(.success(.none))
  356. return
  357. }
  358. // Cache the disk image to memory.
  359. // We are passing `false` to `toDisk`, the memory cache does not change
  360. // callback queue, we can call `completionHandler` without another dispatch.
  361. self.store(
  362. image,
  363. forKey: key,
  364. processorIdentifier: options.processor.identifier,
  365. cacheSerializer: options.cacheSerializer,
  366. toDisk: false)
  367. {
  368. _ in
  369. completionHandler(.success(.disk(image)))
  370. }
  371. case .failure(let error):
  372. completionHandler(.failure(error))
  373. }
  374. }
  375. }
  376. }
  377. /// Gets an image for a given key from the memory storage.
  378. ///
  379. /// - Parameters:
  380. /// - key: The key used for caching the image.
  381. /// - options: The `KingfisherOptionsInfo` options setting used for retrieving the image.
  382. /// - Returns: The image stored in memory cache, if exists and valid. Otherwise, if the image does not exist or
  383. /// has already expired, `nil` is returned.
  384. open func retrieveImageInMemoryCache(
  385. forKey key: String,
  386. options: KingfisherOptionsInfo? = nil) -> Image?
  387. {
  388. let options = options ?? .empty
  389. let computedKey = key.computedKey(with: options.processor.identifier)
  390. do {
  391. return try memoryStorage.value(forKey: computedKey)
  392. } catch {
  393. return nil
  394. }
  395. }
  396. /// Gets an image for a given key from the disk storage.
  397. ///
  398. /// - Parameters:
  399. /// - key: The key used for caching the image.
  400. /// - options: The `KingfisherOptionsInfo` options setting used for retrieving the image.
  401. /// - callbackQueue: The callback queue on which `completionHandler` is invoked. Default is `.untouch`.
  402. /// - completionHandler: A closure which is invoked when the operation finishes.
  403. open func retrieveImageInDiskCache(
  404. forKey key: String,
  405. options: KingfisherOptionsInfo? = nil,
  406. callbackQueue: CallbackQueue = .untouch,
  407. completionHandler: @escaping (Result<Image?, KingfisherError>) -> Void)
  408. {
  409. let options = options ?? .empty
  410. let computedKey = key.computedKey(with: options.processor.identifier)
  411. let loadingQueue: CallbackQueue = options.loadDiskFileSynchronously ? .untouch : .dispatch(ioQueue)
  412. loadingQueue.execute {
  413. do {
  414. var image: Image? = nil
  415. if let data = try self.diskStorage.value(forKey: computedKey) {
  416. image = options.cacheSerializer.image(with: data, options: options)
  417. }
  418. callbackQueue.execute { completionHandler(.success(image)) }
  419. } catch {
  420. if let error = error as? KingfisherError {
  421. callbackQueue.execute { completionHandler(.failure(error)) }
  422. } else {
  423. assertionFailure("The internal thrown error should be a `KingfisherError`.")
  424. }
  425. }
  426. }
  427. }
  428. // MARK: - Clear & Clean
  429. /// Clears the memory storage of this cache.
  430. @objc public func clearMemoryCache() {
  431. try? memoryStorage.removeAll()
  432. }
  433. /// Clears the disk storage of this cache. This is an async operation.
  434. ///
  435. /// - Parameter handler: A closure which is invoked when the cache clearing operation finishes.
  436. /// This `handler` will be called from the main queue.
  437. open func clearDiskCache(completion handler: (()->())? = nil) {
  438. ioQueue.async {
  439. do {
  440. try self.diskStorage.removeAll()
  441. } catch _ { }
  442. if let handler = handler {
  443. DispatchQueue.main.async { handler() }
  444. }
  445. }
  446. }
  447. /// Clears the expired images from disk storage. This is an async operation.
  448. @objc func cleanExpiredDiskCache() {
  449. cleanExpiredDiskCache(completion: nil)
  450. }
  451. /// Clears the expired images from disk storage. This is an async operation.
  452. ///
  453. /// - Parameter handler: A closure which is invoked when the cache clearing operation finishes.
  454. /// This `handler` will be called from the main queue.
  455. open func cleanExpiredDiskCache(completion handler: (() -> Void)? = nil) {
  456. ioQueue.async {
  457. do {
  458. var removed: [URL] = []
  459. let removedExpired = try self.diskStorage.removeExpiredValues()
  460. removed.append(contentsOf: removedExpired)
  461. let removedSizeExceeded = try self.diskStorage.removeSizeExceededValues()
  462. removed.append(contentsOf: removedSizeExceeded)
  463. if !removed.isEmpty {
  464. DispatchQueue.main.async {
  465. let cleanedHashes = removed.map { $0.lastPathComponent }
  466. NotificationCenter.default.post(
  467. name: .KingfisherDidCleanDiskCache,
  468. object: self,
  469. userInfo: [KingfisherDiskCacheCleanedHashKey: cleanedHashes])
  470. }
  471. }
  472. if let handler = handler {
  473. DispatchQueue.main.async { handler() }
  474. }
  475. } catch {}
  476. }
  477. }
  478. #if !os(macOS) && !os(watchOS)
  479. /// Clears the expired images from disk storage when app is in background. This is an async operation.
  480. /// In most cases, you should not call this method explicitly.
  481. /// It will be called automatically when `UIApplicationDidEnterBackgroundNotification` received.
  482. @objc public func backgroundCleanExpiredDiskCache() {
  483. // if 'sharedApplication()' is unavailable, then return
  484. guard let sharedApplication = KingfisherClass<UIApplication>.shared else { return }
  485. func endBackgroundTask(_ task: inout UIBackgroundTaskIdentifier) {
  486. sharedApplication.endBackgroundTask(task)
  487. task = UIBackgroundTaskIdentifier.invalid
  488. }
  489. var backgroundTask: UIBackgroundTaskIdentifier!
  490. backgroundTask = sharedApplication.beginBackgroundTask {
  491. endBackgroundTask(&backgroundTask!)
  492. }
  493. cleanExpiredDiskCache {
  494. endBackgroundTask(&backgroundTask!)
  495. }
  496. }
  497. #endif
  498. /// Returns the cache type for a given `key` and `identifier` combination.
  499. /// This method is used for checking whether an image is cached in current cache.
  500. /// It also provides information on which kind of cache can it be found in the return value.
  501. ///
  502. /// - Parameters:
  503. /// - key: The key used for caching the image.
  504. /// - identifier: Processor identifier which used for this image. Default is the `identifier` of
  505. /// `DefaultImageProcessor.default`.
  506. /// - Returns: A `CacheType` instance which indicates the cache status.
  507. /// `.none` means the image is not in cache or it is already expired.
  508. open func imageCachedType(
  509. forKey key: String,
  510. processorIdentifier identifier: String = DefaultImageProcessor.default.identifier) -> CacheType
  511. {
  512. let computedKey = key.computedKey(with: identifier)
  513. if memoryStorage.isCached(forKey: computedKey) { return .memory }
  514. if diskStorage.isCached(forKey: computedKey) { return .disk }
  515. return .none
  516. }
  517. /// Returns whether the file exists in cache for a given `key` and `identifier` combination.
  518. ///
  519. /// - Parameters:
  520. /// - key: The key used for caching the image.
  521. /// - identifier: Processor identifier which used for this image. Default is the `identifier` of
  522. /// `DefaultImageProcessor.default`.
  523. /// - Returns: A `Bool` which indicates whether a cache could match the given `key` and `identifier` combination.
  524. ///
  525. /// - Note:
  526. /// The return value does not contain information about from which kind of storage the cache matches.
  527. /// To get the information about cache type according `CacheType`,
  528. /// use `imageCachedType(forKey:processorIdentifier:)` instead.
  529. public func isCached(
  530. forKey key: String,
  531. processorIdentifier identifier: String = DefaultImageProcessor.default.identifier) -> Bool
  532. {
  533. return imageCachedType(forKey: key, processorIdentifier: identifier).cached
  534. }
  535. /// Gets the hash used as cache file name for the key.
  536. ///
  537. /// - Parameters:
  538. /// - key: The key used for caching the image.
  539. /// - identifier: Processor identifier which used for this image. Default is the `identifier` of
  540. /// `DefaultImageProcessor.default`.
  541. /// - Returns: The hash which is used as the cache file name.
  542. ///
  543. /// - Note:
  544. /// By default, for a given combination of `key` and `identifier`, `ImageCache` will use the value
  545. /// returned by this method as the cache file name. You can use this value to check and match cache file
  546. /// if you need.
  547. open func hash(
  548. forKey key: String,
  549. processorIdentifier identifier: String = DefaultImageProcessor.default.identifier) -> String
  550. {
  551. let computedKey = key.computedKey(with: identifier)
  552. return diskStorage.cacheFileName(forKey: computedKey)
  553. }
  554. /// Calculates the size taken by the disk storage.
  555. /// It is the total file size of all cached files in the `diskStorage` on disk in bytes.
  556. ///
  557. /// - Parameter handler: Called with the size calculating finishes. This closure is invoked from the main queue.
  558. open func calculateDiskStorageSize(completion handler: @escaping ((Result<UInt, KingfisherError>) -> Void)) {
  559. ioQueue.async {
  560. do {
  561. let size = try self.diskStorage.totalSize()
  562. DispatchQueue.main.async { handler(.success(size)) }
  563. } catch {
  564. if let error = error as? KingfisherError {
  565. DispatchQueue.main.async { handler(.failure(error)) }
  566. } else {
  567. assertionFailure("The internal thrown error should be a `KingfisherError`.")
  568. }
  569. }
  570. }
  571. }
  572. /// Gets the cache path for the key.
  573. /// It is useful for projects with web view or anyone that needs access to the local file path.
  574. ///
  575. /// i.e. Replacing the `<img src='path_for_key'>` tag in your HTML.
  576. ///
  577. /// - Parameters:
  578. /// - key: The key used for caching the image.
  579. /// - identifier: Processor identifier which used for this image. Default is the `identifier` of
  580. /// `DefaultImageProcessor.default`.
  581. /// - Returns: The disk path of cached image under the given `key` and `identifier`.
  582. ///
  583. /// - Note:
  584. /// This method does not guarantee there is an image already cached in the returned path. It just gives your
  585. /// the path that the image should be, if it exists in disk storage.
  586. ///
  587. /// You could use `isImageCached(forKey:)` method to check whether the image is cached under that key in disk.
  588. open func cachePath(
  589. forKey key: String,
  590. processorIdentifier identifier: String = DefaultImageProcessor.default.identifier) -> String
  591. {
  592. let computedKey = key.computedKey(with: identifier)
  593. return diskStorage.cacheFileURL(forKey: computedKey).path
  594. }
  595. }
  596. extension Dictionary {
  597. func keysSortedByValue(_ isOrderedBefore: (Value, Value) -> Bool) -> [Key] {
  598. return Array(self).sorted{ isOrderedBefore($0.1, $1.1) }.map{ $0.0 }
  599. }
  600. }
  601. #if !os(macOS) && !os(watchOS)
  602. // MARK: - For App Extensions
  603. extension UIApplication: KingfisherClassCompatible { }
  604. extension KingfisherClass where Base: UIApplication {
  605. public static var shared: UIApplication? {
  606. let selector = NSSelectorFromString("sharedApplication")
  607. guard Base.responds(to: selector) else { return nil }
  608. return Base.perform(selector).takeUnretainedValue() as? UIApplication
  609. }
  610. }
  611. #endif
  612. extension String {
  613. func computedKey(with identifier: String) -> String {
  614. if identifier.isEmpty {
  615. return self
  616. } else {
  617. return appending("@\(identifier)")
  618. }
  619. }
  620. }