FIX: Save resume data at request cancel.

kanyun-inc · Sep 1, 2017 · efb9c69 · efb9c69
1 parent 277de20
commit efb9c69
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/YTKNetwork/YTKNetworkAgent.m b/YTKNetwork/YTKNetworkAgent.m
@@ -241,7 +241,16 @@ - (void)addRequest:(YTKBaseRequest *)request {
 - (void)cancelRequest:(YTKBaseRequest *)request {
     NSParameterAssert(request != nil);
 
-    [request.requestTask cancel];
+    if (request.resumableDownloadPath) {
+        NSURLSessionDownloadTask *requestTask = (NSURLSessionDownloadTask *)request.requestTask;
+        [requestTask cancelByProducingResumeData:^(NSData *resumeData) {
+            NSURL *localUrl = [self incompleteDownloadTempPathForDownloadPath:request.resumableDownloadPath];
+            [resumeData writeToURL:localUrl atomically:YES];
+        }];
+    } else {
+        [request.requestTask cancel];
+    }
+
     [self removeRequestFromRecord:request];
     [request clearCompletionBlock];
 }