收到了 Apple 的“爱斯基摩人”Quinn 提供的大量信息。
唉,你的做法搞错了。 URLSessionStreamTask 用于处理裸 TCP(或 TCP 上的 TLS)连接,顶部没有 HTTP 框架。您可以将其视为相当于 BSD Sockets API 的高级版本。
分块传输编码是 HTTP 的一部分,因此受到所有其他 URLSession 任务类型(数据任务、上传任务、下载任务)的支持。您无需执行任何特殊操作即可实现此功能。分块传输编码是 HTTP 1.1 标准的强制部分,因此始终启用。
但是,您可以选择如何接收返回的数据。如果您使用 URLSession 便利 API (dataTask(with:completionHandler:)
等等),URLSession 将缓冲所有传入的数据,然后将其传递给一个大的完成处理程序Data
价值。这在很多情况下都很方便,但对于流式资源来说效果不佳。在这种情况下,您需要使用基于 URLSession 委托的 API (dataTask(with:)
等等),这将调用urlSession(_:dataTask:didReceive:)
会话委托方法在数据块到达时进行处理。
至于我正在测试的特定端点,发现了以下内容:
看来服务器只有在客户端向其发送流请求时才启用其流响应(分块传输编码)。这有点奇怪,并且绝对不是 HTTP 规范所要求的。
幸运的是,可以强制 URLSession 发送流请求:
创建您的任务uploadTask(withStreamedRequest:)
实施urlSession(_:task:needNewBodyStream:)
用于返回输入流的委托方法,该输入流在读取时返回请求正文
Profit!
我附上了一些测试代码来展示这一点。在这种情况下,它使用一对绑定的流,将输入流传递给请求(按照上面的步骤 2)并保留输出流。
如果您想实际发送数据作为请求正文的一部分,您可以通过写入输出流来实现。
class NetworkManager : NSObject, URLSessionDataDelegate {
static var shared = NetworkManager()
private var session: URLSession! = nil
override init() {
super.init()
let config = URLSessionConfiguration.default
config.requestCachePolicy = .reloadIgnoringLocalCacheData
self.session = URLSession(configuration: config, delegate: self, delegateQueue: .main)
}
private var streamingTask: URLSessionDataTask? = nil
var isStreaming: Bool { return self.streamingTask != nil }
func startStreaming() {
precondition( !self.isStreaming )
let url = URL(string: "ENTER STREAMING URL HERE")!
let request = URLRequest(url: url)
let task = self.session.uploadTask(withStreamedRequest: request)
self.streamingTask = task
task.resume()
}
func stopStreaming() {
guard let task = self.streamingTask else {
return
}
self.streamingTask = nil
task.cancel()
self.closeStream()
}
var outputStream: OutputStream? = nil
private func closeStream() {
if let stream = self.outputStream {
stream.close()
self.outputStream = nil
}
}
func urlSession(_ session: URLSession, task: URLSessionTask, needNewBodyStream completionHandler: @escaping (InputStream?) -> Void) {
self.closeStream()
var inStream: InputStream? = nil
var outStream: OutputStream? = nil
Stream.getBoundStreams(withBufferSize: 4096, inputStream: &inStream, outputStream: &outStream)
self.outputStream = outStream
completionHandler(inStream)
}
func urlSession(_ session: URLSession, dataTask: URLSessionDataTask, didReceive data: Data) {
NSLog("task data: %@", data as NSData)
}
func urlSession(_ session: URLSession, task: URLSessionTask, didCompleteWithError error: Error?) {
if let error = error as NSError? {
NSLog("task error: %@ / %d", error.domain, error.code)
} else {
NSLog("task complete")
}
}
}
您可以从任何地方调用网络代码,例如:
class MainViewController : UITableViewController {
override func tableView(_ tableView: UITableView, didSelectRowAt indexPath: IndexPath) {
if NetworkManager.shared.isStreaming {
NetworkManager.shared.stopStreaming()
} else {
NetworkManager.shared.startStreaming()
}
self.tableView.deselectRow(at: indexPath, animated: true)
}
}
希望这可以帮助。