blob: 2d80e96265f9b2715319f5aec746da988e594033 (
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
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
|
require 'set'
module ActionCable
class Server < Cramp::Websocket
on_start :initialize_subscriptions
on_data :received_data
on_finish :cleanup_subscriptions
class_attribute :registered_channels
self.registered_channels = Set.new
class << self
def register_channels(*channel_classes)
registered_channels.merge(channel_classes)
end
end
def initialize_subscriptions
@subscriptions = {}
end
def received_data(data)
data = ActiveSupport::JSON.decode data
case data['action']
when 'subscribe'
subscribe_channel(data)
when 'unsubscribe'
unsubscribe_channel(data)
when 'message'
process_message(data)
end
end
def cleanup_subscriptions
@subscriptions.each do |id, channel|
channel.unsubscribe
end
end
def publish(data)
render data
end
private
def subscribe_channel(data)
id_key = data['identifier']
id_options = ActiveSupport::JSON.decode(id_key).with_indifferent_access
if subscription = registered_channels.detect { |channel_klass| channel_klass.matches?(id_options) }
@subscriptions[id_key] = subscription.new(self, id_key, id_options)
@subscriptions[id_key].subscribe
else
# No channel found
end
end
def process_message(message)
id_key = message['identifier']
if @subscriptions[id_key]
@subscriptions[id_key].receive(ActiveSupport::JSON.decode message['data'])
end
end
def unsubscribe_channel(data)
id_key = data['identifier']
@subscriptions[id_key].unsubscribe
@subscriptions.delete(id_key)
end
end
end
|
