rabbit-wq 0.0.1

data/.gitignore

@@ -0,0 +1,17 @@
+*.gem
+*.rbc
+.bundle
+.config
+.yardoc
+Gemfile.lock
+InstalledFiles
+_yardoc
+coverage
+doc/
+lib/bundler/man
+pkg
+rdoc
+spec/reports
+test/tmp
+test/version_tmp
+tmp

data/.ruby-gemset

@@ -0,0 +1 @@
+gemdev

data/.ruby-version

@@ -0,0 +1 @@
+1.9.3-p392

data/Gemfile

@@ -0,0 +1,4 @@
+source 'https://rubygems.org'
+
+# Specify your gem's dependencies in brer_rabbit.gemspec
+gemspec

data/LICENSE.txt

@@ -0,0 +1,22 @@
+Copyright (c) 2013 Jason Harrelson
+
+MIT License
+
+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,28 @@
+# RabbitWQ
+
+A work queue built on RabbitMQ and Celluloid.
+
+This is NOT production ready.  Released only to reserve gem name.
+
+
+## Installation
+
+Add this line to your application's Gemfile:
+
+    gem 'rabbit-wq'
+
+And then execute:
+
+    $ bundle
+
+Or install it yourself as:
+
+    $ gem install rabbit-wq
+
+
+## Usage
+
+TODO: Write usage instructions here
+
+
+## Contributing

data/Rakefile

@@ -0,0 +1 @@
+require "bundler/gem_tasks"

data/bin/rabbit-wq

@@ -0,0 +1,2 @@
+#!/usr/bin/env ruby
+require 'rabbit_wq/cli'

data/lib/rabbit-wq.rb

@@ -0,0 +1 @@
+require "rabbit_wq"

data/lib/rabbit_wq.rb

@@ -0,0 +1,34 @@
+require "rabbit_wq/version"
+
+module RabbitWQ
+
+  APP_ID                = "rabbit-wq"
+  APP_NAME              = "Rabbit Work Queue"
+  COMPANY               = "Look Forward Enterprises"
+  DELAY_QUEUE_PREFIX    = "work-delay" # TODO: Make this configurable (from ENV, or file?)
+  DELAY_EXCHANGE_PREFIX = "work-delay" # TODO: Make this configurable (from ENV, or file?)
+  INT                   = "INT"
+  QUEUE                 = "work" # TODO: Make this configurable (from ENV, or file?)
+  VERSION_COPYRIGHT     = "v#{VERSION} \u00A9#{Time.now.year} #{COMPANY}"
+  WORK_EXCHANGE         = "work" # TODO: Make this configurable (from ENV, or file?)
+
+  autoload :Configuration,  'rabbit_wq/configuration'
+  autoload :Logging,        'rabbit_wq/logging'
+  autoload :MessageHandler, 'rabbit_wq/message_handler'
+  autoload :Server,         'rabbit_wq/server'
+  autoload :ServerLogging,  'rabbit_wq/server_logging'
+  autoload :Work,           'rabbit_wq/work'
+
+  def self.configuration
+    @configuration ||= Configuration.new
+  end
+
+  def self.configure
+    yield( configuration ) if block_given?
+  end
+
+  class << self
+    attr_accessor :logger
+  end
+
+end

data/lib/rabbit_wq/cli.rb

@@ -0,0 +1,107 @@
+require 'rubygems'
+require 'rabbit_wq'
+require 'trollop'
+require 'yell'
+
+module RabbitWQ
+  module Cli
+
+    SUB_COMMANDS = %w(
+      restart
+      start
+      status
+      stop
+    )
+
+    def self.start( options )
+      if options[:interactive]
+        start_interactive options
+      else
+        start_daemon options
+      end
+    end
+
+    def self.start_interactive( options )
+      server = RabbitWQ::Server.new( options.merge( log: nil ) )
+      server.start
+    end
+
+    def self.start_daemon( options )
+      server = RabbitWQ::ServerDaemon.new( options )
+      server.start
+    end
+
+  end
+end
+
+DEFAULT_LOG_PATH = "/var/log/rabbit-wq/#{RabbitWQ::APP_ID}.log"
+DEFAULT_PID_PATH = "/var/run/rabbit-wq/#{RabbitWQ::APP_ID}.pid"
+
+global_opts = Trollop::options do
+  version RabbitWQ::VERSION_COPYRIGHT
+  banner <<-EOS
+#{RabbitWQ::APP_NAME} #{RabbitWQ::VERSION_COPYRIGHT}
+
+Usage:
+  #{RabbitWQ::APP_ID} [command] [options]
+
+  commands:
+#{RabbitWQ::Cli::SUB_COMMANDS.map { |cmd| "    #{cmd}" }.join( "\n" )}
+
+  (For help with a command: #{RabbitWQ::APP_ID} [command] -h)
+
+options:
+EOS
+  stop_on RabbitWQ::Cli::SUB_COMMANDS
+end
+
+# Get the sub-command and its options
+#
+cmd = ARGV.shift || ''
+cmd_opts = case( cmd )
+  #when "restart"
+    #Trollop::options do
+      #opt :pid, "The path for the PID file", :type => String, :default => DEFAULT_PID_PATH
+    #end
+  when "start"
+    Trollop::options do
+      opt :interactive, "Execute the server in interactive mode", :short => '-i'
+      opt :log_level, "The log level", :type => String, :default => 'info'
+      opt :log, "The path for the log file", :type => String, :short => '-l', :default => DEFAULT_LOG_PATH
+      opt :pid, "The path for the PID file", :type => String, :default => DEFAULT_PID_PATH
+    end
+  #when "status"
+    #Trollop::options do
+      #opt :pid, "The path for the PID file", :type => String, :default => DEFAULT_PID_PATH
+    #end
+  #when "stop"
+    #Trollop::options do
+      #opt :pid, "The path for the PID file", :type => String, :default => DEFAULT_PID_PATH
+    #end
+  else
+    Trollop::die "unknown command #{cmd.inspect}"
+  end
+
+if cmd == 'start'
+  unless cmd_opts[:interactive]
+    Trollop::die( :log, "is required when running as daemon" ) unless cmd_opts[:log]
+    Trollop::die( :pid, "is required when running as daemon" ) unless cmd_opts[:pid]
+  end
+end
+
+if %w(restart status stop).include?( cmd )
+  Trollop::die( :pid, "is required" ) unless cmd_opts[:pid]
+end
+
+# Execute the command
+#
+case cmd
+  when "restart"
+    RabbitWQ::ServerDaemon.new( cmd_opts ).restart
+  when "start"
+    RabbitWQ::Cli.start cmd_opts
+  when "status"
+    RabbitWQ::ServerDaemon.new( cmd_opts ).status
+  when "stop"
+    RabbitWQ::ServerDaemon.new( cmd_opts ).stop
+  end

data/lib/rabbit_wq/configuration.rb

@@ -0,0 +1,13 @@
+module RabbitWQ
+  class Configuration
+
+    #def authentication_token
+      #@authentication_token
+    #end
+
+    #def authentication_token=( authentication_token )
+      #@authentication_token = authentication_token
+    #end
+
+  end
+end

data/lib/rabbit_wq/logging.rb

@@ -0,0 +1,21 @@
+module RabbitWQ
+  module Logging
+
+    %w(
+      debug
+      error
+      fatal
+      info
+      warn
+    ).each do |level|
+
+      define_method level do |*messages|
+        messages.each do |message|
+          RabbitWQ.logger.send level, message
+        end
+      end
+
+    end
+
+  end
+end

data/lib/rabbit_wq/message_handler.rb

@@ -0,0 +1,29 @@
+require 'celluloid/autostart'
+
+module RabbitWQ
+  class MessageHandler
+
+    include Celluloid
+    include Logging
+
+    REQUEUE = true
+
+    def call( options )
+      channel       = options[:channel]
+      delivery_info = options[:delivery_info]
+      metadata      = options[:metadata]
+      payload       = options[:payload]
+
+      info "PAYLOAD ARRIVED #{payload}"
+      channel.ack delivery_info.delivery_tag
+    end
+
+  protected
+
+    def requeue( channel, delivery_info, e=nil )
+      info ANSI.yellow { 'REQUEUE ' + e.message }
+      channel.reject delivery_info.delivery_tag, REQUEUE
+    end
+
+  end
+end

data/lib/rabbit_wq/server.rb

@@ -0,0 +1,76 @@
+require 'bunny'
+
+module RabbitWQ
+  class Server
+
+    include ServerLogging
+    include Logging
+
+    attr_reader :message_consumer,
+                :options,
+                :work_exchange
+
+    def initialize( options )
+      @options = options
+      options[:pool_size] ||= 2 # TODO move to external configuration
+
+      configure_server
+    end
+
+    def start
+      log_startup
+      run
+
+      trap( INT ) { finalize; exit }
+      sleep
+    end
+
+  protected
+
+    def finalize
+      info "SHUTTING DOWN"
+      message_consumer.cancel
+      mq.close
+    end
+
+    def mq
+      @mq ||= ::Bunny.new.tap do |bunny|
+        bunny.start
+      end
+    end
+
+    def channel
+      @channel ||= mq.create_channel
+    end
+
+    def work_exchange
+      @work_exchange ||= channel.direct( WORK_EXCHANGE, durable: true )
+    end
+
+    def queue
+      @queue ||= channel.queue( QUEUE,
+                                durable: true ).
+                         bind( work_exchange )
+    end
+
+    def pool
+      @pool ||= MessageHandler.pool( size: options[:pool_size] )
+    end
+
+    def run
+      @message_consumer = queue.subscribe( manual_ack: true ) do |delivery_info, metadata, payload|
+        debug "LISTENER RECEIVED #{payload}"
+
+        pool.async.call( payload: payload,
+                         delivery_info: delivery_info,
+                         metadata: metadata,
+                         channel: channel )
+      end
+    end
+
+    def configure_server
+      initialize_loggers
+    end
+
+  end
+end

data/lib/rabbit_wq/server_logging.rb

@@ -0,0 +1,53 @@
+require 'yell'
+
+# Provides logging services for the base server.
+#
+module RabbitWQ
+  module ServerLogging
+
+  protected
+
+    def initialize_loggers
+      if options[:interactive] || options[:log].nil? || options[:log].empty?
+        RabbitWQ.logger = Yell.new do |l|
+          l.level = log_level
+          l.adapter $stdout, :level => [:debug, :info, :warn]
+          l.adapter $stderr, :level => [:error, :fatal]
+        end
+      else
+        RabbitWQ.logger = Yell.new do |l|
+          l.level = log_level
+          l.adapter :file, options[:log]
+        end
+
+        Celluloid.logger = Yell.new do |l|
+          l.level = :info
+          l.adapter :file, File.join( File.dirname( options[:log] ), "#{APP_ID}-celluloid.log" )
+        end
+      end
+    end
+
+    def log_startup
+      start_banner( options ).each do |line|
+        RabbitWQ.logger.info line
+      end
+    end
+
+    def start_banner( options )
+      [
+        "",
+        "***",
+        "* #{RabbitWQ::APP_NAME} started",
+        "*",
+        "* #{VERSION_COPYRIGHT}",
+        "***",
+        "",
+      ]
+    end
+
+    def log_level
+      options.fetch( :log_level, :info ).to_sym
+    end
+
+  end
+end

data/lib/rabbit_wq/version.rb

@@ -0,0 +1,3 @@
+module RabbitWQ
+  VERSION = "0.0.1"
+end

data/lib/rabbit_wq/work.rb

@@ -0,0 +1,30 @@
+require 'bunny'
+
+module RabbitWQ
+  module Work
+
+    def self.enqueue( worker, options={} )
+      options[:delay] = nil if options[:delay] && options[:delay] < 5000
+
+      mq = ::Bunny.new.tap { |bunny| bunny.start }
+      channel = mq.create_channel
+
+      if options[:delay]
+        delay_x = channel.direct( "#{DELAY_EXCHANGE_PREFIX}-#{options[:delay]}ms", durable: true )
+        work_x = channel.direct( WORK_EXCHANGE, durable: true )
+        queue = channel.queue( "#{DELAY_QUEUE_PREFIX}-#{options[:delay]}ms",
+                               durable: true,
+                               arguments: { "x-dead-letter-exchange" => work_x.name,
+                                            "x-message-ttl" => options[:delay] } ).
+                        bind( delay_x )
+
+        delay_x.publish( 'hello', durable: true )
+        return
+      end
+
+      work_q = channel.queue( QUEUE, durable: true )
+      work_q.publish( 'hello', durable: true )
+    end
+
+  end
+end

data/rabbit-wq.gemspec

@@ -0,0 +1,29 @@
+# coding: utf-8
+lib = File.expand_path('../lib', __FILE__)
+$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
+require 'rabbit_wq/version'
+
+Gem::Specification.new do |spec|
+  spec.name          = "rabbit-wq"
+  spec.version       = RabbitWQ::VERSION
+  spec.authors       = ["C. Jason Harrelson"]
+  spec.email         = ["jason@lookforwardenterprises.com"]
+  spec.description   = %q{A work queue built on RabbitMQ and Celluloid.  See README for more details.}
+  spec.summary       = %q{A work queue built on RabbitMQ and Celluloid.}
+  spec.homepage      = ""
+  spec.license       = "MIT"
+
+  spec.files         = `git ls-files`.split($/)
+  spec.executables   = ['rabbit-wq'] #spec.files.grep(%r{^bin/}) { |f| File.basename(f) }
+  spec.test_files    = spec.files.grep(%r{^(test|spec|features)/})
+  spec.require_paths = ["lib"]
+
+  spec.add_development_dependency "bundler", "~> 1.3"
+  spec.add_development_dependency "rake"
+
+  spec.add_dependency "bunny", "~> 1"
+  spec.add_dependency "celluloid", "~> 0"
+  spec.add_dependency "trollop", "~> 2"
+  spec.add_dependency "yell", "~> 1"
+
+end

metadata

@@ -0,0 +1,168 @@
+--- !ruby/object:Gem::Specification
+name: rabbit-wq
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+  prerelease: 
+platform: ruby
+authors:
+- C. Jason Harrelson
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2013-11-27 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: bundler
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '1.3'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '1.3'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: bunny
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '1'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '1'
+- !ruby/object:Gem::Dependency
+  name: celluloid
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: trollop
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '2'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '2'
+- !ruby/object:Gem::Dependency
+  name: yell
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '1'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '1'
+description: A work queue built on RabbitMQ and Celluloid.  See README for more details.
+email:
+- jason@lookforwardenterprises.com
+executables:
+- rabbit-wq
+extensions: []
+extra_rdoc_files: []
+files:
+- .gitignore
+- .ruby-gemset
+- .ruby-version
+- Gemfile
+- LICENSE.txt
+- README.md
+- Rakefile
+- bin/rabbit-wq
+- lib/rabbit-wq.rb
+- lib/rabbit_wq.rb
+- lib/rabbit_wq/cli.rb
+- lib/rabbit_wq/configuration.rb
+- lib/rabbit_wq/logging.rb
+- lib/rabbit_wq/message_handler.rb
+- lib/rabbit_wq/server.rb
+- lib/rabbit_wq/server_logging.rb
+- lib/rabbit_wq/version.rb
+- lib/rabbit_wq/work.rb
+- rabbit-wq.gemspec
+homepage: ''
+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'
+      segments:
+      - 0
+      hash: -1285656029023150188
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+      segments:
+      - 0
+      hash: -1285656029023150188
+requirements: []
+rubyforge_project: 
+rubygems_version: 1.8.25
+signing_key: 
+specification_version: 3
+summary: A work queue built on RabbitMQ and Celluloid.
+test_files: []