Skip to content

Instantly share code, notes, and snippets.

@lastk
Forked from chsh/pg_pub_sub.rb
Created August 20, 2020 20:01
Show Gist options
  • Save lastk/2dfcab4802adfd2233556147ad9306dd to your computer and use it in GitHub Desktop.
Save lastk/2dfcab4802adfd2233556147ad9306dd to your computer and use it in GitHub Desktop.
PostgreSQL LISTEN/NOTIFY example for ruby
#
# A:
# pubsub = PgPubSub.new('channelname')
# pubsub.subscribe do |data|
# puts "data: #{data} is coming!"
# end
#
# B:
# pubsub = PgPubSub.new('channelname')
# pubsub.publish("hello world")
#
# becomes
# A:
# => data: hello world is coming!
#
#
# "C" before channel name is not required.
# But when channel name starts with digit, e.g. LISTEN 1NAME
# it causes syntax error. So, it fix erorr.
#
class PgPubSub
def initialize(channel)
@channel = channel
end
def subscribe
connection.execute "LISTEN C#{@channel}"
loop do
connection.raw_connection.wait_for_notify do |event, id, data|
yield data
end
end
ensure
connection.execute "UNLISTEN C#{@channel}"
end
def publish(json)
connection.execute "NOTIFY C#{@channel}, #{connection.quote json}"
end
private
def connection
ActiveRecord::Base.connection
end
end
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment