Fix #185: allow Util::Pool to be asked

Author: pitr-ch

lib/concurrent/actor/utils/balancer.rb

@@ -25,14 +25,15 @@ def on_message(message)
           when :subscribed?
             @receivers.include? envelope.sender
           else
-            @buffer << message
+            @buffer << envelope
             distribute
+            Behaviour::MESSAGE_PROCESSED
           end
         end
 
         def distribute
           while !@receivers.empty? && !@buffer.empty?
-            @receivers.shift << @buffer.shift
+            redirect @receivers.shift, @buffer.shift
           end
         end
       end

lib/concurrent/actor/utils/pool.rb

@@ -34,7 +34,7 @@ def initialize(size, &worker_initializer)
         end
 
         def on_message(message)
-          @balancer << message
+          redirect @balancer
         end
       end
 

spec/concurrent/actor_spec.rb

@@ -377,6 +377,22 @@ def on_message(message)
 
       end
 
+      describe 'pool' do
+        it 'supports asks' do
+          worker = Class.new Concurrent::Actor::Utils::AbstractWorker do
+            def work(message)
+              5 + message
+            end
+          end
+
+          pool = Concurrent::Actor::Utils::Pool.spawn! 'pool', 5 do |balancer, index|
+            worker.spawn name: "worker-#{index}", supervise: true, args: [balancer]
+          end
+
+          expect(pool.ask!(5)).to eq 10
+          terminate_actors pool
+        end
+      end
 
     end
   end