1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
|
module Concurrent
module Actor
module Utils
# Distributes messages between subscribed actors. Each actor'll get only one message then
# it's unsubscribed. The actor needs to resubscribe when it's ready to receive next message.
# It will buffer the messages if there is no worker registered.
# @see Pool
class Balancer < RestartingContext
def initialize
@receivers = []
@buffer = []
end
def on_message(message)
command, who = message
case command
when :subscribe
@receivers << (who || envelope.sender)
distribute
true
when :unsubscribe
@receivers.delete(who || envelope.sender)
true
when :subscribed?
@receivers.include?(who || envelope.sender)
else
@buffer << envelope
distribute
Behaviour::MESSAGE_PROCESSED
end
end
def distribute
while !@receivers.empty? && !@buffer.empty?
redirect @receivers.shift, @buffer.shift
end
end
end
end
end
end
|
