lib/async/tools.rb in async-tools-0.2.2 vs lib/async/tools.rb in async-tools-0.2.4

@@ -20,11 +20,13 @@
   module Tools # rubocop:disable Style/ClassAndModuleChildren
     class Error < StandardError
     end
   end
 
-  def self.map(collection, **params, &)
-    WorkerPool.with(queue_limit: collection.count, **params) do |pool|
-      pool.schedule_all(collection, &).map(&:wait)
+  def self.map(collection, concurrency: nil, parent: Async::Task.current, &)
+    Async::Semaphore.new(concurrency || collection.count, parent:).then do |s|
+      collection.map do |item|
+        s.async { yield(item) }
+      end.map(&:wait)
     end
   end
 end