Improved RPC mechanism.

Author: ioquatix

async-container-supervisor.gemspec

@@ -24,8 +24,8 @@ Gem::Specification.new do |spec|
 
 	spec.required_ruby_version = ">= 3.1"
 
-	spec.add_dependency "async-container"
+	spec.add_dependency "async-container", "~> 0.22"
 	spec.add_dependency "async-service"
 	spec.add_dependency "io-stream"
-	spec.add_dependency "memory-leak", "~> 0.2"
+	spec.add_dependency "memory-leak", "~> 0.3"
 end

example/simple/simple.rb

@@ -6,20 +6,23 @@ class SleepService < Async::Service::Generic
 	def setup(container)
 		super
 
-		container.run(count: 1, restart: true, health_check_timeout: 2) do |instance|
+		container.run(name: self.class.name, count: 4, restart: true, health_check_timeout: 2) do |instance|
 			Async do
 				client = Async::Container::Supervisor::Client.new(instance, @evaluator.supervisor_endpoint)
 				client.run
 
+				start_time = Time.now
+				
 				instance.ready!
 
 				chunks = []
 				while true
-					Console.info(self, "Allocating memory...")
-					# Allocate 10MB of memory every second:
-					chunks << " " * 1024 * 1024
-					sleep 0.1
+					chunks << " " * 1024 * 1024 * rand(10) 
+					sleep 1
 					instance.ready!
+					
+					uptime = Time.now - start_time
+					instance.name = "Sleeping for #{uptime.to_i} seconds..."
 				end
 			ensure
 				Console.info(self, "Exiting...")
@@ -38,6 +41,6 @@ def setup(container)
 	include Async::Container::Supervisor::Environment
 
 	monitors do
-		[Async::Container::Supervisor::Monitor::MemoryMonitor.new(interval: 1, limit: 1024 * 1024 * 100)]
+		[Async::Container::Supervisor::MemoryMonitor.new(interval: 1, limit: 1024 * 1024 * 100)]
 	end
 end

gems.locked

@@ -1,17 +1,11 @@
-PATH
-  remote: ../async-container
-  specs:
-    async-container (0.21.0)
-      async (~> 2.22)
-
 PATH
   remote: .
   specs:
     async-container-supervisor (0.0.0)
-      async-container
+      async-container (~> 0.22)
       async-service
       io-stream
-      memory-leak (~> 0.2)
+      memory-leak (~> 0.3)
 
 GEM
   remote: https://rubygems.org/
@@ -23,6 +17,8 @@ GEM
       io-event (~> 1.9)
       metrics (~> 0.12)
       traces (~> 0.15)
+    async-container (0.22.0)
+      async (~> 2.22)
     async-http (0.87.0)
       async (>= 2.10.2)
       async-pool (~> 0.9)
@@ -104,7 +100,7 @@ GEM
       net-smtp
     mapping (1.1.1)
     markly (0.12.1)
-    memory-leak (0.2.0)
+    memory-leak (0.4.0)
     metrics (0.12.1)
     mime-types (3.6.0)
       logger
@@ -148,7 +144,7 @@ GEM
       variant
     rainbow (3.1.1)
     regexp_parser (2.10.0)
-    rubocop (1.72.2)
+    rubocop (1.73.0)
       json (~> 2.3)
       language_server-protocol (~> 3.17.0.2)
       lint_roller (~> 1.1.0)
@@ -206,7 +202,6 @@ PLATFORMS
   ruby
 
 DEPENDENCIES
-  async-container!
   async-container-supervisor!
   bake-gem
   bake-modernize

gems.rb

@@ -7,8 +7,6 @@
 
 gemspec
 
-gem "async-container", path: "../async-container"
-
 group :maintenance, optional: true do
 	gem "bake-gem"
 	gem "bake-modernize"

lib/async/container/supervisor.rb

@@ -7,7 +7,8 @@
 
 require_relative "supervisor/server"
 require_relative "supervisor/client"
-require_relative "supervisor/monitor"
+
+require_relative "supervisor/memory_monitor"
 
 require_relative "supervisor/environment"
 require_relative "supervisor/service"

lib/async/container/supervisor/client.rb

@@ -4,7 +4,7 @@
 # Copyright, 2025, by Samuel Williams.
 
 require "io/stream"
-require_relative "wrapper"
+require_relative "connection"
 
 module Async
 	module Container
@@ -19,47 +19,96 @@ def initialize(instance, endpoint = Supervisor.endpoint)
 					@endpoint = endpoint
 				end
 
+				def dispatch(call)
+					method_name = "do_#{call.message[:do]}"
+					self.public_send(method_name, call)
+				end
+				
 				def connect
-					unless @wrapper
+					unless @connection
 						peer = @endpoint.connect
 						stream = IO::Stream(peer)
-						@wrapper = Wrapper.new(stream)
+						@connection = Connection.new(stream, 0, instance: @instance)
 
-						@wrapper.write(action: "register", instance: @instance)
+						# Register the instance with the server:
+						Async do
+							@connection.call(do: :register, state: @instance)
+						end
 					end
 
-					return @wrapper unless block_given?
+					return @connection unless block_given?
 
 					begin
-						yield @wrapper
+						yield @connection
 					ensure
-						@wrapper.close
+						@connection.close
 					end
 				end
 
 				def close
-					if wrapper = @wrapper
-						@wrapper = nil
-						wrapper.close
+					if connection = @connection
+						@connection = nil
+						connection.close
+					end
+				end
+				
+				private def dump(call)
+					if path = call[:path]
+						File.open(path, "w") do |file|
+							yield file
+						end
+						
+						call.finish(path: path)
+					else
+						buffer = StringIO.new
+						yield buffer
+						
+						call.finish(data: buffer.string)
+					end
+				end
+				
+				def do_scheduler_dump(call)
+					dump(call) do |file|
+						Fiber.scheduler.print_hierarchy(file)
 					end
 				end
 
-				def do_memory_dump(wrapper, message)
-					Console.info(self, "Memory dump:", message)
-					path = message[:path]
+				def do_memory_dump(call)
+					require "objspace"
 
-					File.open(path, "w") do |file|
+					dump(call) do |file|
 						ObjectSpace.dump_all(output: file)
 					end
 				end
 
+				def do_thread_dump(call)
+					dump(call) do |file|
+						Thread.list.each do |thread|
+							file.puts(thread.inspect)
+							file.puts(thread.backtrace)
+						end
+					end
+				end
+				
+				def do_garbage_profile_start(connection, message)
+					Console.info(self, "Garbage profile start:", message)
+					GC::Profiler.enable
+				end
+				
+				def do_garbage_profile_stop(connection, message)
+					Console.info(self, "Garbage profile stop:", message)
+					GC::Profiler.disable
+					
+					dump(connection, message) do |file|
+						file.puts GC::Profiler.result
+					end
+				end
+				
 				def run
 					Async do |task|
 						loop do
-							Console.info(self, "Connecting to supervisor...")
-							connect do |wrapper|
-								Console.info(self, "Connected to supervisor.")
-								wrapper.run(self)
+							connect do |connection|
+								connection.run(self)
 							end
 						rescue => error
 							Console.error(self, "Unexpected error while running client!", exception: error)