首页 > 解决方案 > Swift Group Tasks 未并行运行

问题描述

问题:

输出:

started 24
finished 24
--- 24
started 12
finished 12
--- 12
started 16
finished 16
--- 16
started 4
finished 4
--- 4
started 10
finished 10
--- 10
started 20
finished 20
--- 20
started 19
finished 19
--- 19

环境:

代码:

func groupTask() async -> [Int : Int] {
    let ids = [24, 12, 16, 4, 10, 20, 19]
    var prices = [Int : Int]()
    
    for id in ids {
        await withTaskGroup(of: (Int, Int).self) { group in
            group.addTask(priority: .background) {
                let price = await computePrice(for: id)
                return (id, price)
            }
            
            for await (id, price) in group {
                prices[id] = price
                print("--- \(id)")
            }
        }
    }
    
    return prices
}


func computePrice(for id: Int) async -> Int {
    print("started \(id)")
    let duration = UInt64(id * 100_000_000)
    await Task.sleep(duration)
    print("finished \(id)")
    return id * 2
}

标签: swiftasync-awaitconcurrencyswift-concurrency

解决方案


为什么组任务不并行运行?

您有await withTaskGroup内部 for-in,因此您的代码在每次迭代时等待组任务完成。

你可能想做这样的事情?

    func groupTask() async -> [Int : Int] {
        let ids = [24, 12, 16, 4, 10, 20, 19]
        var prices = [Int : Int]()
        
        await withTaskGroup(of: (Int, Int).self) { group in
            for id in ids {
                group.addTask(priority: .background) {
                    let price = await computePrice(for: id)
                    return (id, price)
                }
            }
            
            for await (id, price) in group {
                prices[id] = price
                print("--- \(id)")
            }
        }
        
        return prices
    }

推荐阅读