具有动态下载大小检查功能的 Swift 下载器

如何解决具有动态下载大小检查功能的 Swift 下载器

我目前正在 Swift 中开发一个简单的文本下载器，并且正在努力在一个函数中返回下载结果，同时在一个单独的委托中检查下载是否小于 10MB（如果更大则取消）。
此下载器应按如下方式调用：

let simpleDownloader = SimpleDownloader()
simpleDownloader.download(urlString: "https://official-joke-api.appspot.com/random_joke")

下载器仅由一个函数组成，该函数将所需的 URL 作为参数，如果下载过大，应返回结果（字符串）或 nil。

public class SimpleDownloader : NSObject {

    var task: URLSessionDownloadTask?

    func download(urlString: String) -> String? {
        let config = URLSessionConfiguration.ephemeral
        let session = URLSession(configuration: config,delegate: self,delegateQueue: nil)

        let url = NSURL(string: urlString)
        task = session.downloadTask(with: url! as URL)
        task?.resume()
        
        // if the download was successful - return the data as a String - otherwise nil
        return nil
    }
}

检查下载是否过大的委托如下所示：

extension SimpleDownloader : URLSessionDownloadDelegate {

    public func urlSession(_ session: URLSession,downloadTask: URLSessionDownloadTask,didFinishDownloadingTo location: URL) {
        print("Finished downloading!")
    }

    public func urlSession(_ session: URLSession,didWriteData bytesWritten: Int64,totalBytesWritten: Int64,totalBytesExpectedToWrite: Int64) -> Void {

        if totalBytesExpectedToWrite > 10000000 { // 10MB
            task?.cancel()
        }
    }
}

我现在的问题是如何返回下载的文本（如果成功），同时检查下载是否变得太大。更详细地说：在检查委托中的下载大小时，我无法使用 session.downloadTask 完成处理程序。

我的第一个解决方案是预先获取下载的大小（通过 HTTP 标头），但一些 API 调用（如我使用的 API 调用）根本不返回标头，因此无法使用这种方法。

有没有办法像我写的（SimpleDownloader）那样编写一个简单的下载器，它还会检查下载量是否过大？

谢谢！