romq 0.1.0

data/Gemfile

@@ -0,0 +1,10 @@
+ruby '1.9.3'
+
+source 'https://rubygems.org'
+
+gemspec
+
+group :development do
+  gem 'awesome_print', '~> 1.0.2'
+  gem 'pry', '~> 0.9.10'
+end

data/README.md

@@ -0,0 +1,32 @@
+# RoMQ
+
+Stands for **Robust MQ** and makes your RabbitMQ clients handle the most
+common exceptions gracefully. It's cluster-aware, so if you define a
+list of rabbitmq hosts, it will keep re-trying to 
+connect by picking hosts at random.
+
+## Installation
+
+Add this line to your application's Gemfile:
+
+    gem 'romq'
+
+And then execute:
+
+    $ bundle
+
+Or install it yourself as:
+
+    $ gem install romq
+
+## Usage
+
+See the examples directory.
+
+## 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

data/examples/client.rb

@@ -0,0 +1,18 @@
+#!/usr/bin/env ruby
+
+require 'ap'
+require_relative '../lib/romq'
+
+RoMQ::Connection.new do |channel|
+  exchange = channel.direct("romq")
+
+  EM.add_periodic_timer(2) do
+    exchange.publish("Time is: #{Time.now}")
+  end
+
+  channel.queue("client.rb", exclusive: true).
+    bind(exchange).
+    subscribe do |metadata, payload|
+      ap payload
+    end
+end

data/lib/romq.rb

@@ -0,0 +1,5 @@
+require 'romq/version'
+require 'romq/connection'
+
+module Romq
+end

data/lib/romq/config.rb

@@ -0,0 +1,30 @@
+module RoMQ
+  module Config
+    extend self
+
+    LOG_EVERY = ENV.fetch('LOG_EVERY') { 2 }.to_i
+
+    RABBITMQ_CLUSTER = ENV.fetch('RABBITMQ_CLUSTER') {
+      "localhost"
+    }.split(",")
+
+    RABBITMQ_PORT         = ENV.fetch('RABBITMQ_PORT') { 5672 }
+    RABBITMQ_TIMEOUT      = ENV.fetch('RABBITMQ_TIMEOUT') { 1 }.to_i
+    RABBITMQ_USER         = ENV.fetch('RABBITMQ_USER') { "guest" }
+    RABBITMQ_PASSWORD     = ENV.fetch('RABBITMQ_PASSWORD') { "guest" }
+    RABBITMQ_VHOST        = ENV.fetch('RABBITMQ_VHOST') { "/" }
+    RABBITMQ_RECONNECT_IN = ENV.fetch('RABBITMQ_RECONNECT_IN') { 5 }.to_i
+    RABBITMQ_PREFECTH     = ENV.fetch('RABBITMQ_PREFECTH') { 100 }.to_i
+
+    def rabbitmq_connection
+      {
+        :host     => RABBITMQ_CLUSTER.sample,
+        :port     => RABBITMQ_PORT,
+        :vhost    => RABBITMQ_VHOST,
+        :user     => RABBITMQ_USER,
+        :password => RABBITMQ_PASSWORD,
+        :timeout  => RABBITMQ_TIMEOUT
+      }
+    end
+  end
+end

data/lib/romq/connection.rb

@@ -0,0 +1,75 @@
+require 'amqp'
+
+require_relative 'config'
+require_relative 'helpers'
+
+module RoMQ
+  class Connection
+    include Helpers
+
+    def initialize(&block)
+      stop_gracefully
+
+      EM.run do
+        flush_logs_periodically.()
+
+        connection = Config.rabbitmq_connection
+        @amqp_connection = AMQP.connect(connection)
+
+        @amqp_connection.on_error do |channel, connection_close|
+          raise channel_close.reply_text
+        end
+
+        @amqp_connection.on_open do
+          logger.info("[RoMQ] connected to #{connection[:host]}")
+          @amqp_connection.on_tcp_connection_loss do
+            reconnect.(block)
+          end
+
+          channel  = AMQP::Channel.new(@amqp_connection, auto_recovery: true)
+          channel.on_error(&handle_channel_error)
+          channel.prefetch(Config::RABBITMQ_PREFECTH)
+
+          yield(channel)
+        end
+      end
+    rescue AMQP::TCPConnectionFailed
+      reconnect.(block)
+    rescue AMQP::PossibleAuthenticationFailureError
+      logger.info("[RoMQ] could not authenticate, check your credentials: #{@amqp_connection.settings}")
+      stop.()
+    end
+
+    def reconnect
+      Proc.new do |block|
+        logger.info("[RoMQ] will try to re-connect in #{Config::RABBITMQ_RECONNECT_IN} seconds...")
+        EM.stop if EM.reactor_running?
+        sleep Config::RABBITMQ_RECONNECT_IN
+        Connection.new(&block)
+      end
+    end
+
+    def stop
+      Proc.new do
+        if @amqp_connection and @amqp_connection.connected?
+          @amqp_connection.disconnect do
+            logger.info("[RoMQ] closed connection to #{@amqp_connection.settings[:host]}")
+            super
+          end
+        else
+          super
+        end
+      end
+    end
+
+    def handle_channel_error
+      Proc.new do |channel, channel_close|
+        logger.error("[RoMQ] channel-level exception")
+        logger.error("[RoMQ] AMQP class id : #{channel_close.class_id}")
+        logger.error("[RoMQ] AMQP method id: #{channel_close.method_id}")
+        logger.error("[RoMQ] Status code   : #{channel_close.reply_code}")
+        logger.error("[RoMQ] Error message : #{channel_close.reply_text}")
+      end
+    end
+  end
+end

data/lib/romq/helpers.rb

@@ -0,0 +1,44 @@
+require_relative 'config'
+require 'logger'
+
+module RoMQ
+  module Helpers
+    DEBUG = ENV.fetch('DEBUG') { false }
+
+    def flush_logs_periodically
+      Proc.new do |seconds = Config::LOG_EVERY|
+        EM.add_periodic_timer(seconds) do
+          EM.defer { $stdout.flush }
+        end
+      end
+    end
+
+    def stop
+      if @connections_to_stop <= 0
+        EM.stop if EM.reactor_running?
+        logger.info("Stopping #{self.class} service... done!")
+        exit 0
+      else
+        logger.info("Waiting on #{@connections_to_stop} connection(s) to close...")
+      end
+    end
+
+    # If you need to tear down any other connections, keep track of
+    # their status via @connections_to_stop
+    def stop_gracefully
+      @connections_to_stop = 0
+
+      Signal.trap "SIGTERM", stop
+      Signal.trap "SIGINT", stop
+      Signal.trap "SIGQUIT", stop
+    end
+
+    def logger
+      return @logger if @logger
+
+      @logger = Logger.new(STDOUT)
+      @logger.level = DEBUG ? Logger::DEBUG : Logger::INFO
+      @logger
+    end
+  end
+end

data/lib/romq/version.rb

@@ -0,0 +1,3 @@
+module RoMQ
+  VERSION = "0.1.0"
+end

metadata

@@ -0,0 +1,69 @@
+--- !ruby/object:Gem::Specification
+name: romq
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+  prerelease: 
+platform: ruby
+authors:
+- Gerhard Lazu
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2012-09-14 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: amqp
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 0.9.7
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 0.9.7
+description: Gracefully handle common amqp exceptions for more robust clients.
+email:
+- gerhard@lazu.co.uk
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- lib/romq/config.rb
+- lib/romq/connection.rb
+- lib/romq/helpers.rb
+- lib/romq/version.rb
+- lib/romq.rb
+- examples/client.rb
+- Gemfile
+- README.md
+homepage: https://github.com/gosquared/romq
+licenses: []
+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.23
+signing_key: 
+specification_version: 3
+summary: Cluster-aware amqp client library, never gives up on reconnecting.
+test_files: []