Fix progress reporting

kean · Jun 7, 2019 · 69eae92 · 69eae92
1 parent f763234
commit 69eae92
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/Sources/ImagePipeline.swift b/Sources/ImagePipeline.swift
@@ -85,8 +85,7 @@ public /* final */ class ImagePipeline {
                     progressHandler?(response, task.completedUnitCount, task.totalUnitCount)
                 }
             case let .progress(progress):
-                task.setProgress(progress)
-                progressHandler?(nil, task.completedUnitCount, task.totalUnitCount)
+                progressHandler?(nil, progress.completed, progress.total)
             case let .error(error):
                 completion?(.failure(error))
             }
@@ -149,6 +148,9 @@ public /* final */ class ImagePipeline {
 
             DispatchQueue.main.async {
                 guard !task.isCancelled else { return }
+                if case let .progress(progress) = event {
+                    task.setProgress(progress)
+                }
                 observer(task, event)
             }
         }