Add cached image data for token's icon

keccak384 · Jan 9, 2020 · 2e27377 · 2e27377
1 parent 445d8b5
commit 2e27377
Show file tree

Hide file tree

Showing 3 changed files with 21 additions and 2 deletions.
diff --git a/KyberNetwork/KyberNetwork/Extensions/UIButton+Kyber.swift b/KyberNetwork/KyberNetwork/Extensions/UIButton+Kyber.swift
@@ -17,10 +17,18 @@ extension UIButton {
     size: CGSize? = nil,
     state: UIControlState = .normal
     ) {
+    if let cachedImg = UIImage.imageCache.object(forKey: url as AnyObject) as? UIImage {
+      self.setImage(cachedImg.resizeImage(to: size), for: .normal)
+      self.layoutIfNeeded()
+      return
+    }
     self.setImage(placeHolder?.resizeImage(to: size), for: state)
-    URLSession.shared.dataTask(with: url) { (data, _, error) in
+    self.layoutIfNeeded()
+    URLSession.shared.dataTask(with: url) { [weak self] (data, _, error) in
+      guard let `self` = self else { return }
       if error == nil, let data = data, let image = UIImage(data: data) {
         DispatchQueue.main.async {
+          UIImage.imageCache.setObject(image, forKey: url as AnyObject)
           self.setImage(image.resizeImage(to: size), for: .normal)
           self.layoutIfNeeded()
         }

diff --git a/KyberNetwork/KyberNetwork/Extensions/UIImage+Kyber.swift b/KyberNetwork/KyberNetwork/Extensions/UIImage+Kyber.swift
@@ -4,6 +4,9 @@ import UIKit
 import JdenticonSwift
 
 extension UIImage {
+
+  static let imageCache = NSCache<AnyObject, AnyObject>()
+
   static func generateQRCode(from string: String) -> UIImage? {
     let context = CIContext()
     let data = string.data(using: String.Encoding.ascii)

diff --git a/KyberNetwork/KyberNetwork/Extensions/UIImageView+Kyber.swift b/KyberNetwork/KyberNetwork/Extensions/UIImageView+Kyber.swift
@@ -4,10 +4,18 @@ import UIKit
 
 extension UIImageView {
   func setImage(with url: URL, placeholder: UIImage?, size: CGSize? = nil, applyNoir: Bool = false) {
+    if let cachedImg = UIImage.imageCache.object(forKey: url as AnyObject) as? UIImage {
+      self.image = applyNoir ? cachedImg.resizeImage(to: size)?.noir : cachedImg.resizeImage(to: size)
+      self.layoutIfNeeded()
+      return
+    }
     self.image = applyNoir ? placeholder?.resizeImage(to: size)?.noir : placeholder?.resizeImage(to: size)
-    URLSession.shared.dataTask(with: url) { (data, _, error) in
+    self.layoutIfNeeded()
+    URLSession.shared.dataTask(with: url) { [weak self] (data, _, error) in
+      guard let `self` = self else { return }
       if error == nil, let data = data, let image = UIImage(data: data) {
         DispatchQueue.main.async {
+          UIImage.imageCache.setObject(image, forKey: url as AnyObject)
           self.image = applyNoir ? image.resizeImage(to: size)?.noir : image.resizeImage(to: size)
           self.layoutIfNeeded()
         }