Add semaphore to run 10 scans at once

duckduckgo · Mar 14, 2024 · f04373e · f04373e
1 parent b9321fb
commit f04373e
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/...kerProtection/Sources/DataBrokerProtection/DebugUI/DataBrokerRunCustomJSONViewModel.swift b/...kerProtection/Sources/DataBrokerProtection/DebugUI/DataBrokerRunCustomJSONViewModel.swift
@@ -184,13 +184,18 @@ final class DataBrokerRunCustomJSONViewModel: ObservableObject {
 
         Task.detached {
             var scanResults = [DebugScanReturnValue]()
+            let semaphore = DispatchSemaphore(value: 10)
             try await withThrowingTaskGroup(of: DebugScanReturnValue.self) { group in
                 for queryData in brokerProfileQueryData {
+                    semaphore.wait()
                     let debugScanOperation = DebugScanOperation(privacyConfig: self.privacyConfigManager, prefs: self.contentScopeProperties, query: queryData) {
                         true
                     }
 
                     group.addTask {
+                        defer {
+                            semaphore.signal()
+                        }
                         do {
                             return try await debugScanOperation.run(inputValue: (), stageCalculator: FakeStageDurationCalculator(), showWebView: false)
                         } catch {