Respond to CR.

// FREEBIE
8 years ago · c2a17920ba
parent 004c9e4809
commit c2a17920ba
1 changed files with 2 additions and 1 deletions
--- a/Signal/src/network/GiphyDownloader.swift
+++ b/Signal/src/network/GiphyDownloader.swift
@ -644,7 +644,7 @@ extension URLSessionTask {
                }
                self?.handleAssetSizeResponse(assetRequest:assetRequest, response:response, error:error)
            })
-
+            assetRequest.contentLengthTask = task
            task.resume()
        } else {
            // Start a download task.
@ -661,6 +661,7 @@ extension URLSessionTask {
            let task = downloadSession.dataTask(with:request)
            task.assetRequest = assetRequest
            task.assetSegment = assetSegment
+            assetSegment.task = task
            task.resume()
        }