rabbithutch 0.1.2 → 0.1.4

data/lib/configurator.rb

@@ -1,38 +1,38 @@
-require 'yaml'
-    
-module RabbitHutch
-  class Configurator
-
-    attr_accessor :config
-
-    def initialize options
-      
-      file = ARGV[0] || (File.dirname(__FILE__) + '/../config.yaml')
-      
-      puts "Using config from #{file}"
-      
-      unless File.exists? file
-        raise "Configuration file [#{file}] doesn't exist"
-      end
-      @config = YAML::load(File.open(file))
-    end
-    
-    def application
-      @config['application']
-    end
-
-    def log_config
-      @config['log4r_config']
-    end
-    
-    def consumers
-      @config['consumers_config']["consumers"]
-    end
-        
-    def rabbitmq_hosts
-      @config['rabbitmq']['hosts']
-    end
-        
-  end
-end
-
+require 'yaml'
+    
+module RabbitHutch
+  class Configurator
+
+    attr_accessor :config
+
+    def initialize options
+      
+      file = options['config'] || (File.dirname(__FILE__) + '/../config.yaml')
+      
+      puts "Using config from #{file}"
+      
+      unless File.exists? file
+        raise "Configuration file [#{file}] doesn't exist"
+      end
+      @config = YAML::load(File.open(file))
+    end
+    
+    def application
+      @config['application']
+    end
+
+    def log_config
+      @config['log4r_config']
+    end
+    
+    def consumers
+      @config['consumers_config']["consumers"]
+    end
+        
+    def rabbitmq_hosts
+      @config['rabbitmq']['hosts']
+    end
+        
+  end
+end
+

data/lib/consumer.rb

@@ -1,32 +1,32 @@
-require "rubygems"
-require_relative "configurator"
-require_relative "consumers/mongo_consumer"
-
-module RabbitHutch
-  # Controls the consumers that are to be kicked off based on the config file
-  class Consumer
-    def initialize(consumers)
-      @consumers = consumers
-    end
-    
-    # Raised on receipt of a message from RabbitMq and uses the appropriate appender
-    def handle_message(metadata, payload)
-       # loop through appenders and fire each as required
-       @consumers.each do |consumer|
-         action = metadata.routing_key.split('.', 2).first
-         if(action == "publish")
-           exchange = metadata.attributes[:headers]["exchange_name"]
-           queue = metadata.routing_key.split('.', 2).last
-           item = {:date => Time.now, 
-                :exchange => exchange,
-                :queue => queue,
-                :routing_keys => metadata.attributes[:headers]["routing_keys"].inspect,
-                :attributes => metadata.attributes.inspect,
-                :payload => payload.inspect
-           }
-           consumer.log_event(item)
-         end 
-       end
-    end
-  end
+require "rubygems"
+require_relative "configurator"
+require_relative "consumers/mongo_consumer"
+
+module RabbitHutch
+  # Controls the consumers that are to be kicked off based on the config file
+  class Consumer
+    def initialize(consumers)
+      @consumers = consumers
+    end
+    
+    # Raised on receipt of a message from RabbitMq and uses the appropriate appender
+    def handle_message(metadata, payload)
+       # loop through appenders and fire each as required
+       @consumers.each do |consumer|
+         action = metadata.routing_key.split('.', 2).first
+         if(action == "publish")
+           exchange = metadata.attributes[:headers]["exchange_name"]
+           queue = metadata.routing_key.split('.', 2).last
+           item = {:date => Time.now, 
+                :exchange => exchange,
+                :queue => queue,
+                :routing_keys => metadata.attributes[:headers]["routing_keys"].inspect,
+                :attributes => metadata.attributes.inspect,
+                :payload => payload.inspect
+           }
+           consumer.log_event(item)
+         end 
+       end
+    end
+  end
 end

data/lib/consumers/console_consumer.rb

@@ -1,20 +1,20 @@
-# encoding: utf-8
-require 'rubygems'
-require "mongo"
-
-module RabbitHutch
-  class ConsoleConsumer
-      
-    def initialize()
-      puts "\tInitializing Console Consumer"
-    end
-        
-    def log_event(item)
-      begin  
-        puts item
-      rescue Exception => e
-        puts "Error occurred Message Handler trying to write messages to Log #{e.inspect}" 
-      end
-    end 
-  end
-end
+# encoding: utf-8
+require 'rubygems'
+require "mongo"
+
+module RabbitHutch
+  class ConsoleConsumer
+      
+    def initialize()
+      puts "\tInitializing Console Consumer"
+    end
+        
+    def log_event(item)
+      begin  
+        puts item
+      rescue Exception => e
+        puts "Error occurred Message Handler trying to write messages to Log #{e.inspect}" 
+      end
+    end 
+  end
+end

data/lib/consumers/log4r_consumer.rb

@@ -1,35 +1,35 @@
-# encoding: utf-8
-require 'rubygems'
-require 'log4r'
-#require 'log4r/outputter/syslogoutputter'
-
-module RabbitHutch
-  class Log4rConsumer
-      
-    def initialize(rabbitmq_host, config)
-      puts "\tInitializing Log4r Consumer"
-      @rabbitmq_host = rabbitmq_host
-      @config = config
-      @log_name = rabbitmq_host["displayname"]
-      @config.consumers.each do |consumer|
-        if consumer["name"] == 'log4r_consumer'
-          @log_prefix = consumer['log_prefix']
-          @log_location = consumer['log_location']
-        end
-      end
-      
-      
-      @logger = Log4r::Logger.new("#{@log_name}#_log")
-      @logger.outputters << Log4r::FileOutputter.new("#{@log_name}_filelog", :filename =>  "#{@log_location}/#{@log_prefix}#{@log_name}.log")
-    end
-        
-    def log_event(item)
-      begin  
-        @logger.info(item)    
-      rescue Exception => e
-        puts "Error occurred Message Handler trying to write messages to Log #{e.inspect}" 
-        #@log.error("Error occurred Message Handler trying to write messages to MONGODB #{e.inspect}") 
-      end
-    end 
-  end
-end
+# encoding: utf-8
+require 'rubygems'
+require 'log4r'
+#require 'log4r/outputter/syslogoutputter'
+
+module RabbitHutch
+  class Log4rConsumer
+      
+    def initialize(rabbitmq_host, config)
+      puts "\tInitializing Log4r Consumer"
+      @rabbitmq_host = rabbitmq_host
+      @config = config
+      @log_name = rabbitmq_host["displayname"]
+      @config.consumers.each do |consumer|
+        if consumer["name"] == 'log4r_consumer'
+          @log_prefix = consumer['log_prefix']
+          @log_location = consumer['log_location']
+        end
+      end
+      
+      
+      @logger = Log4r::Logger.new("#{@log_name}#_log")
+      @logger.outputters << Log4r::FileOutputter.new("#{@log_name}_filelog", :filename =>  "#{@log_location}/#{@log_prefix}#{@log_name}.log")
+    end
+        
+    def log_event(item)
+      begin  
+        @logger.info(item)    
+      rescue Exception => e
+        puts "Error occurred Message Handler trying to write messages to Log #{e.inspect}" 
+        #@log.error("Error occurred Message Handler trying to write messages to MONGODB #{e.inspect}") 
+      end
+    end 
+  end
+end

data/lib/consumers/mongo_consumer.rb

@@ -1,34 +1,34 @@
-# encoding: utf-8
-require 'rubygems'
-require "mongo"
-
-module RabbitHutch
-  class MongoConsumer
-      
-    def initialize(rabbitmq_host, config)
-      puts "\tInitializing MongoDb Consumer"
-      @config = config
-      @rabbitmq_host = rabbitmq_host
-      
-      @config.consumers.each do |consumer|
-        if consumer["name"] == 'mongo_consumer'
-          @host = consumer['hostname']
-          @port = consumer["port"]
-          @database_prefix = consumer['database_prefix']
-          @database = "#{@database_prefix}#{rabbitmq_host["displayname"]}"
-        end
-      end
-      @connection = Mongo::Connection.new(@host, @port)
-    end
-        
-    def log_event(item)
-      begin  
-        db = @connection.db(@database)
-        coll = db.collection(item[:exchange])
-        coll.insert(item)
-      rescue Exception => e
-        puts "Error occurred Message Handler trying to write messages to MONGODB #{e.inspect}" 
-      end
-    end 
-  end
-end
+# encoding: utf-8
+require 'rubygems'
+require "mongo"
+
+module RabbitHutch
+  class MongoConsumer
+      
+    def initialize(rabbitmq_host, config)
+      puts "\tInitializing MongoDb Consumer"
+      @config = config
+      @rabbitmq_host = rabbitmq_host
+      
+      @config.consumers.each do |consumer|
+        if consumer["name"] == 'mongo_consumer'
+          @host = consumer['hostname']
+          @port = consumer["port"]
+          @database_prefix = consumer['database_prefix']
+          @database = "#{@database_prefix}#{rabbitmq_host["displayname"]}"
+        end
+      end
+      @connection = Mongo::Connection.new(@host, @port)
+    end
+        
+    def log_event(item)
+      begin  
+        db = @connection.db(@database)
+        coll = db.collection(item[:exchange])
+        coll.insert(item)
+      rescue Exception => e
+        puts "Error occurred Message Handler trying to write messages to MONGODB #{e.inspect}" 
+      end
+    end 
+  end
+end

data/lib/logger.rb

@@ -1,16 +1,16 @@
-require 'log4r'
-require 'log4r/yamlconfigurator'
-require 'log4r/outputter/syslogoutputter'
-
-  class Logger
-  	@@log = nil
-  	def self.init(config)
-  		if !@@log.nil? 
-  			return @@log
-  		end
-  
-  		configurator = Log4r::YamlConfigurator 
-  		configurator.decode_yaml config.log_config
-  		@@log = Log4r::Logger['main']
-  	end
-  end
+require 'log4r'
+require 'log4r/yamlconfigurator'
+require 'log4r/outputter/syslogoutputter'
+
+  class Logger
+  	@@log = nil
+  	def self.init(config)
+  		if !@@log.nil? 
+  			return @@log
+  		end
+  
+  		configurator = Log4r::YamlConfigurator 
+  		configurator.decode_yaml config.log_config
+  		@@log = Log4r::Logger['main']
+  	end
+  end

data/lib/rabbithutch.rb

@@ -1,78 +1,78 @@
-require 'rubygems'
-require "amqp"
-require_relative "configurator"
-require_relative "consumers/mongo_consumer"
-require_relative "consumers/log4r_consumer"
-require_relative "consumers/console_consumer"
-require_relative "worker"
-
-  @config = RabbitHutch::Configurator.new({})
-  
-  puts "\tEnvironment Settings"
-  @config.rabbitmq_hosts.each do |rabbitmq_host|
-    puts "\tDisplay Name: #{rabbitmq_host["displayname"]}, host: #{rabbitmq_host["hostname"]}, username: #{rabbitmq_host["username"]}, enabled #{rabbitmq_host["enabled"]}"
-  end
-  
-  # Initialize all enabled consumners
-  # NOTE: this method will be replaced with a routine to reflect through all valid consumers and initialze them implicitly   
-  def initialize_consumers(rabbitmq_host)
-    puts "Initializing Consumers for #{rabbitmq_host["displayname"]}"
-    consumers = []
-    @config.consumers.each do |consumer|
-      if consumer["enabled"] == true
-        case consumer["name"]
-          when "console_consumer"
-            consumers << RabbitHutch::ConsoleConsumer.new()
-          when "mongo_consumer"
-            consumers << RabbitHutch::MongoConsumer.new(rabbitmq_host, @config)
-          when "log4r_consumer"
-            consumers << RabbitHutch::Log4rConsumer.new(rabbitmq_host, @config)
-          end 
-        end
-    end
-    puts "\tdone"
-    consumers 
-  end
-   
-  # Start the worker process to listen to a RabbitMq Node 
-  def start_worker(rabbitmq_host)
-    displayname = rabbitmq_host["displayname"]
-    hostname = rabbitmq_host["hostname"]
-    username = rabbitmq_host["username"]
-    password = rabbitmq_host["password"]
-    
-    consumers = initialize_consumers(rabbitmq_host)
-       
-    puts "Listening to RabbitMq #{displayname}, host: #{hostname}, username #{username}"
-    AMQP.connect(:host => hostname, :user => username, :password => password) do |connection|
-      channel = AMQP::Channel.new(connection)
-      worker = RabbitHutch::Worker.new(channel, @config, consumers)
-      worker.start
-    end
-  end
-  
-  # Entry Point to the application, Begins queue listener and initializes all consmers
-  def start
-    begin
-      EventMachine.run do
-        
-        @config.rabbitmq_hosts.each do |rabbitmq_host|
-          if rabbitmq_host["enabled"] == true
-            start_worker(rabbitmq_host)
-          end
-        end
-        Signal.trap("INT"){EventMachine.stop}
-        Signal.trap("QUIT"){EventMachine.stop}
-        Signal.trap("TERM"){EventMachine.stop}
-        Signal.trap("TSTP"){EventMachine.stop}
-      end
-    rescue Exception=>e
-      puts "#{e.message} #{e.backtrace}"
-    end
-  end
-
-# Kick off the App
-start
-
-
-
+require 'rubygems'
+require "amqp"
+require_relative "configurator"
+require_relative "consumers/mongo_consumer"
+require_relative "consumers/log4r_consumer"
+require_relative "consumers/console_consumer"
+require_relative "worker"
+
+  @config = RabbitHutch::Configurator.new({})
+  
+  puts "\tEnvironment Settings"
+  @config.rabbitmq_hosts.each do |rabbitmq_host|
+    puts "\tDisplay Name: #{rabbitmq_host["displayname"]}, host: #{rabbitmq_host["hostname"]}, username: #{rabbitmq_host["username"]}, enabled #{rabbitmq_host["enabled"]}"
+  end
+  
+  # Initialize all enabled consumners
+  # NOTE: this method will be replaced with a routine to reflect through all valid consumers and initialze them implicitly   
+  def initialize_consumers(rabbitmq_host)
+    puts "Initializing Consumers for #{rabbitmq_host["displayname"]}"
+    consumers = []
+    @config.consumers.each do |consumer|
+      if consumer["enabled"] == true
+        case consumer["name"]
+          when "console_consumer"
+            consumers << RabbitHutch::ConsoleConsumer.new()
+          when "mongo_consumer"
+            consumers << RabbitHutch::MongoConsumer.new(rabbitmq_host, @config)
+          when "log4r_consumer"
+            consumers << RabbitHutch::Log4rConsumer.new(rabbitmq_host, @config)
+          end 
+        end
+    end
+    puts "\tdone"
+    consumers 
+  end
+   
+  # Start the worker process to listen to a RabbitMq Node 
+  def start_worker(rabbitmq_host)
+    displayname = rabbitmq_host["displayname"]
+    hostname = rabbitmq_host["hostname"]
+    username = rabbitmq_host["username"]
+    password = rabbitmq_host["password"]
+    
+    consumers = initialize_consumers(rabbitmq_host)
+       
+    puts "Listening to RabbitMq #{displayname}, host: #{hostname}, username #{username}"
+    AMQP.connect(:host => hostname, :user => username, :password => password) do |connection|
+      channel = AMQP::Channel.new(connection)
+      worker = RabbitHutch::Worker.new(channel, @config, consumers)
+      worker.start
+    end
+  end
+  
+  # Entry Point to the application, Begins queue listener and initializes all consmers
+  def start
+    begin
+      EventMachine.run do
+        
+        @config.rabbitmq_hosts.each do |rabbitmq_host|
+          if rabbitmq_host["enabled"] == true
+            start_worker(rabbitmq_host)
+          end
+        end
+        Signal.trap("INT"){EventMachine.stop}
+        Signal.trap("QUIT"){EventMachine.stop}
+        Signal.trap("TERM"){EventMachine.stop}
+        Signal.trap("TSTP"){EventMachine.stop}
+      end
+    rescue Exception=>e
+      puts "#{e.message} #{e.backtrace}"
+    end
+  end
+
+# Kick off the App
+start
+
+
+