Add support for Process.fork within an active scheduler.

Author: samuel-williams-shopify

lib/async/fork_handler.rb

@@ -0,0 +1,36 @@
+# frozen_string_literal: true
+
+# Released under the MIT License.
+# Copyright, 2025, by Samuel Williams.
+
+module Async
+	# Private module that hooks into Process._fork to handle fork events.
+	module ForkHandler
+		def _fork(&block)
+			if block_given?
+				super do
+					# Child process:
+					if scheduler = Fiber.scheduler
+						scheduler.process_fork if scheduler.respond_to?(:process_fork)
+					end
+					
+					yield
+				end
+			else
+				unless pid = super
+					# Child process:
+					if scheduler = Fiber.scheduler
+						scheduler.process_fork if scheduler.respond_to?(:process_fork)
+					end
+				end
+				
+				return pid
+			end
+		end
+	end
+	
+	private_constant :ForkHandler
+	
+	# Hook into Process._fork to handle fork events automatically:
+	::Process.singleton_class.prepend(ForkHandler)
+end

lib/async/node.rb

@@ -214,7 +214,8 @@ def parent=(parent)
 		end
 
 		protected def remove_child(child)
-			@children.remove(child)
+			# In the case of a fork, the children list may be nil:
+			@children&.remove(child)
 			child.set_parent(nil)
 		end
 

lib/async/scheduler.rb

@@ -9,6 +9,7 @@
 require_relative "clock"
 require_relative "task"
 require_relative "timeout"
+require_relative "fork_handler"
 
 require "io/event"
 
@@ -146,24 +147,26 @@ def terminate
 		# Terminate all child tasks and close the scheduler.
 		# @public Since *Async v1*.
 		def close
-			self.run_loop do
-				until self.terminate
-					self.run_once!
+			unless @children.nil?
+				self.run_loop do
+					until self.terminate
+						self.run_once!
+					end
 				end
 			end
 
 			Kernel.raise "Closing scheduler with blocked operations!" if @blocked > 0
 		ensure
 			# We want `@selector = nil` to be a visible side effect from this point forward, specifically in `#interrupt` and `#unblock`. If the selector is closed, then we don't want to push any fibers to it.
-			selector = @selector
-			@selector = nil
-			
-			selector&.close
-			
-			worker_pool = @worker_pool
-			@worker_pool = nil
+			if selector = @selector
+				@selector = nil
+				selector.close
+			end
 
-			worker_pool&.close
+			if worker_pool = @worker_pool
+				@worker_pool = nil
+				worker_pool.close
+			end
 
 			consume
 		end
@@ -642,5 +645,27 @@ def timeout_after(duration, exception, message, &block)
 				yield duration
 			end
 		end
+		
+		# Handle fork in the child process. This method is called automatically when Process.fork is invoked.
+		#
+		# This method:
+		# - Terminates all tasks forcefully (without raising exceptions)
+		# - Closes the selector (kernel state doesn't survive fork)
+		# - Resets scheduler state
+		# - Closes worker pool if present
+		# - Unsets the scheduler from Fiber.scheduler
+		#
+		# The child process starts with a clean slate - no scheduler is set.
+		# Users can create a new scheduler if needed.
+		#
+		# @public Since *Async v2*.
+		def process_fork
+			@children = nil
+			@selector = nil
+			@timers = nil
+			
+			# Close the scheduler:
+			Fiber.set_scheduler(nil)
+		end
 	end
 end

test/process/fork.rb

@@ -0,0 +1,47 @@
+# frozen_string_literal: true
+
+# Released under the MIT License.
+# Copyright, 2025, by Samuel Williams.
+
+require "sus/fixtures/async"
+require "async"
+
+describe Process do
+	describe ".fork" do
+		it "can fork with block form" do
+			r, w = IO.pipe
+			
+			Async do
+				pid = Process.fork do
+					# Child process:
+					w.write("hello")
+				end
+				
+				# Parent process:
+				w.close
+				expect(r.read).to be == "hello"
+			ensure
+				Process.waitpid(pid) if pid
+			end
+		end
+		
+		it "can fork with non-block form" do
+			r, w = IO.pipe
+			
+			Async do
+				unless pid = Process.fork
+					# Child process:
+					w.write("hello")
+					
+					exit!
+				end
+				
+				# Parent process:
+				w.close
+				expect(r.read).to be == "hello"
+			ensure
+				Process.waitpid(pid) if pid
+			end
+		end
+	end
+end