Added a shutdown hook for ActorContext and SimpleActorRef.

Author: jdantonio

demos/global_thread_pool-demo.rb

@@ -19,6 +19,8 @@ def initialize(actor, opts = {})
       @reset_on_error = opts.fetch(:reset_on_error, true)
       @exception_class = opts.fetch(:rescue_exception, false) ? Exception : StandardError
       @observers = CopyOnNotifyObserverSet.new
+
+      @actor.define_singleton_method(:shutdown, &method(:set_stop_event))
     end
 
     def running?
@@ -70,6 +72,10 @@ def join(timeout = nil)
 
     Message = Struct.new(:payload, :ivar, :callback)
 
+    def set_stop_event
+      @stop_event.set
+    end
+
     def supervise
       if @thread.nil?
         @actor.on_start
@@ -109,7 +115,10 @@ def run_message_loop
 
           observers.notify_observers(now, message.payload, result, ex)
         end
+
+        break if @stop_event.set?
       end
+      @actor.on_shutdown
     end
   end
 end

lib/concurrent/simple_actor_ref.rb

@@ -16,6 +16,8 @@ def receive(*msg)
         raise StandardError
       when :bullet
         raise Exception
+      when :stop
+        shutdown
       when :terminate
         Thread.current.kill
       when :sleep
@@ -48,6 +50,13 @@ def receive(*msg)
       sleep(0.1)
       subject.should be_shutdown
     end
+
+    it 'defines a shutdown method on the actor(s)' do
+      subject << :foo
+      subject << :stop
+      sleep(0.1)
+      subject.should be_shutdown
+    end
   end
 
   context '#post' do