basic proxy and consumer done
This commit is contained in:
88
consumer/bettermail_consumer.rb
Normal file
88
consumer/bettermail_consumer.rb
Normal file
@@ -0,0 +1,88 @@
|
||||
# frozen_string_literal: true
|
||||
|
||||
require 'dotenv/load'
|
||||
require 'pg'
|
||||
require 'mail'
|
||||
require 'bunny'
|
||||
|
||||
# Server class
|
||||
class BettermailConsumer
|
||||
def initialize
|
||||
@pg = PG::Connection.new(ENV['DATABASE_URL'])
|
||||
end
|
||||
|
||||
def start
|
||||
begin
|
||||
puts ' [*] Waiting for messages. To exit press CTRL+C'
|
||||
mail_queue = bunny_channel.queue(ENV['RABBIT_MAIL_QUEUE'])
|
||||
mail_queue.subscribe(block: true) do |_delivery_info, _properties, body|
|
||||
mail = Mail.read_from_string(body)
|
||||
puts mail
|
||||
end
|
||||
sns_queue = bunny_channel.queue(ENV['RABBIT_SNS_QUEUE'])
|
||||
sns_queue.subscribe(block: true) do |_delivery_info, _properties, body|
|
||||
puts " [x] Received #{_delivery_info}"
|
||||
end
|
||||
rescue Interrupt => _
|
||||
stop
|
||||
end
|
||||
|
||||
end
|
||||
|
||||
def stop
|
||||
@bunny_conn&.close
|
||||
@pg&.close
|
||||
end
|
||||
|
||||
def bunny_channel
|
||||
if @bunny_channel.nil?
|
||||
@bunny_conn = Bunny.new
|
||||
@bunny_conn.start
|
||||
@bunny_channel = @bunny_conn.create_channel
|
||||
end
|
||||
@bunny_channel
|
||||
end
|
||||
end
|
||||
|
||||
service_name='BetterMail::Consumer'
|
||||
# Create a new server instance for listening at localhost interfaces 127.0.0.1:2525
|
||||
# and accepting a maximum of 4 simultaneous connections per default
|
||||
service = BettermailConsumer.new
|
||||
|
||||
# save flag for Ctrl-C pressed
|
||||
flag_status_ctrl_c_pressed = false
|
||||
|
||||
# try to gracefully shutdown on Ctrl-C
|
||||
trap('INT') do
|
||||
# print an empty line right after ^C
|
||||
puts
|
||||
# notify flag about Ctrl-C was pressed
|
||||
flag_status_ctrl_c_pressed = true
|
||||
# signal exit to app
|
||||
exit 0
|
||||
end
|
||||
|
||||
# Output for debug
|
||||
puts("Starting #{service_name}")
|
||||
|
||||
# setup exit code
|
||||
at_exit do
|
||||
# check to shutdown connection
|
||||
if service
|
||||
# Output for debug
|
||||
puts('Ctrl-C interrupted, exit now...') if flag_status_ctrl_c_pressed
|
||||
# info about shutdown
|
||||
puts("Shutdown #{service_name}...")
|
||||
# stop all threads and connections gracefully
|
||||
service.stop
|
||||
end
|
||||
|
||||
# Output for debug
|
||||
puts "#{service_name} stopped!"
|
||||
end
|
||||
|
||||
# Start the server
|
||||
service.start
|
||||
|
||||
# Run on server forever
|
||||
service.join
|
||||
Reference in New Issue
Block a user