Merge pull request #580 from Fnoz/fix/cancelDownloadTask

(Supplement)Modify disk and download task cancel logic.
onevcat · Feb 5, 2017 · d11dbb9 · d11dbb9
2 parents 8c9d3f6 + 42e0fea
commit d11dbb9
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/Sources/NSButton+Kingfisher.swift b/Sources/NSButton+Kingfisher.swift
@@ -95,7 +95,7 @@ extension Kingfisher where Base: NSButton {
      Nothing will happen if the downloading has already finished.
      */
     public func cancelImageDownloadTask() {
-        imageTask?.downloadTask?.cancel()
+        imageTask?.cancel()
     }
 
     /**
@@ -164,7 +164,7 @@ extension Kingfisher where Base: NSButton {
     /// Cancel the alternate image download task bounded to the image view if it is running. 
     /// Nothing will happen if the downloading has already finished.
     public func cancelAlternateImageDownloadTask() {
-        alternateImageTask?.downloadTask?.cancel()
+        alternateImageTask?.cancel()
     }
 }
 

diff --git a/Sources/UIButton+Kingfisher.swift b/Sources/UIButton+Kingfisher.swift
@@ -99,7 +99,7 @@ extension Kingfisher where Base: UIButton {
      Nothing will happen if the downloading has already finished.
      */
     public func cancelImageDownloadTask() {
-        imageTask?.downloadTask?.cancel()
+        imageTask?.cancel()
     }
 
     /**
@@ -168,7 +168,7 @@ extension Kingfisher where Base: UIButton {
      Nothing will happen if the downloading has already finished.
      */
     public func cancelBackgroundImageDownloadTask() {
-        backgroundImageTask?.downloadTask?.cancel()
+        backgroundImageTask?.cancel()
     }
 
 }