kennethkalmer/daemon-kit

require 'yaml'
require 'amqp'

module DaemonKit
  # Thin wrapper around the amqp gem, specifically designed to ease
  # configuration of a AMQP consumer daemon and provide some added
  # simplicity
  class AMQP

    @@instance = nil

    class << self

      def instance
        @instance ||= new
      end

      private :new

      def run(&block)
        instance.run(&block)
      end
    end

    def initialize( config = {} )
      @config = DaemonKit::Config.load('amqp').to_h( true )
    end

    def run(&block)
      # Start our event loop and AMQP client
      DaemonKit.logger.debug("AMQP.start(#{@config.inspect})")
      ::AMQP.start(@config) do |connection|
        # Ensure graceful shutdown of the connection to the broker
        hook = Proc.new { connection.close { EventMachine.stop } }
        DaemonKit.trap('INT', hook)
        DaemonKit.trap('TERM', hook)

        yield connection
      end
    end
  end
end