Merge pull request mrackwitz#58 from mrackwitz/fix/57_afnetworking_su…

…pport_operation_finished Fixes mrackwitz#57: Hides overlay if task/operation is finished
SolarCS · Aug 20, 2014 · e5f258d · e5f258d
2 parents 0f308ea + 25e2f83
commit e5f258d
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 2 deletions.
diff --git a/Example/MRAFNetworkingSupportOperationViewController.m b/Example/MRAFNetworkingSupportOperationViewController.m
@@ -74,8 +74,16 @@ - (IBAction)onOverlayViewGo:(id)sender {
                                                            }];
 
     MRProgressOverlayView *overlayView = [MRProgressOverlayView showOverlayAddedTo:self.view animated:YES];
-    [overlayView setModeAndProgressWithStateOfOperation:operation];
-    [overlayView setStopBlockForOperation:operation];
+
+    dispatch_async(dispatch_get_global_queue(DISPATCH_QUEUE_PRIORITY_HIGH, 0), ^{
+        // Do an expensive background operation, before observing the operation
+        sleep(2);
+
+        dispatch_async(dispatch_get_main_queue(), ^{
+            [overlayView setModeAndProgressWithStateOfOperation:operation];
+            [overlayView setStopBlockForOperation:operation];
+        });
+    });
 }
 
 - (IBAction)onOverlayViewUpload:(id)sender {

diff --git a/src/Support/AFNetworking/MRProgressOverlayView+AFNetworking.m b/src/Support/AFNetworking/MRProgressOverlayView+AFNetworking.m
@@ -84,6 +84,8 @@ - (void)setModeAndProgressWithStateOfTask:(NSURLSessionTask *)task {
             [task addObserver:self forKeyPath:NSStringFromSelector(@selector(countOfBytesSent))     options:0 context:MRTaskCountOfBytesSentContext];
             [task addObserver:self forKeyPath:NSStringFromSelector(@selector(countOfBytesReceived)) options:0 context:MRTaskCountOfBytesReceivedContext];
 #pragma clang diagnostic pop
+        } else {
+            [self dismiss:YES];
         }
     }
 }
@@ -151,6 +153,8 @@ - (void)setModeAndProgressWithStateOfOperation:(AFURLConnectionOperation *)opera
                 // Unregister
                 [weakSelf.operation setDownloadProgressBlock:originalDownloadProgressBlock];
             }];
+        } else {
+            [self dismiss:YES];
         }
     }
 }