快速从 URL 获取文件大小 - Swift

在开发iOS应用程序中，有时我们需要从远程URL获取文件的大小。这对于下载管理器、文件预览和网络优化等功能非常有用。在Swift中，我们可以使用URLSession来实现快速从URL获取文件大小的功能。

实现

首先，我们需要创建一个URLSession对象来处理网络请求。然后，我们可以使用dataTask(with:completionHandler:)方法来发送HTTP HEAD请求并获取文件的元数据。在响应的头部信息中，我们可以找到Content-Length字段来获得文件的大小。

下面是一个例子，展示了如何从URL获取文件大小：

func getFileSize(url: URL, completion: @escaping (Int64?) -> Void) {
    let request = URLRequest(url: url)
    let session = URLSession.shared

    let task = session.dataTask(with: request) { (data, response, error) in
        if let error = error {
            print("Error: \(error.localizedDescription)")
            completion(nil)
            return
        }

        guard let httpResponse = response as? HTTPURLResponse else {
            print("Invalid HTTP response")
            completion(nil)
            return
        }

        if httpResponse.statusCode != 200 {
            print("HTTP Error: \(httpResponse.statusCode)")
            completion(nil)
            return
        }

        if let contentLength = httpResponse.allHeaderFields["Content-Length"] as? String,
           let size = Int64(contentLength) {
            completion(size)
        } else {
            print("Invalid Content-Length header")
            completion(nil)
        }
    }

    task.resume()
}

使用示例

let url = URL(string: "https://example.com/file.pdf")!

getFileSize(url: url) { (size) in
    if let size = size {
        print("File Size: \(size) bytes")
    } else {
        print("Unable to retrieve file size")
    }
}

以上代码会从指定URL异步获取文件大小，并输出文件大小信息。

总结

在本文中，我们学习了如何使用Swift的URLSession来从URL获取文件大小。通过发送HTTP HEAD请求，我们可以获取到文件的元数据中的Content-Length字段，并从中提取文件大小信息。这个功能对于网络优化、下载管理和资源预览等应用非常有用。