从零打造一个iOS图片加载框架（四）

前言

在上一篇文章中，我们讨论了如何使用NSURLSession从URL下载图像。然而，仅仅获取最终结果是不够的。在许多情况下，我们还需要能够跟踪图像下载的进度。例如，我们可以显示一个进度条或在下载完成后通知用户。

要获取下载进度，我们需要自己实现NSURLSession的NSURLSessionDelegate协议。NSURLSession具有大量协议，但我们只需要关注以下几个：

1. func urlSession(_ session: URLSession, downloadTask: URLSessionDownloadTask, didWriteData bytesWritten: Int64, totalBytesWritten: Int64, totalBytesExpectedToWrite: Int64)：此方法在收到部分数据时调用。它提供下载的当前进度。
2. func urlSession(_ session: URLSession, task: URLSessionTask, didCompleteWithError error: Error?)：此方法在下载完成或发生错误时调用。我们可以使用它来完成下载并更新UI。

实现NSURLSessionDelegate

现在我们已经了解了所需的协议方法，让我们实现它们。首先，我们需要创建一个新的类并使其遵循NSURLSessionDelegate协议：

class ImageDownloaderDelegate: NSObject, URLSessionDelegate, URLSessionDownloadDelegate {

    // ...

}

然后，我们需要在ImageDownloader类中实现NSURLSessionDelegate协议方法。以下是获取下载进度的示例实现：

func urlSession(_ session: URLSession, downloadTask: URLSessionDownloadTask, didWriteData bytesWritten: Int64, totalBytesWritten: Int64, totalBytesExpectedToWrite: Int64) {

    // 计算进度
    let progress = Float(totalBytesWritten) / Float(totalBytesExpectedToWrite)

    // 更新UI（例如，显示进度条）

}

同样，以下是下载完成或发生错误时的示例实现：

func urlSession(_ session: URLSession, task: URLSessionTask, didCompleteWithError error: Error?) {

    // 如果没有错误，完成下载
    if error == nil {

        // ...

    } else {

        // 处理错误

    }

    // 更新UI（例如，隐藏进度条）

}

使用自定义的NSURLSessionDelegate

现在我们已经实现了自定义的NSURLSessionDelegate，我们需要在ImageDownloader类中使用它：

let session = URLSession(configuration: .default, delegate: ImageDownloaderDelegate(), delegateQueue: nil)

通过提供我们的自定义委托，我们可以跟踪下载进度并相应地更新UI。

结论

在本教程中，我们学习了如何实现NSURLSessionDelegate协议以获取图像下载进度。这使我们能够构建更强大的图像加载框架，能够跟踪下载的实时进度并向用户提供反馈。在下一篇文章中，我们将探讨如何使用自定义的NSURLSessionDelegate来处理错误并取消下载。