如何在Swift并发中限制并发性（例如`maxConcurrentOperationCount`）？

Question

如何在Swift并发中限制并发性（例如`maxConcurrentOperationCount`）？

19

我正在尝试执行一系列网络请求，并希望在新的 Swift 并发系统中限制并行任务的数量。在操作队列中，我们会使用maxConcurrentOperationCount。在 Combine 中，使用flatMap(maxPublishers:_:)。在新的 Swift 并发系统中有什么等效的方法？

例如，这可能不是特别相关，但考虑一下：

func downloadAll() async throws {
    try await withThrowingTaskGroup(of: Void.self) { group in
        for index in 0..<20 {
            group.addTask { try await self.download(index) }
        }

        try await group.waitForAll()
    }
}

这导致所有请求同时运行。

事实上，URLSession不遵守httpMaximumConnectionsPerHost的规定确实很有趣，但这并不是关键问题。我更关心的是如何在一系列并行运行的异步任务中限制并发程度。

- Rob

1个回答

网页内容由stack overflow 提供, 点击上面的

可以查看英文原文，
原文链接

- Rob · Accepted Answer

在达到一定计数之后，可以在循环中插入一个group.next()调用，例如：

func downloadAll() async throws {
    try await withThrowingTaskGroup(of: Void.self) { group in
        for index in 0..<20 {
            if index >= 6 { try await group.next() }
            group.addTask { try await self.download(index) }
        }

        try await group.waitForAll()
    }
}

每次最多只能有六个结果。

为了完整起见，我应该注意到在WWDC 2023中，苹果提出了一种替代模式：超越结构化并发的基础知识。

withTaskGroup(of: Something.self) { group in
    for _ in 0 ..< maxConcurrentTasks {
        group.addTask { … }
    }
    while let <partial result> = await group.next() {
        if !shouldStop {
            group.addTask { … }
        }
    }
}

在这个例子中，可能会翻译成：

func downloadAll() async throws {
    try await withThrowingTaskGroup(of: Void.self) { group in
        for index in 0..<6 {
            group.addTask { try await self.download(index) }
        }
        var index = 6
        while try await group.next() != nil {
            if index < 20 {
                group.addTask { [index] in try await self.download(index) }
            }
            index += 1
        }
    }
}

投资品中的收益：

这个想法非常相似，即你可以在最大并发度上使用 group.addTask {…}，但之后再使用 group.next() 来添加每个后续任务。这是另一种解决方法。