: hammer: Update main queue return completion

noblakit01 · Jul 16, 2017 · 3b31cb1 · 3b31cb1
1 parent f4cc130
commit 3b31cb1
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/PhotoCollectionView/PhotoCache.swift b/PhotoCollectionView/PhotoCache.swift
@@ -28,7 +28,9 @@ open class PhotoCache {
         if let workItem = workItems.object(forKey: urlString as NSString) {
             workItem.notify(queue: queue, execute: { [weak self] in
                 if let image = self?.images.object(forKey: urlString as NSString) {
-                    completion(image)
+                    DispatchQueue.main.async {
+                        completion(image)
+                    }
                 }
             })
             return
@@ -47,7 +49,9 @@ open class PhotoCache {
                     }
                 }
             } catch ( _) {
-                completion(nil)
+                DispatchQueue.main.async {
+                    completion(nil)
+                }
             }
         }
         workItems.setObject(workItem, forKey: urlString as NSString)