Fix the issue in stopPreheatingImagesForRequests

ggthedev · Mar 19, 2015 · 92bf711 · 92bf711
1 parent 7aa6330
commit 92bf711
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/DFImageManager/Source/Core/Managing/DFImageManager.m b/DFImageManager/Source/Core/Managing/DFImageManager.m
@@ -436,8 +436,11 @@ - (void)stopPreheatingImagesForRequests:(NSArray *)requests {
     dispatch_async(_queue, ^{
         for (DFImageRequest *request in [self _canonicalRequestsForRequests:requests]) {
             _DFImageRequestKey *key = DFImageCacheKeyCreate(request);
-            [self _setImageTaskState:_DFImageTaskStateCancelled task:_preheatingTasks[key]];
-            [_preheatingTasks removeObjectForKey:key];
+            _DFImageTask *task = _preheatingTasks[key];
+            if (task) {
+                [self _setImageTaskState:_DFImageTaskStateCancelled task:task];
+                [_preheatingTasks removeObjectForKey:key];
+            }
         }
     });
 }