Add health check support to supervisor.

Author: ioquatix

lib/falcon/environment/supervisor.rb

@@ -12,6 +12,12 @@ module Falcon
 	module Environment
 		# Provides an environment for hosting a supervisor which can monitor multiple applications.
 		module Supervisor
+			# The service class to use for the supervisor.
+			# @returns [Class]
+			def service_class
+				::Falcon::Service::Supervisor
+			end
+			
 			# The name of the supervisor
 			# @returns [String]
 			def name
@@ -30,10 +36,9 @@ def endpoint
 				::IO::Endpoint.unix(ipc_path)
 			end
 
-			# The service class to use for the supervisor.
-			# @returns [Class]
-			def service_class
-				::Falcon::Service::Supervisor
+			# Options to use when creating the container.
+			def container_options
+				{restart: true, count: 1, health_check_timeout: 30}
 			end
 		end
 

lib/falcon/service/supervisor.rb

@@ -69,7 +69,10 @@ def start
 			# Start the supervisor process which accepts connections from the bound endpoint and processes JSON formatted messages.
 			# @parameter container [Async::Container::Generic]
 			def setup(container)
-				container.run(name: self.name, restart: true, count: 1) do |instance|
+				container_options = @evaluator.container_options
+				health_check_timeout = container_options[:health_check_timeout]
+				
+				container.run(name: self.name, **container_options) do |instance|
 					Async do
 						@bound_endpoint.accept do |peer|
 							stream = ::IO::Stream(peer)
@@ -81,6 +84,15 @@ def setup(container)
 						end
 
 						instance.ready!
+						
+						if health_check_timeout
+							Async(transient: true) do
+								while true
+									sleep(health_check_timeout / 2)
+									instance.ready!
+								end
+							end
+						end
 					end
 				end