energizer_bunny 0.3.0-java

data/LICENSE.txt

@@ -0,0 +1,20 @@
+Copyright (c) 2012 Joel Friedman
+
+Permission is hereby granted, free of charge, to any person obtaining
+a copy of this software and associated documentation files (the
+"Software"), to deal in the Software without restriction, including
+without limitation the rights to use, copy, modify, merge, publish,
+distribute, sublicense, and/or sell copies of the Software, and to
+permit persons to whom the Software is furnished to do so, subject to
+the following conditions:
+
+The above copyright notice and this permission notice shall be
+included in all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
+LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
+OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
+WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

data/README.md

@@ -0,0 +1,27 @@
+Energizer Bunny
+===============
+
+A thin wrapper around [hot_bunnies][], the JRuby wrapper around the
+[RabbitMQ Java Client][api-guide].
+
+Usage
+-----
+
+TODO
+
+Contributing
+------------
+
+1. Fork it
+2. Create your feature branch (`git checkout -b my-new-feature`)
+3. Commit your changes (`git commit -am 'Add some feature'`)
+4. Push to the branch (`git push origin my-new-feature`)
+5. Create new Pull Request
+
+Copyright
+---------
+
+Copyright (c) 2012 Joel Friedman. See LICENSE.txt for further details.
+
+[hot_bunnies]: <https://github.com/ruby-amqp/hot_bunnies>
+[api-guide]: <http://www.rabbitmq.com/api-guide.html>

data/lib/energizer_bunny.rb

@@ -0,0 +1,2 @@
+require 'hot_bunnies'
+require 'energizer_bunny/connection'

data/lib/energizer_bunny/connection.rb

@@ -0,0 +1,93 @@
+require_relative 'subscription'
+
+module EnergizerBunny
+  class Connection
+    class ConfigurationError < StandardError ; end
+
+    def initialize config, logger = Rails.logger
+      @config = config
+      @logger = logger
+      return unless enabled?
+      at_exit { close }
+      @exchanges = Hash.new { |exchanges, topic_key| exchanges[topic_key] = create_exchange topic_key }
+      @queues    = Hash.new { |queues, topic_key|    queues[topic_key]    = create_queue topic_key }
+      @subscriptions = []
+      create_connection
+    end
+
+    def subscribe topic_key, subscription_opts = {}, &block
+      return unless enabled?
+      @subscriptions << Subscription.new(@queues[topic_key], subscription_opts, @logger, block)
+    end
+
+    def publish topic_key, message, opts = {}
+      add_message_id_to_header! opts
+      @exchanges[topic_key].publish message, opts if enabled?
+    end
+
+    def connected?
+      @connection && @connection.open?
+    end
+
+    def enabled?
+      @config[:enabled]
+    end
+
+    def connection_url
+      @config[:broker][:url]
+    end
+
+    private
+    def create_exchange topic_key
+      exchange_hash = @config[:exchanges][topic_key]
+      channel.exchange exchange_hash[:name], exchange_hash[:opts]
+    end
+
+    def create_connection
+      url = connection_url
+      raise ConfigurationError.new('Please set a broker url for RabbitMQ') if enabled? && url.nil?
+      @logger.info "Connecting bunny to: #{url}"
+      @connection = HotBunnies.connect :uri => url
+    end
+
+    #TODO: is having one channel bad, should this be one per exchange?
+    def channel
+      @channel ||= @connection.create_channel.tap { |c| c.prefetch = (@config.delete(:prefetch_count) || 1) }
+    end
+
+    def create_queue topic_key
+      queue_hash = @config[:queues][topic_key]
+      raise ConfigurationError.new("Cannot find queue configuration for queue: #{topic_key}") if queue_hash.nil?
+      queue_opts = queue_hash[:opts]
+      queue_opts[:durable] = true unless queue_opts.key? :durable
+      channel.queue(queue_hash[:name], queue_opts).tap do |queue|
+        queue_bind_opts = queue_hash[:bind_opts] || {}
+        Array(queue_hash[:bindings]).each do |binding|
+          binding_opts = queue_bind_opts.merge(binding[:binding_opts] || {})
+          queue.bind @exchanges[binding[:exchange_topic_key]], binding_opts
+        end
+      end
+    end
+
+    def close
+      @logger.info 'Closing rabbit connection'
+      @subscriptions.each &:close
+      return unless connected?
+      if @channel
+        @channel.close
+        @channel = nil
+      end
+      @connection.close
+      @connection = nil
+    end
+
+    def add_message_id_to_header!(opts)
+      opts[:headers] ||= {}
+      opts[:headers]['message_id'] ||= guid_generator.generate
+    end
+
+    def guid_generator
+      @guid_generator ||= UUID.new
+    end
+  end
+end

data/lib/energizer_bunny/subscription.rb

@@ -0,0 +1,52 @@
+module EnergizerBunny
+  class Subscription
+    import java.util.concurrent.Executors
+
+    def initialize queue, subscription_opts, logger, handle_message
+      @queue, @subscription_opts, @logger, @handle_message = queue, subscription_opts.dup, logger, handle_message
+      @reraise_errors = @subscription_opts.delete(:reraise_errors)
+      @subscription_opts[:executor] = executor unless @subscription_opts[:executor]
+      listen!
+    end
+
+    def close
+      close_executor
+      return if @subscription.nil?
+      @subscription.shutdown!
+      @subscription = nil
+    end
+
+    private
+    def listen!
+      @subscription = @queue.subscribe(@subscription_opts) do |headers, msg|
+        begin
+          @handle_message.call headers.properties.headers, msg
+          headers.ack
+        rescue => e
+          @logger.error "!!!Error handling message: #{e.message}"
+          headers.reject
+          raise if @reraise_errors
+        end
+      end
+      self
+    end
+
+    def log_error e
+      @logger.error "An error has occured on another thread: #{e}#{$/}#{e.backtrace}"
+    end
+
+    def thread_pool_size
+      @thread_pool_size ||= (@subscription_opts.delete(:thread_pool_size) || 1)
+    end
+
+    def executor
+      @executor ||= Executors.new_fixed_thread_pool(thread_pool_size)
+    end
+
+    def close_executor
+      return unless @executor
+      @executor.shutdown_now
+      @executor = nil
+    end
+  end
+end

data/lib/energizer_bunny/version.rb

@@ -0,0 +1,3 @@
+module EnergizerBunny
+  VERSION = '0.3.0'
+end

metadata

@@ -0,0 +1,101 @@
+--- !ruby/object:Gem::Specification
+name: energizer_bunny
+version: !ruby/object:Gem::Version
+  version: 0.3.0
+  prerelease: 
+platform: java
+authors:
+- Joel Friedman
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2013-05-14 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: hot_bunnies
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '1.4'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '1.4'
+- !ruby/object:Gem::Dependency
+  name: uuid
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '2.0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '2.0'
+- !ruby/object:Gem::Dependency
+  name: rspec
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '2.11'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '2.11'
+description: Extra hot_bunnies.
+email: asher.friedman@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files:
+- LICENSE.txt
+- README.md
+files:
+- lib/energizer_bunny/connection.rb
+- lib/energizer_bunny/subscription.rb
+- lib/energizer_bunny/version.rb
+- lib/energizer_bunny.rb
+- LICENSE.txt
+- README.md
+homepage: http://github.com/trunkclub/energizer_bunny
+licenses:
+- MIT
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 1.8.24
+signing_key: 
+specification_version: 3
+summary: Declaratively configure hot_bunnies.
+test_files: []