修正寫法上的錯誤

DaidoujiChen · Jan 16, 2015 · 7a3b05a · 7a3b05a
1 parent 1027ebd
commit 7a3b05a
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 2 deletions.
diff --git a/e-Hentai/HentaiDownloadCenter/HentaiDownloadBookOperation.m b/e-Hentai/HentaiDownloadCenter/HentaiDownloadBookOperation.m
@@ -106,11 +106,17 @@ - (void)downloadResult:(NSString *)urlString heightOfSize:(CGFloat)height isSucc
             self.failCount++;
             self.maxHentaiCount = [NSString stringWithFormat:@"%ld", [self.maxHentaiCount integerValue] - 1];
 
+            NSUInteger removeIndex = NSNotFound;
             for (NSString *eachURLString in self.hentaiImageURLs) {
                 if ([eachURLString isEqualToString:urlString]) {
-                    [self.hentaiImageURLs removeObject:eachURLString];
+                    removeIndex = [self.hentaiImageURLs indexOfObject:eachURLString];
+                    break;
                 }
             }
+
+            if (removeIndex != NSNotFound) {
+                [self.hentaiImageURLs removeObjectAtIndex:removeIndex];
+            }
         }
     }
     self.status = HentaiDownloadBookOperationStatusDownloading;

diff --git a/e-Hentai/PhotoViewController/PhotoViewController.m b/e-Hentai/PhotoViewController/PhotoViewController.m
@@ -252,11 +252,17 @@ - (void)downloadResult:(NSString *)urlString heightOfSize:(CGFloat)height isSucc
             self.failCount++;
             self.maxHentaiCount = [NSString stringWithFormat:@"%ld", [self.maxHentaiCount integerValue] - 1];
 
+            NSUInteger removeIndex = NSNotFound;
             for (NSString *eachURLString in self.hentaiImageURLs) {
                 if ([eachURLString isEqualToString:urlString]) {
-                    [self.hentaiImageURLs removeObject:eachURLString];
+                    removeIndex = [self.hentaiImageURLs indexOfObject:eachURLString];
+                    break;
                 }
             }
+
+            if (removeIndex != NSNotFound) {
+                [self.hentaiImageURLs removeObjectAtIndex:removeIndex];
+            }
         }
     }
 }