Class: ActionCable::SubscriptionAdapter::PostgreSQL

Inherits:
Base
  • Object
show all
Includes:
ChannelPrefix
Defined in:
lib/action_cable/subscription_adapter/postgresql.rb

Overview

:nodoc:

Defined Under Namespace

Classes: Listener

Instance Attribute Summary

Attributes inherited from Base

#logger, #server

Instance Method Summary collapse

Methods inherited from Base

#identifier

Constructor Details

#initializePostgreSQL

Returns a new instance of PostgreSQL.



13
14
15
16
# File 'lib/action_cable/subscription_adapter/postgresql.rb', line 13

def initialize(*)
  super
  @listener = nil
end

Instance Method Details

#broadcast(channel, payload) ⇒ Object



18
19
20
21
22
# File 'lib/action_cable/subscription_adapter/postgresql.rb', line 18

def broadcast(channel, payload)
  with_broadcast_connection do |pg_conn|
    pg_conn.exec("NOTIFY #{pg_conn.escape_identifier(channel_identifier(channel))}, '#{pg_conn.escape_string(payload)}'")
  end
end

#shutdownObject



32
33
34
# File 'lib/action_cable/subscription_adapter/postgresql.rb', line 32

def shutdown
  listener.shutdown
end

#subscribe(channel, callback, success_callback = nil) ⇒ Object



24
25
26
# File 'lib/action_cable/subscription_adapter/postgresql.rb', line 24

def subscribe(channel, callback, success_callback = nil)
  listener.add_subscriber(channel_identifier(channel), callback, success_callback)
end

#unsubscribe(channel, callback) ⇒ Object



28
29
30
# File 'lib/action_cable/subscription_adapter/postgresql.rb', line 28

def unsubscribe(channel, callback)
  listener.remove_subscriber(channel_identifier(channel), callback)
end

#with_broadcast_connection(&block) ⇒ Object

:nodoc:



51
52
53
54
55
56
57
# File 'lib/action_cable/subscription_adapter/postgresql.rb', line 51

def with_broadcast_connection(&block) # :nodoc:
  ActiveRecord::Base.connection_pool.with_connection do |ar_conn|
    pg_conn = ar_conn.raw_connection
    verify!(pg_conn)
    yield pg_conn
  end
end

#with_subscriptions_connection(&block) ⇒ Object

:nodoc:



36
37
38
39
40
41
42
43
44
45
46
47
48
49
# File 'lib/action_cable/subscription_adapter/postgresql.rb', line 36

def with_subscriptions_connection(&block) # :nodoc:
  ar_conn = ActiveRecord::Base.connection_pool.checkout.tap do |conn|
    # Action Cable is taking ownership over this database connection, and
    # will perform the necessary cleanup tasks
    ActiveRecord::Base.connection_pool.remove(conn)
  end
  pg_conn = ar_conn.raw_connection

  verify!(pg_conn)
  pg_conn.exec("SET application_name = #{pg_conn.escape_identifier(identifier)}")
  yield pg_conn
ensure
  ar_conn.disconnect!
end