Allow PoolManager to incorrectly resume a task from within a task

lib/celluloid/pool_manager.rb

@@ -91,6 +91,7 @@ def idle_size
 
     # Provision a new worker
     def __provision_worker__
+      Task.current.guard_warnings = true
       while @idle.empty?
         # Wait for responses from one of the busy workers
         response = exclusive { receive { |msg| msg.is_a?(Response) } }

lib/celluloid/tasks.rb

@@ -25,7 +25,7 @@ def self.suspend(status)
     end
 
     attr_reader :type, :meta, :status
-    attr_accessor :chain_id
+    attr_accessor :chain_id, :guard_warnings
 
     # Create a new task
     def initialize(type, meta)
@@ -35,6 +35,7 @@ def initialize(type, meta)
 
       @exclusive         = false
       @dangerous_suspend = @meta ? @meta.delete(:dangerous_suspend) : false
+      @guard_warnings    = false
 
       actor     = Thread.current[:celluloid_actor]
       @chain_id = CallChain.current_id
@@ -93,7 +94,7 @@ def suspend(status)
 
     # Resume a suspended task, giving it a value to return if needed
     def resume(value = nil)
-      raise "Cannot resume a task from inside of a task" if Thread.current[:celluloid_task]
+      guard "Cannot resume a task from inside of a task" if Thread.current[:celluloid_task]
       deliver(value)
       nil
     end
@@ -141,6 +142,14 @@ def running?; @status != :dead; end
     def inspect
       "#<#{self.class}:0x#{object_id.to_s(16)} @type=#{@type.inspect}, @meta=#{@meta.inspect}, @status=#{@status.inspect}>"
     end
+
+    def guard(message)
+      if @guard_warnings
+        Logger.warn message if $CELLULOID_DEBUG
+      else
+        raise message
+      end
+    end
   end
 end
 

spec/celluloid/pool_spec.rb

@@ -49,4 +49,11 @@ def crash
   it "terminates" do
     expect { subject.terminate }.to_not raise_exception
   end
+
+  it "handles many requests" do
+    futures = 10.times.map do
+      subject.future.process
+    end
+    futures.map(&:value)
+  end
 end