dafiti-rabbit-hutch 0.1.5 → 0.1.6

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 16043f53d44831db55c4fbcdb307cbe1fa70f1130c628f3d2691e4f4df06c257
-  data.tar.gz: d7ce49872570a3b4dd9e3a74dda66c481149af8d585650e4d8746208d48bf878
+  metadata.gz: 9b744707fbf0a08c3516948d2755b8d73c9bf5677f93c8c519489f4265c327dc
+  data.tar.gz: d943b769c98a9bc4c9d0c4d70e589037f56dc9d25dba1c29da7c4acfe610dd35
 SHA512:
-  metadata.gz: c2ec20711106a19d2e17ef3317d1c1de45ed67edc5541ab382c2aaa58476ec30c248d9c6c27621f054321c026f5b76ebe9126641e2acd39826094e26e81aca0c
-  data.tar.gz: 72dbb1286e3d8421ce853a06231e04f4d9e318f119c59d0d1f8ea0064c194f6d49219e21866bc7db24e3f826513c1a594d1aef7a275057153dc7955ef23a223f
+  metadata.gz: 9e08d0c24abae122426dfb7a7c302927e8196684039e33f741442dd43d4cb588d7f367feb41be97a37cdb2e8ad226337f11a50c3e444610ff9148bfe6ee94363
+  data.tar.gz: 5cfc3f9e1f49c9451ece540bf088e4b77fcdb055282f86bbb363db1d1b11345e854184ad8f5d903d76ccdf1264463fa99afaf2b73b835db29d68f02555413f14

data/web/rabbithutchweb.rb

@@ -0,0 +1,106 @@
+require 'sinatra'
+require 'haml'
+require 'yaml'
+
+module RabbitHutchWeb
+  class RabbitSettings
+    
+    attr_accessor :settings, :application, :rabbitmq_hosts, :consumers
+    
+    def initialize
+      file = File.dirname(__FILE__) + '/../config.yaml'
+      @settings = YAML::load(File.open(file))
+      @application = settings['application']
+      @rabbitmq_hosts = settings['rabbitmq']['hosts']
+      @consumers = settings['consumers_config']['consumers']
+    end
+    
+    def save
+      file = File.dirname(__FILE__) + '/../config.yaml'
+      File.open(file, "w") do|f|
+        f.write self.settings.to_yaml
+      end
+    end
+    
+    def parse(data)
+      @settings = YAML::load(data)
+      @application = settings['application']
+      @rabbitmq_hosts = settings['rabbitmq']['hosts']
+      @consumers = settings['consumers_config']['consumers']
+    end
+    
+  end
+end
+
+get '/' do
+  settings = RabbitHutchWeb::RabbitSettings.new
+  
+  haml :index, :locals => { 
+    :app => settings.application,
+    :rabbit => settings.rabbitmq_hosts,
+    :consumers => settings.consumers
+  }
+end
+
+get '/rabbitmqsettings/add' do
+  settings = RabbitHutchWeb::RabbitSettings.new
+  params["id"] = settings.rabbitmq_hosts.length + 1
+  haml :rabbitmqsettings, :locals=>{:settings=>params} 
+end
+
+get '/rabbitmqsettings/edit/:id' do |id| 
+  setting = nil
+  settings = RabbitHutchWeb::RabbitSettings.new
+  settings.rabbitmq_hosts.each{|item| setting = item if item["id"]==id.to_i }
+  haml :rabbitmqsettings, :locals=>{:settings=>setting} 
+end
+
+post '/rabbitmqsettings/save' do
+  exists = false
+  settings = RabbitHutchWeb::RabbitSettings.new
+  settings.rabbitmq_hosts.each{|item|
+    if item["id"]==params["id"].to_i 
+      item["displayname"] = params["displayname"]
+      item["enabled"] = params["enabled"]
+      item["hostname"] = params["hostname"]
+      item["username"] = params["username"]
+      item["password"] = params["password"]
+      exists = true
+    end
+  }
+  
+  unless exists
+    setting = 
+    {
+    "id" => params["id"],
+    "displayname" => params["displayname"],
+    "enabled" => params["enabled"],
+    "hostname" => params["hostname"],
+    "username" => params["username"],
+    "password" => params["password"]
+    }
+    settings.rabbitmq_hosts << setting
+  end
+    
+  settings.save
+  redirect "/"
+end
+
+post '/rabbitmqsettings/delete' do
+  redirect "/"
+end
+
+get '/manageconfig' do
+  haml :manageconfig, :locals=>{:data=>params}
+end
+
+post '/manageconfig' do
+  haml :manageconfig, :locals=>{:data=>params}
+end
+
+post '/manageconfig/save' do
+  settings = RabbitHutchWeb::RabbitSettings.new
+  settings.parse(params["filetext"])
+  settings.save
+  redirect "/"
+end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: dafiti-rabbit-hutch
 version: !ruby/object:Gem::Version
-  version: 0.1.5
+  version: 0.1.6
 platform: ruby
 authors:
 - Luiz Bartolomeu
@@ -248,16 +248,7 @@ files:
 - README.md
 - bin/rabbithutch
 - bin/rabbithutchweb
-- lib/configurator.rb
-- lib/consumer.rb
-- lib/consumers/console_consumer.rb
-- lib/consumers/log4r_consumer.rb
-- lib/consumers/mongo_consumer.rb
-- lib/logger.rb
-- lib/rabbithutch.rb
-- lib/rabbithutchmgr.rb
-- lib/rabbithutchservice.rb
-- lib/worker.rb
+- web/rabbithutchweb.rb
 homepage: https://rubygems.org/gems/example
 licenses:
 - MIT

data/lib/configurator.rb

@@ -1,38 +0,0 @@
-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 +0,0 @@
-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,21 +0,0 @@
-# encoding: utf-8
-require 'rubygems'
-require "mongo"
-
-module RabbitHutch
-  class ConsoleConsumer
-      
-    def initialize()
-      puts "\tInitializing Console Consumer"
-    end
-        
-    def log_event(item)
-      begin  
-        puts "console"
-        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,36 +0,0 @@
-# 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  
-        puts "log_consumer"
-        @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,35 +0,0 @@
-# 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  
-        puts "mongo"
-        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 +0,0 @@
-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 +0,0 @@
-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
-
-
-

data/lib/rabbithutchmgr.rb

@@ -1,48 +0,0 @@
-require 'thor'
-require 'mustache'
-require_relative 'configurator.rb'
-
-module RabbitHutch
-  #This class controls the Command Line Interface
-  class CLI < Thor
-    
-StandardTemplate =<<-TEMPLATE
-{{#config}}
-
-{{#application}}
-Application
-+------------------------+------------------------+
-|Exchange                |Queue                   |
-+------------------------+------------------------+
-|{{exchangename}}        |  {{queuename}}         |
-+------------------------+------------------------+
-{{/application}}
-
-{{#rabbitmq}}
-RabbitMq Servers
-+------------------------+------------------------+
-|Display Name                |Host                   |
-+------------------------+------------------------+
-{{#hosts}}
-|{{displayname}}        |  {{hostname}}         |
-+------------------------+------------------------+
-{{/hosts}}
-{{/rabbitmq}}
-
-{{/config}}
-TEMPLATE
-
-    desc "list", "List all config settings"
-    #method_option :errors, :aliases => "-e", :type => :boolean, :default => false, :desc => "true = display files that could not be processed, false = do not display skipped files"
-    #method_option :hashtype, :aliases => "-h", :default => "cmd5", :desc => "Choose the hash algorithm to use - md5 or sha"
-    method_option :config, :aliases => "-c", :default => nil, :desc => "parse the config file from a specified location "
-    #method_option :recursive, :aliases => "-r", :type => :boolean, :default => "false", :desc => "true = recurse through sub directories, false = only do top directory"
-    def list()
-      @config = RabbitHutch::Configurator.new(options)
-      puts Mustache.render(StandardTemplate,  :config => @config);
-    end
-    
-  end  
-  
-  CLI.start()
-end

data/lib/rabbithutchservice.rb

@@ -1,18 +0,0 @@
-require 'rubygems'
-require 'daemons'
-
-# The Service controller.
-def start_service
-  begin
-    puts "-------------------------"
-    puts "Starting RabbitHutch #1"
-    Daemons.run(File.dirname(__FILE__) + '/rabbithutch.rb')
-  rescue SystemExit=>e
-    puts e.inspect
-  rescue Exception=>e
-    puts e.inspect
-  end
-end
-
-start_service()
-

data/lib/worker.rb

@@ -1,33 +0,0 @@
-require "rubygems"
-require "amqp"
-require_relative "configurator"
-require_relative "consumer"
-
-module RabbitHutch
-  @exchange_name = "amq.rabbitmq.trace"
-  
-  class Worker
-    
-    def initialize(channel, config, consumers)
-      @channel = channel
-      @channel.on_error(&method(:handle_channel_exception))
-      @consumer = Consumer.new(consumers)
-      @exchange_name = config.application['exchangename']
-      @queue_name = config.application['queuename']
-    end
-  
-    # begin listening for all topics in publish.#
-    def start
-      @exchange = @channel.topic(@exchange_name, :durable => true, :auto_delete => false, :internal => true)
-      @queue = @channel.queue(@queue_name, :durable => true, :auto_delete => false)
-      @queue.bind(@exchange, :routing_key => 'publish.#')
-      @queue.subscribe(&@consumer.method(:handle_message))
-    end
-  
-    def handle_channel_exception(channel, channel_close)
-      puts "Oops... a channel-level exception: code = #{channel_close.reply_code}, message = #{channel_close.reply_text}"
-    end
-    
-  end
-  
-end