文章详情

短信预约-IT技能 免费直播动态提醒

请输入下面的图形验证码

提交验证

短信预约提醒成功

Swift AsyncThrowingStream 和 AsyncStream 代码实例详解

2024-12-13 15:40

关注

前言

AsyncThrowingStream 和 AsyncStream 是 Swift 5.5 中由 SE-314[1] 引入的并发框架的一部分。异步流允许你替换基于闭包或 Combine 发布器的现有代码。

在深入研究围绕抛出流的细节之前,如果你还没有阅读我的文章,我建议你先阅读我的文章,内容包括async-await。本文解释的大部分代码将使用那里解释的API。

什么是 AsyncThrowingStream?

你可以把 AsyncThrowingStream 看作是一个有可能导致抛出错误的元素流。他的值随着时间的推移而传递,流可以通过一个结束事件来关闭。一旦发生错误,结束事件既可以是成功,也可以是失败。

什么是 AsyncStream?

AsyncStream 类似于抛出的变体,但绝不会导致抛出错误。一个非抛出型的异步流会根据明确的完成调用或流的取消而完成。

注意: 在这篇文章中,我们将解释如何使用AsyncThrowingStream。除了发生错误处理的部分,代码示例与AsyncStream类似。

AsyncThrowingStream

如何使用 AsyncThrowingStream

AsyncThrowingStream 可以很好地替代现有的基于闭包的代码,如进度和完成处理程序。为了更好地理解我的意思,我将向你介绍我们在 WeTransfer 应用程序中遇到的一个场景。

在我们的应用程序中,我们有一个基于闭包的现有类,叫做 FileDownloader:

struct FileDownloader {
enum Status {
case downloading(Float)
case finished(Data)
}

func download(_ url: URL, progressHandler: (Float) -> Void, completion: (Result<Data, Error>) -> Void) throws {
// .. Download implementation
}
}

文件下载器接受一个URL,报告进度情况,并完成一个包含下载数据的结果或在失败时显示一个错误。

文件下载器在文件下载过程中报告一个数值流。在这种情况下,它报告的是一个状态值流,以报告正在运行的下载的当前状态。FileDownloader 是一个完美的例子,你可以重写一段代码来使用 AsyncThrowingStream。然而,重写需要你在实现层面上也重写你的代码,所以让我们定义一个重载方法来代替:

extension FileDownloader {
func download(_ url: URL) -> AsyncThrowingStream<Status, Error> {
return AsyncThrowingStream { continuation in
do {
try self.download(url, progressHandler: { progress in
continuation.yield(.downloading(progress))
}, completion: { result in
switch result {
case .success(let data):
continuation.yield(.finished(data))
continuation.finish()
case .failure(let error):
continuation.finish(throwing: error)
}
})
} catch {
continuation.finish(throwing: error)
}
}
}
}

正如你所看到的,我们把下载方法包裹在一个 AsyncThrowingStream 里面。我们将流的值 Status 的类型描述为一个通用的类型,允许我们用状态更新来延续流。

只要有错误发生,我们就会通过抛出一个错误来完成流。在完成处理程序的情况下,我们要么通过抛出一个错误来完成,要么用一个不抛出的完成回调来跟进数据的产生。

switch result {
case .success(let data):
continuation.yield(.finished(data))
continuation.finish()
case .failure(let error):
continuation.finish(throwing: error)
}

在收到最后的状态更新后,不要忘记 finish() 回调,这一点至关重要。否则,我们将保持流的存活,而实现层面的代码将永远不会继续。

我们可以通过使用另一个 yield 方法来重写上述代码,接受一个 Result 枚举作为参数:

continuation.yield(with: result.map { .finished($0) })
continuation.finish()

重写后的代码简化了我们的代码,并去掉了 switch-case 代码。我们必须映射我们的 Reslut 枚举以匹配预期的 Status 值。如果我们产生一个失败的结果,我们的流将在抛出包含的错误后结束。

AsyncThrowingStream 迭代

一旦你配置好你的异步抛出流,你就可以开始在数值流上进行迭代。在我们的 FileDownloader 例子中,它将看起来如下所示:

do {
for try await status in download(url) {
switch status {
case .downloading(let progress):
print("Downloading progress: \(progress)")
case .finished(let data):
print("Downloading completed with data: \(data)")
}
}
print("Download finished and stream closed")
} catch {
print("Download failed with \(error)")
}

我们处理任何状态的更新,并且我们可以使用 catch 闭包来处理任何发生的错误。你可以使用基于 AsyncSequence 接口的 for ... in 循环进行迭代,这对 AsyncStream 来说是一样的。

如果你遇到了类似的编译错误:

‘async’ in a function that does not support concurrency

上述代码示例中的打印语句有助于你理解 AsyncThrowingStream 的生命周期。你可以替换打印语句来处理进度更新和处理数据,为你的用户实现可视化。

调试 AsyncStream

如果一个流不能报告数值,我们可以通过放置断点来调试流产生的回调。虽然也可能是上面的 “Download finished and stream closed” 的打印语句不会调用,这意味着你在实现层的代码永远不会继续。后者可能是一个未完成的流的结果。

为了验证,我们可以利用 onTermination 回调:

func download(_ url: URL) -> AsyncThrowingStream {
return AsyncThrowingStream { continuation in

/// 配置一个终止回调,以了解你的流的生命周期。
continuation.onTermination = { @Sendable status in
print("Stream terminated with status \(status)")
}

// ..
}
}

回调在流终止时被调用,它将告诉你你的流是否还活着。

如果出现了错误,输出结果可能如下:

Stream terminated with status finished(Optional(FileDownloader.FileDownloadingError.example))

上述输出只有在使用 AsyncThrowingStream 时才能实现。如果是一个普通的 AsyncStream,完成的输出看起来如下:

Stream terminated with status finished

而取消的结果对这两种类型的流来说都是这样的:

Stream terminated with status cancelled

你也可以在流结束后使用这个终止回调进行任何清理。例如,删除任何观察者或在文件下载后清理磁盘空间。

取消一个 AsyncStream

一个 AsyncStream 或 AsyncThrowingStream 可以由于一个封闭的任务被取消而取消。一个例子可以如下:

let task = Task.detached {
do {
for try await status in download(url) {
switch status {
case .downloading(let progress):
print("Downloading progress: \(progress)")
case .finished(let data):
print("Downloading completed with data: \(data)")
}
}
} catch {
print("Download failed with \(error)")
}
}
task.cancel()

一个流在超出范围或包围的任务取消时就会取消。如前所述,取消将相应地触发 onTermination 回调。

结论

AsyncThrowingStream 或 AsyncStream 是重写基于闭包的现有代码到支持 async-awai t的替代品的好方法。你可以提供一个连续的值流,并在成功或失败时完成一个流。你可以使用基于 AsyncSequence APIs 的 for 循环在实现层面上迭代值。

参考资料

[1]SE-314: ​https://github.com/apple/swift-evolution/blob/main/proposals/0314-async-stream.md。​

来源:Swift社区内容投诉

免责声明:

① 本站未注明“稿件来源”的信息均来自网络整理。其文字、图片和音视频稿件的所属权归原作者所有。本站收集整理出于非商业性的教育和科研之目的,并不意味着本站赞同其观点或证实其内容的真实性。仅作为临时的测试数据,供内部测试之用。本站并未授权任何人以任何方式主动获取本站任何信息。

② 本站未注明“稿件来源”的临时测试数据将在测试完成后最终做删除处理。有问题或投稿请发送至: 邮箱/279061341@qq.com QQ/279061341

软考中级精品资料免费领

  • 历年真题答案解析
  • 备考技巧名师总结
  • 高频考点精准押题
  • 2024年上半年信息系统项目管理师第二批次真题及答案解析(完整版)

    难度     813人已做
    查看
  • 【考后总结】2024年5月26日信息系统项目管理师第2批次考情分析

    难度     354人已做
    查看
  • 【考后总结】2024年5月25日信息系统项目管理师第1批次考情分析

    难度     318人已做
    查看
  • 2024年上半年软考高项第一、二批次真题考点汇总(完整版)

    难度     435人已做
    查看
  • 2024年上半年系统架构设计师考试综合知识真题

    难度     224人已做
    查看

相关文章

发现更多好内容

猜你喜欢

AI推送时光机
位置:首页-资讯-后端开发
咦!没有更多了?去看看其它编程学习网 内容吧
首页课程
资料下载
问答资讯