iOS 15环境下TaskGroup任务无法并行运行的问题修复方案咨询

老哥，我之前也踩过iOS15结构化并发的坑，咱们先拆解你遇到的两个核心问题：iOS15下TaskGroup任务无法并行执行、iOS15下所有任务都会跑完而不是仅第一个成功后取消其余任务，这俩其实都和iOS15对结构化并发的早期实现限制有关。

问题根源分析 #

1. TaskGroup调度逻辑的版本差异：
   iOS16对TaskGroup的任务调度做了优化，会默认把任务分发到不同的执行线程；但iOS15里，TaskGroup的任务大概率会被调度到同一个串行执行上下文，导致看起来是单线程运行，CPU占用也就只有100%左右。
   另外你代码里用的Int.random(in:)，在iOS15的早期实现里可能存在隐性全局锁（虽然官方标注线程安全），多任务抢锁时直接就变成了串行执行，进一步放大了问题。

2. 取消信号传递不及时：
   你用defer { group.cancelAll() }来在拿到第一个结果后取消其余任务，但iOS15里group.next()返回后，取消信号没法及时传递给正在运行的任务；再加上如果任务里没有频繁的await暂停点（比如你一开始没加Task.yield()），任务根本没机会检查Task.isCancelled，自然会一直运行到完成，这就是iOS15下四个任务都打印成功的原因。

具体修复方案 #

咱们从调度策略、取消检查、性能优化三个方面调整代码：

1. 强制任务在独立上下文运行（解决并行问题） #

iOS15里仅给TaskGroup任务设置优先级不够，咱们用Task.detached强制每个生成任务脱离当前上下文，让系统把它们分发到不同的后台线程：

private nonisolated func generatePuzzle() async -> Puzzle? {
    await withTaskGroup(of: Puzzle?.self) { group in
        for i in 0..<4 {
            group.addTask(priority: .userInitiated) {
                // 用detached创建独立执行的任务，强制iOS15做并行调度
                await Task.detached(priority: .userInitiated) {
                    let worker = PuzzleGenerator(index: i)
                    return await worker.generatePuzzle()
                }.value
            }
        }
        
        defer { group.cancelAll() }
        return await group.next() ?? nil
    }
}

2. 确保任务频繁检查取消状态（解决取消不及时问题） #

你已经发现加await Task.yield()能缓解问题，这是因为yield()会给任务一个暂停点，让系统有机会传递取消信号。在生成循环里必须保证每次循环都有await操作，让Task.isCancelled能被及时检测到：

func generatePuzzle() async -> Puzzle? {
    var val: Int
    repeat {
        val = Int.random(in: 0..<100_000_000)
        // 每次循环都让出执行权，触发取消检查
        await Task.yield()
    } while val > 0 && !Task.isCancelled
    
    if val == 0 {
        print("puzzle generated successully by generator \(index)")
    }
    return val > 0 ? nil : Puzzle(val: val)
}

如果你的真实生成逻辑有长耗时的无await计算块，最好拆成多个小片段，每个片段后加await Task.yield()，或者手动抛出CancellationError()来快速终止任务。

3. 优化随机数生成（可选性能提升） #

iOS15的Int.random(in:)可能存在全局锁竞争，咱们给每个任务分配独立的随机数生成器，避免锁阻塞：

func generatePuzzle() async -> Puzzle? {
    var rng = SystemRandomNumberGenerator() // 每个任务用独立的生成器
    var val: Int
    repeat {
        val = Int.random(in: 0..<100_000_000, using: &rng)
        await Task.yield()
    } while val > 0 && !Task.isCancelled
    
    if val == 0 {
        print("puzzle generated successully by generator \(index)")
    }
    return val > 0 ? nil : Puzzle(val: val)
}

验证效果 #

调整后在iOS15模拟器上运行，应该能看到CPU占用提升到300%-400%，而且只有第一个成功的任务会打印日志，其余任务会被及时取消。

另外兼容iOS15时还要注意：

• iOS15的withTaskGroup不支持maxConcurrentTaskCount参数（iOS16+才支持），所以用Task.detached是更可靠的并行方式；
• 尽量避免在TaskGroup里嵌套复杂异步逻辑，iOS15对嵌套任务的调度支持不如iOS16稳定。

内容来源于stack exchange