blob: 373b9d7fe32231927bd01ff2702dbbd278df3395 (
plain) (
blame)
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
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
|
# frozen_string_literal: true
module ActionCable
module SubscriptionAdapter
class SubscriberMap
def initialize
@subscribers = Hash.new { |h, k| h[k] = [] }
@sync = Mutex.new
end
def add_subscriber(channel, subscriber, on_success)
@sync.synchronize do
new_channel = !@subscribers.key?(channel)
@subscribers[channel] << subscriber
if new_channel
add_channel channel, on_success
elsif on_success
on_success.call
end
end
end
def remove_subscriber(channel, subscriber)
@sync.synchronize do
@subscribers[channel].delete(subscriber)
if @subscribers[channel].empty?
@subscribers.delete channel
remove_channel channel
end
end
end
def broadcast(channel, message)
list = @sync.synchronize do
return if !@subscribers.key?(channel)
@subscribers[channel].dup
end
list.each do |subscriber|
invoke_callback(subscriber, message)
end
end
def add_channel(channel, on_success)
on_success.call if on_success
end
def remove_channel(channel)
end
def invoke_callback(callback, message)
callback.call message
end
end
end
end
|
