Class: ActionCable::SubscriptionAdapter::PostgreSQL::Listener
Instance Method Summary
collapse
#add_subscriber, #invoke_callback, #remove_subscriber
#add_subscriber, #broadcast, #invoke_callback, #remove_subscriber
Constructor Details
#initialize(adapter, executor) ⇒ Listener
Returns a new instance of Listener.
77
78
79
80
81
82
83
84
85
86
87
|
# File 'lib/action_cable/subscription_adapter/postgresql.rb', line 77
def initialize(adapter, executor)
super(executor)
@adapter = adapter
@queue = Queue.new
@thread = Thread.new do
Thread.current.abort_on_exception = true
listen
end
end
|
Instance Method Details
#add_channel(channel, on_success) ⇒ Object
120
121
122
|
# File 'lib/action_cable/subscription_adapter/postgresql.rb', line 120
def add_channel(channel, on_success)
@queue.push([:listen, channel, on_success])
end
|
#listen ⇒ Object
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
|
# File 'lib/action_cable/subscription_adapter/postgresql.rb', line 89
def listen
@adapter.with_subscriptions_connection do |pg_conn|
catch :shutdown do
loop do
until @queue.empty?
action, channel, callback = @queue.pop(true)
case action
when :listen
pg_conn.exec("LISTEN #{pg_conn.escape_identifier channel}")
@executor.post(&callback) if callback
when :unlisten
pg_conn.exec("UNLISTEN #{pg_conn.escape_identifier channel}")
when :shutdown
throw :shutdown
end
end
pg_conn.wait_for_notify(1) do |chan, pid, message|
broadcast(chan, message)
end
end
end
end
end
|
#remove_channel(channel) ⇒ Object
124
125
126
|
# File 'lib/action_cable/subscription_adapter/postgresql.rb', line 124
def remove_channel(channel)
@queue.push([:unlisten, channel])
end
|
#shutdown ⇒ Object
115
116
117
118
|
# File 'lib/action_cable/subscription_adapter/postgresql.rb', line 115
def shutdown
@queue.push([:shutdown])
Thread.pass while @thread.alive?
end
|