ImageCache.swift 39 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882
  1. //
  2. // ImageCache.swift
  3. // Kingfisher
  4. //
  5. // Created by Wei Wang on 15/4/6.
  6. //
  7. // Copyright (c) 2019 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 KFCrossPlatformImage: CacheCostCalculable {
  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(KFCrossPlatformImage)
  93. /// The image can be retrieved memory cache.
  94. case memory(KFCrossPlatformImage)
  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: KFCrossPlatformImage? {
  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 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. // MARK: Singleton
  124. /// The default `ImageCache` object. Kingfisher will use this cache for its related methods if there is no
  125. /// other cache specified. The `name` of this default cache is "default", and you should not use this name
  126. /// for any of your customize cache.
  127. public static let `default` = ImageCache(name: "default")
  128. // MARK: Public Properties
  129. /// The `MemoryStorage.Backend` object used in this cache. This storage holds loaded images in memory with a
  130. /// reasonable expire duration and a maximum memory usage. To modify the configuration of a storage, just set
  131. /// the storage `config` and its properties.
  132. public let memoryStorage: MemoryStorage.Backend<KFCrossPlatformImage>
  133. /// The `DiskStorage.Backend` object used in this cache. This storage stores loaded images in disk with a
  134. /// reasonable expire duration and a maximum disk usage. To modify the configuration of a storage, just set
  135. /// the storage `config` and its properties.
  136. public let diskStorage: DiskStorage.Backend<Data>
  137. private let ioQueue: DispatchQueue
  138. /// Closure that defines the disk cache path from a given path and cacheName.
  139. public typealias DiskCachePathClosure = (URL, String) -> URL
  140. // MARK: Initializers
  141. /// Creates an `ImageCache` from a customized `MemoryStorage` and `DiskStorage`.
  142. ///
  143. /// - Parameters:
  144. /// - memoryStorage: The `MemoryStorage.Backend` object to use in the image cache.
  145. /// - diskStorage: The `DiskStorage.Backend` object to use in the image cache.
  146. public init(
  147. memoryStorage: MemoryStorage.Backend<KFCrossPlatformImage>,
  148. diskStorage: DiskStorage.Backend<Data>)
  149. {
  150. self.memoryStorage = memoryStorage
  151. self.diskStorage = diskStorage
  152. let ioQueueName = "com.onevcat.Kingfisher.ImageCache.ioQueue.\(UUID().uuidString)"
  153. ioQueue = DispatchQueue(label: ioQueueName)
  154. let notifications: [(Notification.Name, Selector)]
  155. #if !os(macOS) && !os(watchOS)
  156. notifications = [
  157. (UIApplication.didReceiveMemoryWarningNotification, #selector(clearMemoryCache)),
  158. (UIApplication.willTerminateNotification, #selector(cleanExpiredDiskCache)),
  159. (UIApplication.didEnterBackgroundNotification, #selector(backgroundCleanExpiredDiskCache))
  160. ]
  161. #elseif os(macOS)
  162. notifications = [
  163. (NSApplication.willResignActiveNotification, #selector(cleanExpiredDiskCache)),
  164. ]
  165. #else
  166. notifications = []
  167. #endif
  168. notifications.forEach {
  169. NotificationCenter.default.addObserver(self, selector: $0.1, name: $0.0, object: nil)
  170. }
  171. }
  172. /// Creates an `ImageCache` with a given `name`. Both `MemoryStorage` and `DiskStorage` will be created
  173. /// with a default config based on the `name`.
  174. ///
  175. /// - Parameter name: The name of cache object. It is used to setup disk cache directories and IO queue.
  176. /// You should not use the same `name` for different caches, otherwise, the disk storage would
  177. /// be conflicting to each other. The `name` should not be an empty string.
  178. public convenience init(name: String) {
  179. self.init(noThrowName: name, cacheDirectoryURL: nil, diskCachePathClosure: nil)
  180. }
  181. /// Creates an `ImageCache` with a given `name`, cache directory `path`
  182. /// and a closure to modify the cache directory.
  183. ///
  184. /// - Parameters:
  185. /// - name: The name of cache object. It is used to setup disk cache directories and IO queue.
  186. /// You should not use the same `name` for different caches, otherwise, the disk storage would
  187. /// be conflicting to each other.
  188. /// - cacheDirectoryURL: Location of cache directory URL on disk. It will be internally pass to the
  189. /// initializer of `DiskStorage` as the disk cache directory. If `nil`, the cache
  190. /// directory under user domain mask will be used.
  191. /// - diskCachePathClosure: Closure that takes in an optional initial path string and generates
  192. /// the final disk cache path. You could use it to fully customize your cache path.
  193. /// - Throws: An error that happens during image cache creating, such as unable to create a directory at the given
  194. /// path.
  195. public convenience init(
  196. name: String,
  197. cacheDirectoryURL: URL?,
  198. diskCachePathClosure: DiskCachePathClosure? = nil
  199. ) throws
  200. {
  201. if name.isEmpty {
  202. fatalError("[Kingfisher] You should specify a name for the cache. A cache with empty name is not permitted.")
  203. }
  204. let memoryStorage = ImageCache.createMemoryStorage()
  205. let config = ImageCache.createConfig(
  206. name: name, cacheDirectoryURL: cacheDirectoryURL, diskCachePathClosure: diskCachePathClosure
  207. )
  208. let diskStorage = try DiskStorage.Backend<Data>(config: config)
  209. self.init(memoryStorage: memoryStorage, diskStorage: diskStorage)
  210. }
  211. convenience init(
  212. noThrowName name: String,
  213. cacheDirectoryURL: URL?,
  214. diskCachePathClosure: DiskCachePathClosure?
  215. )
  216. {
  217. if name.isEmpty {
  218. fatalError("[Kingfisher] You should specify a name for the cache. A cache with empty name is not permitted.")
  219. }
  220. let memoryStorage = ImageCache.createMemoryStorage()
  221. let config = ImageCache.createConfig(
  222. name: name, cacheDirectoryURL: cacheDirectoryURL, diskCachePathClosure: diskCachePathClosure
  223. )
  224. let diskStorage = DiskStorage.Backend<Data>(noThrowConfig: config, creatingDirectory: true)
  225. self.init(memoryStorage: memoryStorage, diskStorage: diskStorage)
  226. }
  227. private static func createMemoryStorage() -> MemoryStorage.Backend<KFCrossPlatformImage> {
  228. let totalMemory = ProcessInfo.processInfo.physicalMemory
  229. let costLimit = totalMemory / 4
  230. let memoryStorage = MemoryStorage.Backend<KFCrossPlatformImage>(config:
  231. .init(totalCostLimit: (costLimit > Int.max) ? Int.max : Int(costLimit)))
  232. return memoryStorage
  233. }
  234. private static func createConfig(
  235. name: String,
  236. cacheDirectoryURL: URL?,
  237. diskCachePathClosure: DiskCachePathClosure? = nil
  238. ) -> DiskStorage.Config
  239. {
  240. var diskConfig = DiskStorage.Config(
  241. name: name,
  242. sizeLimit: 0,
  243. directory: cacheDirectoryURL
  244. )
  245. if let closure = diskCachePathClosure {
  246. diskConfig.cachePathBlock = closure
  247. }
  248. return diskConfig
  249. }
  250. deinit {
  251. NotificationCenter.default.removeObserver(self)
  252. }
  253. // MARK: Storing Images
  254. open func store(_ image: KFCrossPlatformImage,
  255. original: Data? = nil,
  256. forKey key: String,
  257. options: KingfisherParsedOptionsInfo,
  258. toDisk: Bool = true,
  259. completionHandler: ((CacheStoreResult) -> Void)? = nil)
  260. {
  261. let identifier = options.processor.identifier
  262. let callbackQueue = options.callbackQueue
  263. let computedKey = key.computedKey(with: identifier)
  264. // Memory storage should not throw.
  265. memoryStorage.storeNoThrow(value: image, forKey: computedKey, expiration: options.memoryCacheExpiration)
  266. guard toDisk else {
  267. if let completionHandler = completionHandler {
  268. let result = CacheStoreResult(memoryCacheResult: .success(()), diskCacheResult: .success(()))
  269. callbackQueue.execute { completionHandler(result) }
  270. }
  271. return
  272. }
  273. ioQueue.async {
  274. let serializer = options.cacheSerializer
  275. if let data = serializer.data(with: image, original: original) {
  276. self.syncStoreToDisk(
  277. data,
  278. forKey: key,
  279. processorIdentifier: identifier,
  280. callbackQueue: callbackQueue,
  281. expiration: options.diskCacheExpiration,
  282. writeOptions: options.diskStoreWriteOptions,
  283. completionHandler: completionHandler)
  284. } else {
  285. guard let completionHandler = completionHandler else { return }
  286. let diskError = KingfisherError.cacheError(
  287. reason: .cannotSerializeImage(image: image, original: original, serializer: serializer))
  288. let result = CacheStoreResult(
  289. memoryCacheResult: .success(()),
  290. diskCacheResult: .failure(diskError))
  291. callbackQueue.execute { completionHandler(result) }
  292. }
  293. }
  294. }
  295. /// Stores an image to the cache.
  296. ///
  297. /// - Parameters:
  298. /// - image: The image to be stored.
  299. /// - original: The original data of the image. This value will be forwarded to the provided `serializer` for
  300. /// further use. By default, Kingfisher uses a `DefaultCacheSerializer` to serialize the image to
  301. /// data for caching in disk, it checks the image format based on `original` data to determine in
  302. /// which image format should be used. For other types of `serializer`, it depends on their
  303. /// implementation detail on how to use this original data.
  304. /// - key: The key used for caching the image.
  305. /// - identifier: The identifier of processor being used for caching. If you are using a processor for the
  306. /// image, pass the identifier of processor to this parameter.
  307. /// - serializer: The `CacheSerializer`
  308. /// - toDisk: Whether this image should be cached to disk or not. If `false`, the image is only cached in memory.
  309. /// Otherwise, it is cached in both memory storage and disk storage. Default is `true`.
  310. /// - callbackQueue: The callback queue on which `completionHandler` is invoked. Default is `.untouch`. For case
  311. /// that `toDisk` is `false`, a `.untouch` queue means `callbackQueue` will be invoked from the
  312. /// caller queue of this method. If `toDisk` is `true`, the `completionHandler` will be called
  313. /// from an internal file IO queue. To change this behavior, specify another `CallbackQueue`
  314. /// value.
  315. /// - completionHandler: A closure which is invoked when the cache operation finishes.
  316. open func store(_ image: KFCrossPlatformImage,
  317. original: Data? = nil,
  318. forKey key: String,
  319. processorIdentifier identifier: String = "",
  320. cacheSerializer serializer: CacheSerializer = DefaultCacheSerializer.default,
  321. toDisk: Bool = true,
  322. callbackQueue: CallbackQueue = .untouch,
  323. completionHandler: ((CacheStoreResult) -> Void)? = nil)
  324. {
  325. struct TempProcessor: ImageProcessor {
  326. let identifier: String
  327. func process(item: ImageProcessItem, options: KingfisherParsedOptionsInfo) -> KFCrossPlatformImage? {
  328. return nil
  329. }
  330. }
  331. let options = KingfisherParsedOptionsInfo([
  332. .processor(TempProcessor(identifier: identifier)),
  333. .cacheSerializer(serializer),
  334. .callbackQueue(callbackQueue)
  335. ])
  336. store(image, original: original, forKey: key, options: options,
  337. toDisk: toDisk, completionHandler: completionHandler)
  338. }
  339. open func storeToDisk(
  340. _ data: Data,
  341. forKey key: String,
  342. processorIdentifier identifier: String = "",
  343. expiration: StorageExpiration? = nil,
  344. callbackQueue: CallbackQueue = .untouch,
  345. completionHandler: ((CacheStoreResult) -> Void)? = nil)
  346. {
  347. ioQueue.async {
  348. self.syncStoreToDisk(
  349. data,
  350. forKey: key,
  351. processorIdentifier: identifier,
  352. callbackQueue: callbackQueue,
  353. expiration: expiration,
  354. completionHandler: completionHandler)
  355. }
  356. }
  357. private func syncStoreToDisk(
  358. _ data: Data,
  359. forKey key: String,
  360. processorIdentifier identifier: String = "",
  361. callbackQueue: CallbackQueue = .untouch,
  362. expiration: StorageExpiration? = nil,
  363. writeOptions: Data.WritingOptions = [],
  364. completionHandler: ((CacheStoreResult) -> Void)? = nil)
  365. {
  366. let computedKey = key.computedKey(with: identifier)
  367. let result: CacheStoreResult
  368. do {
  369. try self.diskStorage.store(value: data, forKey: computedKey, expiration: expiration, writeOptions: writeOptions)
  370. result = CacheStoreResult(memoryCacheResult: .success(()), diskCacheResult: .success(()))
  371. } catch {
  372. let diskError: KingfisherError
  373. if let error = error as? KingfisherError {
  374. diskError = error
  375. } else {
  376. diskError = .cacheError(reason: .cannotConvertToData(object: data, error: error))
  377. }
  378. result = CacheStoreResult(
  379. memoryCacheResult: .success(()),
  380. diskCacheResult: .failure(diskError)
  381. )
  382. }
  383. if let completionHandler = completionHandler {
  384. callbackQueue.execute { completionHandler(result) }
  385. }
  386. }
  387. // MARK: Removing Images
  388. /// Removes the image for the given key from the cache.
  389. ///
  390. /// - Parameters:
  391. /// - key: The key used for caching the image.
  392. /// - identifier: The identifier of processor being used for caching. If you are using a processor for the
  393. /// image, pass the identifier of processor to this parameter.
  394. /// - fromMemory: Whether this image should be removed from memory storage or not.
  395. /// If `false`, the image won't be removed from the memory storage. Default is `true`.
  396. /// - fromDisk: Whether this image should be removed from disk storage or not.
  397. /// If `false`, the image won't be removed from the disk storage. Default is `true`.
  398. /// - callbackQueue: The callback queue on which `completionHandler` is invoked. Default is `.untouch`.
  399. /// - completionHandler: A closure which is invoked when the cache removing operation finishes.
  400. open func removeImage(forKey key: String,
  401. processorIdentifier identifier: String = "",
  402. fromMemory: Bool = true,
  403. fromDisk: Bool = true,
  404. callbackQueue: CallbackQueue = .untouch,
  405. completionHandler: (() -> Void)? = nil)
  406. {
  407. let computedKey = key.computedKey(with: identifier)
  408. if fromMemory {
  409. memoryStorage.remove(forKey: computedKey)
  410. }
  411. if fromDisk {
  412. ioQueue.async{
  413. try? self.diskStorage.remove(forKey: computedKey)
  414. if let completionHandler = completionHandler {
  415. callbackQueue.execute { completionHandler() }
  416. }
  417. }
  418. } else {
  419. if let completionHandler = completionHandler {
  420. callbackQueue.execute { completionHandler() }
  421. }
  422. }
  423. }
  424. // MARK: Getting Images
  425. /// Gets an image for a given key from the cache, either from memory storage or disk storage.
  426. ///
  427. /// - Parameters:
  428. /// - key: The key used for caching the image.
  429. /// - options: The `KingfisherParsedOptionsInfo` options setting used for retrieving the image.
  430. /// - callbackQueue: The callback queue on which `completionHandler` is invoked. Default is `.mainCurrentOrAsync`.
  431. /// - completionHandler: A closure which is invoked when the image getting operation finishes. If the
  432. /// image retrieving operation finishes without problem, an `ImageCacheResult` value
  433. /// will be sent to this closure as result. Otherwise, a `KingfisherError` result
  434. /// with detail failing reason will be sent.
  435. open func retrieveImage(
  436. forKey key: String,
  437. options: KingfisherParsedOptionsInfo,
  438. callbackQueue: CallbackQueue = .mainCurrentOrAsync,
  439. completionHandler: ((Result<ImageCacheResult, KingfisherError>) -> Void)?)
  440. {
  441. // No completion handler. No need to start working and early return.
  442. guard let completionHandler = completionHandler else { return }
  443. // Try to check the image from memory cache first.
  444. if let image = retrieveImageInMemoryCache(forKey: key, options: options) {
  445. callbackQueue.execute { completionHandler(.success(.memory(image))) }
  446. } else if options.fromMemoryCacheOrRefresh {
  447. callbackQueue.execute { completionHandler(.success(.none)) }
  448. } else {
  449. // Begin to disk search.
  450. self.retrieveImageInDiskCache(forKey: key, options: options, callbackQueue: callbackQueue) {
  451. result in
  452. switch result {
  453. case .success(let image):
  454. guard let image = image else {
  455. // No image found in disk storage.
  456. callbackQueue.execute { completionHandler(.success(.none)) }
  457. return
  458. }
  459. // Cache the disk image to memory.
  460. // We are passing `false` to `toDisk`, the memory cache does not change
  461. // callback queue, we can call `completionHandler` without another dispatch.
  462. var cacheOptions = options
  463. cacheOptions.callbackQueue = .untouch
  464. self.store(
  465. image,
  466. forKey: key,
  467. options: cacheOptions,
  468. toDisk: false)
  469. {
  470. _ in
  471. callbackQueue.execute { completionHandler(.success(.disk(image))) }
  472. }
  473. case .failure(let error):
  474. callbackQueue.execute { completionHandler(.failure(error)) }
  475. }
  476. }
  477. }
  478. }
  479. /// Gets an image for a given key from the cache, either from memory storage or disk storage.
  480. ///
  481. /// - Parameters:
  482. /// - key: The key used for caching the image.
  483. /// - options: The `KingfisherOptionsInfo` options setting used for retrieving the image.
  484. /// - callbackQueue: The callback queue on which `completionHandler` is invoked. Default is `.mainCurrentOrAsync`.
  485. /// - completionHandler: A closure which is invoked when the image getting operation finishes. If the
  486. /// image retrieving operation finishes without problem, an `ImageCacheResult` value
  487. /// will be sent to this closure as result. Otherwise, a `KingfisherError` result
  488. /// with detail failing reason will be sent.
  489. ///
  490. /// Note: This method is marked as `open` for only compatible purpose. Do not overide this method. Instead, override
  491. /// the version receives `KingfisherParsedOptionsInfo` instead.
  492. open func retrieveImage(forKey key: String,
  493. options: KingfisherOptionsInfo? = nil,
  494. callbackQueue: CallbackQueue = .mainCurrentOrAsync,
  495. completionHandler: ((Result<ImageCacheResult, KingfisherError>) -> Void)?)
  496. {
  497. retrieveImage(
  498. forKey: key,
  499. options: KingfisherParsedOptionsInfo(options),
  500. callbackQueue: callbackQueue,
  501. completionHandler: completionHandler)
  502. }
  503. /// Gets an image for a given key from the memory storage.
  504. ///
  505. /// - Parameters:
  506. /// - key: The key used for caching the image.
  507. /// - options: The `KingfisherParsedOptionsInfo` options setting used for retrieving the image.
  508. /// - Returns: The image stored in memory cache, if exists and valid. Otherwise, if the image does not exist or
  509. /// has already expired, `nil` is returned.
  510. open func retrieveImageInMemoryCache(
  511. forKey key: String,
  512. options: KingfisherParsedOptionsInfo) -> KFCrossPlatformImage?
  513. {
  514. let computedKey = key.computedKey(with: options.processor.identifier)
  515. return memoryStorage.value(forKey: computedKey, extendingExpiration: options.memoryCacheAccessExtendingExpiration)
  516. }
  517. /// Gets an image for a given key from the memory storage.
  518. ///
  519. /// - Parameters:
  520. /// - key: The key used for caching the image.
  521. /// - options: The `KingfisherOptionsInfo` options setting used for retrieving the image.
  522. /// - Returns: The image stored in memory cache, if exists and valid. Otherwise, if the image does not exist or
  523. /// has already expired, `nil` is returned.
  524. ///
  525. /// Note: This method is marked as `open` for only compatible purpose. Do not overide this method. Instead, override
  526. /// the version receives `KingfisherParsedOptionsInfo` instead.
  527. open func retrieveImageInMemoryCache(
  528. forKey key: String,
  529. options: KingfisherOptionsInfo? = nil) -> KFCrossPlatformImage?
  530. {
  531. return retrieveImageInMemoryCache(forKey: key, options: KingfisherParsedOptionsInfo(options))
  532. }
  533. func retrieveImageInDiskCache(
  534. forKey key: String,
  535. options: KingfisherParsedOptionsInfo,
  536. callbackQueue: CallbackQueue = .untouch,
  537. completionHandler: @escaping (Result<KFCrossPlatformImage?, KingfisherError>) -> Void)
  538. {
  539. let computedKey = key.computedKey(with: options.processor.identifier)
  540. let loadingQueue: CallbackQueue = options.loadDiskFileSynchronously ? .untouch : .dispatch(ioQueue)
  541. loadingQueue.execute {
  542. do {
  543. var image: KFCrossPlatformImage? = nil
  544. if let data = try self.diskStorage.value(forKey: computedKey, extendingExpiration: options.diskCacheAccessExtendingExpiration) {
  545. image = options.cacheSerializer.image(with: data, options: options)
  546. }
  547. if options.backgroundDecode {
  548. image = image?.kf.decoded(scale: options.scaleFactor)
  549. }
  550. callbackQueue.execute { completionHandler(.success(image)) }
  551. } catch let error as KingfisherError {
  552. callbackQueue.execute { completionHandler(.failure(error)) }
  553. } catch {
  554. assertionFailure("The internal thrown error should be a `KingfisherError`.")
  555. }
  556. }
  557. }
  558. /// Gets an image for a given key from the disk storage.
  559. ///
  560. /// - Parameters:
  561. /// - key: The key used for caching the image.
  562. /// - options: The `KingfisherOptionsInfo` options setting used for retrieving the image.
  563. /// - callbackQueue: The callback queue on which `completionHandler` is invoked. Default is `.untouch`.
  564. /// - completionHandler: A closure which is invoked when the operation finishes.
  565. open func retrieveImageInDiskCache(
  566. forKey key: String,
  567. options: KingfisherOptionsInfo? = nil,
  568. callbackQueue: CallbackQueue = .untouch,
  569. completionHandler: @escaping (Result<KFCrossPlatformImage?, KingfisherError>) -> Void)
  570. {
  571. retrieveImageInDiskCache(
  572. forKey: key,
  573. options: KingfisherParsedOptionsInfo(options),
  574. callbackQueue: callbackQueue,
  575. completionHandler: completionHandler)
  576. }
  577. // MARK: Cleaning
  578. /// Clears the memory & disk storage of this cache. This is an async operation.
  579. ///
  580. /// - Parameter handler: A closure which is invoked when the cache clearing operation finishes.
  581. /// This `handler` will be called from the main queue.
  582. public func clearCache(completion handler: (() -> Void)? = nil) {
  583. clearMemoryCache()
  584. clearDiskCache(completion: handler)
  585. }
  586. /// Clears the memory storage of this cache.
  587. @objc public func clearMemoryCache() {
  588. memoryStorage.removeAll()
  589. }
  590. /// Clears the disk storage of this cache. This is an async operation.
  591. ///
  592. /// - Parameter handler: A closure which is invoked when the cache clearing operation finishes.
  593. /// This `handler` will be called from the main queue.
  594. open func clearDiskCache(completion handler: (() -> Void)? = nil) {
  595. ioQueue.async {
  596. do {
  597. try self.diskStorage.removeAll()
  598. } catch _ { }
  599. if let handler = handler {
  600. DispatchQueue.main.async { handler() }
  601. }
  602. }
  603. }
  604. /// Clears the expired images from memory & disk storage. This is an async operation.
  605. open func cleanExpiredCache(completion handler: (() -> Void)? = nil) {
  606. cleanExpiredMemoryCache()
  607. cleanExpiredDiskCache(completion: handler)
  608. }
  609. /// Clears the expired images from disk storage.
  610. open func cleanExpiredMemoryCache() {
  611. memoryStorage.removeExpired()
  612. }
  613. /// Clears the expired images from disk storage. This is an async operation.
  614. @objc func cleanExpiredDiskCache() {
  615. cleanExpiredDiskCache(completion: nil)
  616. }
  617. /// Clears the expired images from disk storage. This is an async operation.
  618. ///
  619. /// - Parameter handler: A closure which is invoked when the cache clearing operation finishes.
  620. /// This `handler` will be called from the main queue.
  621. open func cleanExpiredDiskCache(completion handler: (() -> Void)? = nil) {
  622. ioQueue.async {
  623. do {
  624. var removed: [URL] = []
  625. let removedExpired = try self.diskStorage.removeExpiredValues()
  626. removed.append(contentsOf: removedExpired)
  627. let removedSizeExceeded = try self.diskStorage.removeSizeExceededValues()
  628. removed.append(contentsOf: removedSizeExceeded)
  629. if !removed.isEmpty {
  630. DispatchQueue.main.async {
  631. let cleanedHashes = removed.map { $0.lastPathComponent }
  632. NotificationCenter.default.post(
  633. name: .KingfisherDidCleanDiskCache,
  634. object: self,
  635. userInfo: [KingfisherDiskCacheCleanedHashKey: cleanedHashes])
  636. }
  637. }
  638. if let handler = handler {
  639. DispatchQueue.main.async { handler() }
  640. }
  641. } catch {}
  642. }
  643. }
  644. #if !os(macOS) && !os(watchOS)
  645. /// Clears the expired images from disk storage when app is in background. This is an async operation.
  646. /// In most cases, you should not call this method explicitly.
  647. /// It will be called automatically when `UIApplicationDidEnterBackgroundNotification` received.
  648. @objc public func backgroundCleanExpiredDiskCache() {
  649. // if 'sharedApplication()' is unavailable, then return
  650. guard let sharedApplication = KingfisherWrapper<UIApplication>.shared else { return }
  651. func endBackgroundTask(_ task: inout UIBackgroundTaskIdentifier) {
  652. sharedApplication.endBackgroundTask(task)
  653. task = UIBackgroundTaskIdentifier.invalid
  654. }
  655. var backgroundTask: UIBackgroundTaskIdentifier!
  656. backgroundTask = sharedApplication.beginBackgroundTask {
  657. endBackgroundTask(&backgroundTask!)
  658. }
  659. cleanExpiredDiskCache {
  660. endBackgroundTask(&backgroundTask!)
  661. }
  662. }
  663. #endif
  664. // MARK: Image Cache State
  665. /// Returns the cache type for a given `key` and `identifier` combination.
  666. /// This method is used for checking whether an image is cached in current cache.
  667. /// It also provides information on which kind of cache can it be found in the return value.
  668. ///
  669. /// - Parameters:
  670. /// - key: The key used for caching the image.
  671. /// - identifier: Processor identifier which used for this image. Default is the `identifier` of
  672. /// `DefaultImageProcessor.default`.
  673. /// - Returns: A `CacheType` instance which indicates the cache status.
  674. /// `.none` means the image is not in cache or it is already expired.
  675. open func imageCachedType(
  676. forKey key: String,
  677. processorIdentifier identifier: String = DefaultImageProcessor.default.identifier) -> CacheType
  678. {
  679. let computedKey = key.computedKey(with: identifier)
  680. if memoryStorage.isCached(forKey: computedKey) { return .memory }
  681. if diskStorage.isCached(forKey: computedKey) { return .disk }
  682. return .none
  683. }
  684. /// Returns whether the file exists in cache for a given `key` and `identifier` combination.
  685. ///
  686. /// - Parameters:
  687. /// - key: The key used for caching the image.
  688. /// - identifier: Processor identifier which used for this image. Default is the `identifier` of
  689. /// `DefaultImageProcessor.default`.
  690. /// - Returns: A `Bool` which indicates whether a cache could match the given `key` and `identifier` combination.
  691. ///
  692. /// - Note:
  693. /// The return value does not contain information about from which kind of storage the cache matches.
  694. /// To get the information about cache type according `CacheType`,
  695. /// use `imageCachedType(forKey:processorIdentifier:)` instead.
  696. public func isCached(
  697. forKey key: String,
  698. processorIdentifier identifier: String = DefaultImageProcessor.default.identifier) -> Bool
  699. {
  700. return imageCachedType(forKey: key, processorIdentifier: identifier).cached
  701. }
  702. /// Gets the hash used as cache file name for the key.
  703. ///
  704. /// - Parameters:
  705. /// - key: The key used for caching the image.
  706. /// - identifier: Processor identifier which used for this image. Default is the `identifier` of
  707. /// `DefaultImageProcessor.default`.
  708. /// - Returns: The hash which is used as the cache file name.
  709. ///
  710. /// - Note:
  711. /// By default, for a given combination of `key` and `identifier`, `ImageCache` will use the value
  712. /// returned by this method as the cache file name. You can use this value to check and match cache file
  713. /// if you need.
  714. open func hash(
  715. forKey key: String,
  716. processorIdentifier identifier: String = DefaultImageProcessor.default.identifier) -> String
  717. {
  718. let computedKey = key.computedKey(with: identifier)
  719. return diskStorage.cacheFileName(forKey: computedKey)
  720. }
  721. /// Calculates the size taken by the disk storage.
  722. /// It is the total file size of all cached files in the `diskStorage` on disk in bytes.
  723. ///
  724. /// - Parameter handler: Called with the size calculating finishes. This closure is invoked from the main queue.
  725. open func calculateDiskStorageSize(completion handler: @escaping ((Result<UInt, KingfisherError>) -> Void)) {
  726. ioQueue.async {
  727. do {
  728. let size = try self.diskStorage.totalSize()
  729. DispatchQueue.main.async { handler(.success(size)) }
  730. } catch let error as KingfisherError {
  731. DispatchQueue.main.async { handler(.failure(error)) }
  732. } catch {
  733. assertionFailure("The internal thrown error should be a `KingfisherError`.")
  734. }
  735. }
  736. }
  737. #if swift(>=5.5)
  738. #if canImport(_Concurrency)
  739. @available(iOS 13.0, macOS 10.15, tvOS 13.0, watchOS 6.0, *)
  740. open var diskStorageSize: UInt {
  741. get async throws {
  742. try await withCheckedThrowingContinuation { continuation in
  743. calculateDiskStorageSize { result in
  744. continuation.resume(with: result)
  745. }
  746. }
  747. }
  748. }
  749. #endif
  750. #endif
  751. /// Gets the cache path for the key.
  752. /// It is useful for projects with web view or anyone that needs access to the local file path.
  753. ///
  754. /// i.e. Replacing the `<img src='path_for_key'>` tag in your HTML.
  755. ///
  756. /// - Parameters:
  757. /// - key: The key used for caching the image.
  758. /// - identifier: Processor identifier which used for this image. Default is the `identifier` of
  759. /// `DefaultImageProcessor.default`.
  760. /// - Returns: The disk path of cached image under the given `key` and `identifier`.
  761. ///
  762. /// - Note:
  763. /// This method does not guarantee there is an image already cached in the returned path. It just gives your
  764. /// the path that the image should be, if it exists in disk storage.
  765. ///
  766. /// You could use `isCached(forKey:)` method to check whether the image is cached under that key in disk.
  767. open func cachePath(
  768. forKey key: String,
  769. processorIdentifier identifier: String = DefaultImageProcessor.default.identifier) -> String
  770. {
  771. let computedKey = key.computedKey(with: identifier)
  772. return diskStorage.cacheFileURL(forKey: computedKey).path
  773. }
  774. }
  775. #if !os(macOS) && !os(watchOS)
  776. // MARK: - For App Extensions
  777. extension UIApplication: KingfisherCompatible { }
  778. extension KingfisherWrapper where Base: UIApplication {
  779. public static var shared: UIApplication? {
  780. let selector = NSSelectorFromString("sharedApplication")
  781. guard Base.responds(to: selector) else { return nil }
  782. return Base.perform(selector).takeUnretainedValue() as? UIApplication
  783. }
  784. }
  785. #endif
  786. extension String {
  787. func computedKey(with identifier: String) -> String {
  788. if identifier.isEmpty {
  789. return self
  790. } else {
  791. return appending("@\(identifier)")
  792. }
  793. }
  794. }