Add Async::Container::Supervisor::Supervised environemnt for worker setup.

Author: ioquatix

example/simple/simple.rb

@@ -12,7 +12,9 @@ def setup(container)
 
 		container.run(name: self.class.name, count: 4, restart: true, health_check_timeout: 2) do |instance|
 			Async do
-				Async::Container::Supervisor::Worker.new(instance, endpoint: @evaluator.supervisor_endpoint).run
+				if @environment.implements?(Async::Container::Supervisor::Supervised)
+					@evaluator.make_supervised_worker(instance).run
+				end
 
 				start_time = Time.now
 
@@ -37,7 +39,7 @@ def setup(container)
 service "sleep" do
 	service_class SleepService
 
-	supervisor_endpoint {Async::Container::Supervisor.endpoint}
+	include Async::Container::Supervisor::Supervised
 end
 
 service "supervisor" do

lib/async/container/supervisor.rb

@@ -12,4 +12,4 @@
 require_relative "supervisor/memory_monitor"
 
 require_relative "supervisor/environment"
-require_relative "supervisor/service"
+require_relative "supervisor/supervised"

lib/async/container/supervisor/client.rb

@@ -58,7 +58,7 @@ def run
 							Console.error(self, "Connection failed:", exception: error)
 							sleep(rand)
 						ensure
-							connection.close
+							connection&.close
 						end
 					end
 				end

lib/async/container/supervisor/environment.rb

@@ -5,6 +5,8 @@
 
 require "async/service/environment"
 
+require_relative "service"
+
 module Async
 	module Container
 		module Supervisor

lib/async/container/supervisor/supervised.rb

@@ -0,0 +1,30 @@
+# frozen_string_literal: true
+
+# Released under the MIT License.
+# Copyright, 2025, by Samuel Williams.
+
+require "async/service/environment"
+
+module Async
+	module Container
+		module Supervisor
+			module Supervised
+				# The IPC path to use for communication with the supervisor.
+				# @returns [String]
+				def supervisor_ipc_path
+					::File.expand_path("supervisor.ipc", root)
+				end
+				
+				# The endpoint the supervisor will bind to.
+				# @returns [::IO::Endpoint::Generic]
+				def supervisor_endpoint
+					::IO::Endpoint.unix(supervisor_ipc_path)
+				end
+				
+				def make_supervised_worker(instance)
+					Worker.new(instance, endpoint: supervisor_endpoint)
+				end
+			end
+		end
+	end
+end

test/async/container/supervised.rb

@@ -0,0 +1,54 @@
+# frozen_string_literal: true
+
+# Released under the MIT License.
+# Copyright, 2025, by Samuel Williams.
+
+require "async/container/supervisor/a_server"
+require "async/container/supervisor/supervised"
+
+class SleepService < Async::Service::Generic
+	def setup(container)
+		super
+		
+		container.spawn(name: self.class.name) do |instance|
+			Async do
+				if @environment.implements?(Async::Container::Supervisor::Supervised)
+					@evaluator.make_supervised_worker(instance).run
+				end
+				
+				instance.ready!
+				
+				sleep
+			end
+		end
+	end	
+end
+
+describe Async::Container::Supervisor::Supervised do
+	include Async::Container::Supervisor::AServer
+	
+	let(:state) do
+		{process_id: ::Process.pid}
+	end
+	
+	it "can define a supervised service" do
+		environment = Async::Service::Environment.build(root: @root) do
+			service_class {SimpleService}
+			
+			include Async::Container::Supervisor::Supervised
+		end
+		
+		evaluator = environment.evaluator
+		worker = evaluator.make_supervised_worker(state)
+		worker_task = worker.run
+		
+		sleep(0.001) until registration_monitor.registrations.any?
+		
+		connection = registration_monitor.registrations.first
+		expect(connection.state).to have_keys(
+			process_id: be == ::Process.pid
+		)
+	ensure
+		worker_task&.stop
+	end
+end

test/async/container/supervisor.rb

@@ -13,8 +13,8 @@
 	end
 
 	it "can connect to a server" do
-		client = Async::Container::Supervisor::Worker.new(state, endpoint: endpoint)
-		connection = client.connect
+		worker = Async::Container::Supervisor::Worker.new(state, endpoint: endpoint)
+		connection = worker.connect
 
 		# Wait for the client to connect to the server:
 		sleep(0.001) until registration_monitor.registrations.any?
@@ -23,12 +23,14 @@
 		expect(connection.state).to have_keys(
 			process_id: be == ::Process.pid
 		)
+	ensure
+		connection&.close
 	end
 
-	with "do_memory_dump" do
+	with "do: :memory_dump" do
 		it "can dump memory" do
-			client = Async::Container::Supervisor::Worker.new(state, endpoint: endpoint)
-			client_task = client.run
+			worker = Async::Container::Supervisor::Worker.new(state, endpoint: endpoint)
+			worker_task = worker.run
 
 			sleep(0.001) until registration_monitor.registrations.any?
 
@@ -38,7 +40,7 @@
 
 			expect(File.size(path)).to be > 0
 		ensure
-			client_task&.stop
+			worker_task&.stop
 		end
 	end
 end