Contents

# :nodoc:
require "rubygems"
require 'logger'
require "log4r/outputter/outputter"
require 'bunny'
require 'yaml'

module Log4r
  # See log4r/logserver.rb
  class RabbitOutputter < Outputter

    def initialize(_name, hash={})
      # Configuration defaults
      super(_name, hash)
      stderr_log "Unable to find rabbit configuration file" unless load_config
      @config ||= {:host => "localhost"}
      @config.symbolize_keys!
      @queue_name = @config.delete(:queue) || ''
      start_bunny rescue nil
    end

    def load_config_file(name)
      path = "#{Rails.root}/config/#{name}"
      if File.exist?(path)
        @config = YAML::load(IO.read(path)) 
      end
    end

    def load_config
      @config = if load_config_file("bunny.yml")
        @config[Rails.env]
      else
        load_config_file("rabbitmq.yml")
      end
    end

    def start_bunny
      begin
        stderr_log "Starting Bunny Client"
        config = @config.clone
        config[:pass] &&= "**redacted**"
        stderr_log config
        @conn = Bunny.new @config
        @conn.start
        create_channel
      rescue Bunny::TCPConnectionFailed => e
        stderr_log "rescued from: #{e}. Unable to connect to Rabbit Server"
      end
    end

    def stderr_log(msg)
      $stderr.puts "[#{Time.now.utc}] #{msg}"
    end

    def create_channel
      ch = @conn.create_channel
      @queue  = ch.queue(@queue_name, auto_delete: false, durable: true)
    end
    
    private

    def write(data)
      @queue.publish data, { routing_key: @queue.name } if @conn.connected? and @queue
    rescue Exception => e
      @conn.send(:handle_network_failure, e)
      create_channel if @conn.connected?
    end

  end
end

Version data entries

2 entries across 2 versions & 1 rubygems

Version	Path
mtn_log4r-1.1.12	lib/log4r/outputter/rabbitoutputter.rb
mtn_log4r-1.1.11	lib/log4r/outputter/rabbitoutputter.rb