-
Notifications
You must be signed in to change notification settings - Fork 1.1k
/
MemoryCache.swift
118 lines (90 loc) · 3.54 KB
/
MemoryCache.swift
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
import Foundation
import AlamofireImage
import WordPressUI
import SDWebImage
final class MemoryCache {
/// A shared image cache used by the entire system.
static let shared = MemoryCache()
private let cache = NSCache<NSString, AnyObject>()
private init() {
self.cache.totalCostLimit = 256_000_000 // 256 MB
NotificationCenter.default.addObserver(self, selector: #selector(didReceiveMemoryWarning), name: UIApplication.didReceiveMemoryWarningNotification, object: nil)
}
@objc private func didReceiveMemoryWarning() {
cache.removeAllObjects()
}
// MARK: - UIImage
func setImage(_ image: UIImage, forKey key: String) {
cache.setObject(image, forKey: key as NSString, cost: Int(image.sd_memoryCost))
}
func getImage(forKey key: String) -> UIImage? {
cache.object(forKey: key as NSString) as? UIImage
}
func removeImage(forKey key: String) {
cache.removeObject(forKey: key as NSString)
}
// MARK: - Data
func setData(_ data: Data, forKey key: String) {
cache.setObject(data as NSData, forKey: key as NSString, cost: data.count)
}
func geData(forKey key: String) -> Data? {
cache.object(forKey: key as NSString) as? Data
}
func removeData(forKey key: String) {
cache.removeObject(forKey: key as NSString)
}
}
extension MemoryCache {
/// Registers the cache with all the image loading systems used by the app.
func register() {
// WordPressUI
WordPressUI.ImageCache.shared = WordpressUICacheAdapter(cache: .shared)
// AlamofireImage
UIImageView.af_sharedImageDownloader = AlamofireImage.ImageDownloader(
imageCache: AlamofireImageCacheAdapter(cache: .shared)
)
// WordPress.AnimatedImageCache uses WordPress.MemoryCache directly
}
}
private struct WordpressUICacheAdapter: WordPressUI.ImageCaching {
let cache: MemoryCache
func setImage(_ image: UIImage, forKey key: String) {
cache.setImage(image, forKey: key)
}
func getImage(forKey key: String) -> UIImage? {
cache.getImage(forKey: key)
}
}
private struct AlamofireImageCacheAdapter: AlamofireImage.ImageRequestCache {
let cache: MemoryCache
func image(for request: URLRequest, withIdentifier identifier: String?) -> AlamofireImage.Image? {
image(withIdentifier: cacheKey(for: request, identifier: identifier))
}
func add(_ image: AlamofireImage.Image, for request: URLRequest, withIdentifier identifier: String?) {
add(image, withIdentifier: cacheKey(for: request, identifier: identifier))
}
func removeImage(for request: URLRequest, withIdentifier identifier: String?) -> Bool {
removeImage(withIdentifier: cacheKey(for: request, identifier: identifier))
}
func image(withIdentifier identifier: String) -> AlamofireImage.Image? {
cache.getImage(forKey: identifier)
}
func add(_ image: AlamofireImage.Image, withIdentifier identifier: String) {
cache.setImage(image, forKey: identifier)
}
func removeImage(withIdentifier identifier: String) -> Bool {
cache.removeImage(forKey: identifier)
return true
}
func removeAllImages() -> Bool {
// Do nothing (the app decides when to remove images)
return true
}
private func cacheKey(for request: URLRequest, identifier: String?) -> String {
var key = request.url?.absoluteString ?? ""
if let identifier = identifier {
key += "-\(identifier)"
}
return key
}
}