主页 / coding / 问题 / 77045906
Accepted
matt
Asked: 2023-09-05 23:42:07 +0800 CST

相互取消任务

  • 772

这是一个小测试台，显示两个同时启动的任务，无论哪个先完成都会取消另一个：

func startTest() async throws {
    Task { await test() }
    try await Task.sleep(nanoseconds: 1 * NSEC_PER_SEC)
    // comment out this next line to test the timeout
    NotificationCenter.default.post(name: .init("Test"), object: nil)
}

func test() async {
    let notificationTask = Task {
        for await _ in NotificationCenter.default
            .notifications(named: .init("Test"), object: nil)
            .prefix(1) {}
        print("Done waiting for the notification")
    }
    let timeoutTask = Task {
        try await Task.sleep(nanoseconds: 5 * NSEC_PER_SEC)
        notificationTask.cancel()
        print("I timed out and cancelled the notification task")
    }
    await notificationTask.value
    timeoutTask.cancel()
    print("finished!")
}

正如您所看到的，我们的想法是，要么我们在 5 秒内收到通知，在这种情况下，五秒计时器被取消，要么我们在 5 秒内超时，在这种情况下，等待通知被取消。

我的问题是：有没有更简洁的方式来表达这一点？我尝试了各种任务组和 async let 公式，但没有想出任何办法。也许这是“正确”的方式，但不知何故它令人烦恼。

swift

1 个回答

  1. Best Answer
    1. 做一个TaskGroup
    2. 将所有“赛车”任务添加到其中（可以超过 2 个！）
    3. 致电await tg.next()等待“比赛获胜者”的结果
    4. 打电话tg.cancelAll()取消未完成的任务
    5. ???
    6. 利润！
    import Foundation

func startTest() async throws {
    Task { try await test() }
    try await Task.sleep(nanoseconds: 1 * NSEC_PER_SEC)
    // comment out this next line to test the timeout
    NotificationCenter.default.post(name: .init("Test"), object: nil)
}


func test() async throws {
    try await withThrowingTaskGroup(of: Void.self) { group in
        group.addTask { // formerlynotificationTask
            for await _ in NotificationCenter.default
                .notifications(named: .init("Test"), object: nil)
                .prefix(1) {}
            print("Done waiting for the notification")
        }
        
        group.addTask { // formerly "timeoutTask"
            try await Task.sleep(nanoseconds: 5 * NSEC_PER_SEC)
            print("I timed out and cancelled the notification task")
        }
        
        try await group.next() // Wait for the first completed task
        group.cancelAll() // Cancel the rest
    }
    print("finished!")
}

try await startTest()
    • 3

相关问题