multiple_man 1.0.0 → 1.1.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: eff9e5989a0fe4702caf87df4d8a50b0fb5cea03
-  data.tar.gz: 04361d38e491938b791dfb98e16727240ee33daa
+  metadata.gz: 620155bba7ed083d04162d64106eabcaca533bbd
+  data.tar.gz: 313cb2f92c8a02e48fea9b4f4208531c42f97753
 SHA512:
-  metadata.gz: 4fdf426a23e43e4ee9f13d19d2b940d329b8ead830fe0155043370c0341d7e8b1ef012d8712375f85d9235c9aea2a61e9084b80b6837dd8b37e01f0952c7988a
-  data.tar.gz: 092424fa6b574af0eef20f6585b0fe1026cc0baee453eb4d349bb106d9d71ca2f612df050dee82030140dea45f43f017043d784d5bf3ee70c2b2e74ea9fd1ae3
+  metadata.gz: 2afc286f24e1e82ce7809984c67358d71852deab99093d48d5a6aa383877741b754246fa2ea843ceefb04a5f79563a9ce152eaa8cd42e5376ddc90b402e24d22
+  data.tar.gz: ed09d9f96992efaeae2cfb4f55f260b35b89623eb5a17c51bef351ad29672c632d1ac170060d76ff2f980e7af85c09474e6ded21593aacff982ca915701d8dd4

data/bin/multiple_man

@@ -0,0 +1,11 @@
+#!/usr/bin/env ruby
+
+# Add gem directory to load path - (compatible with Ruby 1.9+)
+$LOAD_PATH.unshift File.expand_path('../lib', File.dirname(__FILE__))
+
+# Add the current working directory to the load path so we can load the config/environment
+$LOAD_PATH.unshift Dir.pwd
+
+require 'multiple_man'
+
+MultipleMan::CLI.run(ARGV)

data/lib/multiple_man.rb

@@ -4,7 +4,7 @@ require 'active_support'
 module MultipleMan
   Error = Class.new(StandardError)
   ConsumerError = Class.new(Error)
-  
+
   require 'multiple_man/railtie' if defined?(Rails)
 
   require 'multiple_man/mixins/publisher'
@@ -25,6 +25,8 @@ module MultipleMan
   require 'multiple_man/model_populator'
   require 'multiple_man/identity'
   require 'multiple_man/publish'
+  require 'multiple_man/runner'
+  require 'multiple_man/cli'
 
   require 'multiple_man/channel_maintenance/gc'
   require 'multiple_man/channel_maintenance/reaper'

data/lib/multiple_man/cli.rb

@@ -0,0 +1,39 @@
+require 'optparse'
+
+module MultipleMan
+  class Options < OptionParser
+    def self.parse!(args)
+      parser = new('Multiple Man')
+      parser.set_opts.parse!(args)
+      parser.options
+    end
+
+    attr_reader :options
+
+    def initialize(*args)
+      @options = { mode: :general, environment_path: 'config/environment' }
+      super
+    end
+
+    def set_opts
+      on('--seed', 'listen to the seeding queue') do
+        options[:mode] == :seed
+      end
+      on('-e', '--environment-path PATH', 'Set the path to load the web framework (default: config/environment)') do |value|
+        options[:environment_path] = value
+      end
+    end
+  end
+
+  module CLI
+    module_function
+
+    def run(args)
+      options = Options.parse!(args)
+
+      require options[:environment_path]
+
+      Runner.new(mode: options[:mode]).run
+    end
+  end
+end

data/lib/multiple_man/runner.rb

@@ -0,0 +1,83 @@
+require 'forwardable'
+
+module MultipleMan
+  def self.trap_signals!
+    return if @signals_trapped
+
+    @signals_trapped = true
+
+    handler = proc do |signal|
+      puts "received #{Signal.signame(signal)}"
+      exit
+    end
+
+    %w(INT QUIT TERM).each { |signal| Signal.trap(signal, handler) }
+  end
+
+  class Runner
+    extend Forwardable
+
+    MODES = [:general, :seed].freeze
+
+    def initialize(options = {})
+      @mode = options.fetch(:mode, :general)
+
+      raise ArgumentError, "undefined mode: #{mode}" unless MODES.include?(mode)
+    end
+
+    def run
+      MultipleMan.trap_signals!
+      preload_framework!
+      channel.prefetch(prefetch_size)
+      build_listener.listen
+      sleep
+    end
+
+    private
+
+    attr_reader :mode
+
+    def_delegators :config, :prefetch_size, :queue_name, :listeners, :topic_name
+
+    def preload_framework!
+      Rails.application.eager_load! if defined?(Rails)
+      Hanami::Application.preload_applications! if defined?(Hanami)
+    end
+
+    def build_listener
+      listener_class.new(
+        queue: channel.queue(*queue_params),
+        subscribers: listeners,
+        topic: topic_name
+      )
+    end
+
+    def listener_class
+      if seeding?
+        Consumers::Seed
+      else
+        Consumers::General
+      end
+    end
+
+    def queue_params
+      if seeding?
+        ["#{queue_name}.seed", durable: false, auto_delete: true]
+      else
+        [queue_name, durable: true, auto_delete: false]
+      end
+    end
+
+    def channel
+      @channel ||= Connection.connection.create_channel
+    end
+
+    def config
+      MultipleMan.configuration
+    end
+
+    def seeding?
+      mode == :seed
+    end
+  end
+end

data/lib/multiple_man/tasks/worker.rake

@@ -1,37 +1,12 @@
 namespace :multiple_man do
-  desc "Run multiple man listeners"
+  desc 'Run multiple man listeners'
   task worker: :environment do
-    channel = MultipleMan::Connection.connection.create_channel
-    channel.prefetch(MultipleMan.configuration.prefetch_size)
-    queue_name = MultipleMan.configuration.queue_name
-    queue = channel.queue(queue_name, durable: true, auto_delete: false)
-
-    run_listener(MultipleMan::Consumers::General, queue)
+    MultipleMan::Runner.new(mode: :general).run
   end
 
   desc 'Run a seeding listener'
   task seed: :environment do
-    channel = MultipleMan::Connection.connection.create_channel
-    channel.prefetch(MultipleMan.configuration.prefetch_size)
-    queue_name = MultipleMan.configuration.queue_name + '.seed'
-    queue = channel.queue(queue_name, durable: false, auto_delete: true)
-
-    run_listener(MultipleMan::Consumers::Seed, queue)
-  end
-
-  def run_listener(listener, queue)
-    Rails.application.eager_load! if defined?(Rails)
-
-    subscribers = MultipleMan.configuration.listeners
-    topic = MultipleMan.configuration.topic_name
-
-    listener.new(subscribers: subscribers, queue: queue, topic: topic).listen
-
-    Signal.trap("INT") { puts "received INT"; exit }
-    Signal.trap("QUIT") { puts "received QUIT"; exit }
-    Signal.trap("TERM") { puts "received TERM"; exit }
-
-    sleep
+    MultipleMan::Runner.new(mode: :seed).run
   end
 
   desc 'Run transitional worker'
@@ -54,10 +29,7 @@ namespace :multiple_man do
       MultipleMan::Consumers::Transitional.new(subscription: listener, queue: queue, topic: topic).listen
     end
 
-    Signal.trap("INT") { puts "received INT"; exit }
-    Signal.trap("QUIT") { puts "received QUIT"; exit }
-    Signal.trap("TERM") { puts "received TERM"; exit }
-
+    MultipleMan.trap_signals!
     sleep
   end
 

data/lib/multiple_man/version.rb

@@ -1,3 +1,3 @@
 module MultipleMan
-  VERSION = "1.0.0"
+  VERSION = "1.1.0"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: multiple_man
 version: !ruby/object:Gem::Version
-  version: 1.0.0
+  version: 1.1.0
 platform: ruby
 authors:
 - Ryan Brunner
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2016-06-23 00:00:00.000000000 Z
+date: 2016-06-29 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bunny
@@ -139,7 +139,8 @@ dependencies:
 description: MultipleMan syncs changes to ActiveRecord models via AMQP
 email:
 - ryan@influitive.com
-executables: []
+executables:
+- multiple_man
 extensions: []
 extra_rdoc_files: []
 files:
@@ -150,11 +151,13 @@ files:
 - LICENSE.txt
 - README.md
 - Rakefile
+- bin/multiple_man
 - circle.yml
 - lib/multiple_man.rb
 - lib/multiple_man/attribute_extractor.rb
 - lib/multiple_man/channel_maintenance/gc.rb
 - lib/multiple_man/channel_maintenance/reaper.rb
+- lib/multiple_man/cli.rb
 - lib/multiple_man/configuration.rb
 - lib/multiple_man/connection.rb
 - lib/multiple_man/consumers/general.rb
@@ -170,6 +173,7 @@ files:
 - lib/multiple_man/publish.rb
 - lib/multiple_man/railtie.rb
 - lib/multiple_man/routing_key.rb
+- lib/multiple_man/runner.rb
 - lib/multiple_man/subscribers/base.rb
 - lib/multiple_man/subscribers/model_subscriber.rb
 - lib/multiple_man/subscribers/registry.rb