Fixes unrecognized selector for 'taskIdentifier' and 'originalRequest…

…' methods - Adds 'taskIdentifier' and 'originalRequest' methods for SessionUploadTask and SessionDownloadTask
venmo · Apr 9, 2018 · 7b175e5 · 7b175e5
1 parent 1b7ccc8
commit 7b175e5
Show file tree

Hide file tree

Showing 3 changed files with 29 additions and 1 deletion.
diff --git a/DVR/SessionDataTask.swift b/DVR/SessionDataTask.swift
@@ -34,7 +34,6 @@ final class SessionDataTask: URLSessionDataTask {
       return _originalRequest
     }
 
-
     // MARK: - Initializers
 
     init(session: Session, request: URLRequest, completion: (Completion)? = nil) {

diff --git a/DVR/SessionDownloadTask.swift b/DVR/SessionDownloadTask.swift
@@ -1,3 +1,5 @@
+private var globalTaskIdentifier: Int = 200000
+
 final class SessionDownloadTask: URLSessionDownloadTask {
 
     // MARK: - Types
@@ -10,6 +12,18 @@ final class SessionDownloadTask: URLSessionDownloadTask {
     let request: URLRequest
     let completion: Completion?
 
+    var _taskIdentifier: Int = {
+      globalTaskIdentifier += 1
+      return globalTaskIdentifier
+    }()
+    override var taskIdentifier: Int {
+      return _taskIdentifier
+    }
+
+    var _originalRequest: URLRequest?
+    override var originalRequest: URLRequest? {
+      return _originalRequest
+    }
 
     // MARK: - Initializers
 

diff --git a/DVR/SessionUploadTask.swift b/DVR/SessionUploadTask.swift
@@ -1,3 +1,5 @@
+private var globalTaskIdentifier: Int = 300000
+
 final class SessionUploadTask: URLSessionUploadTask {
 
     // MARK: - Types
@@ -11,6 +13,19 @@ final class SessionUploadTask: URLSessionUploadTask {
     let completion: Completion?
     let dataTask: SessionDataTask
 
+    var _taskIdentifier: Int = {
+      globalTaskIdentifier += 1
+      return globalTaskIdentifier
+    }()
+    override var taskIdentifier: Int {
+      return _taskIdentifier
+    }
+
+    var _originalRequest: URLRequest?
+    override var originalRequest: URLRequest? {
+      return _originalRequest
+    }
+
     // MARK: - Initializers
 
     init(session: Session, request: URLRequest, completion: Completion? = nil) {