Add support for IO::Event::WorkerPool.

Author: samuel-williams-shopify

gems.rb

@@ -9,7 +9,7 @@
 
 gemspec
 
-# gem "io-event", git: "https://github.com/socketry/io-event.git"
+gem "io-event", git: "https://github.com/samuel-williams-shopify/io-event.git", branch: "native-work-pool"
 
 # In order to capture both code paths in coverage, we need to optionally load this gem:
 if ENV["FIBER_PROFILER_CAPTURE"] == "true"

lib/async/scheduler.rb

@@ -8,7 +8,6 @@
 require_relative "clock"
 require_relative "task"
 require_relative "timeout"
-require_relative "worker_pool"
 
 require "io/event"
 
@@ -45,6 +44,22 @@ def initialize(message = "Scheduler is closed!")
 		def self.supported?
 			true
 		end
+
+		# Used to augment the scheduler to add support for blocking operations.
+		module BlockingOperationWait
+			# Wait for the given work to be executed.
+			#
+			# @public Since *Async v2.21* and *Ruby v3.4*.
+			# @asynchronous May be non-blocking.
+			#
+			# @parameter work [Proc] The work to execute on a background thread.
+			# @returns [Object] The result of the work.
+			def blocking_operation_wait(work)
+				@worker_pool.call(work)
+			end
+		end
+
+		private_constant :BlockingOperationWait
 
 		# Create a new scheduler.
 		#
@@ -65,14 +80,19 @@ def initialize(parent = nil, selector: nil, profiler: Profiler&.default, worker_
 			@idle_time = 0.0
 
 			@timers = ::IO::Event::Timers.new
+			
 			if worker_pool == true
-				@worker_pool = WorkerPool.new
+				if ::IO::Event.const_defined?(:WorkerPool)
+					@worker_pool = ::IO::Event::WorkerPool.new
+				else
+					@worker_pool = nil
+				end
 			else
 				@worker_pool = worker_pool
 			end
-			
+
 			if @worker_pool
-				self.singleton_class.prepend(WorkerPool::BlockingOperationWait)
+				self.singleton_class.prepend(BlockingOperationWait)
 			end
 		end