multiple_man 0.3.2 → 0.4.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: e9fe7925c120cc5b704a13b62378b3b459da3b31
-  data.tar.gz: 21d3d6b079125195a0e85b398b5f106e21836a65
+  metadata.gz: 9b69cb35c812dfa3b4814a81a6cdca0aaa495afc
+  data.tar.gz: ea18e5f3984487b6f7278febc96269612b7038b3
 SHA512:
-  metadata.gz: 0bfec446ad1b8f131216f4493a2a63b61febbcefd1a5974ae7d8a1a9e91f22401f4bd20ab080a560e48e728afdfc0bee8b094fa696e8c7b6636eab856c644d48
-  data.tar.gz: adf476aed9b285617450d42cb823febeda32864bc1ebe9e844abd15dd21c30c22abc5b5d661ca87cc601bff75f8c01f35492b24f9fb1d855081da61b1f0c9fd8
+  metadata.gz: 8c1b1b58da01c48d575aea0ce54eac633e9996fed67a3f8860396596c721f122cdb92175823b2bb528d403fb7fb05e7748c55c14f1ead348fd32e2f6ce79a19c
+  data.tar.gz: 93b9be1ffd2e8a8427c35791ad87c76ef4e3a3b4f401cf8150b4a048797af2e0a8c4758826c8aa24b1d3fe120c0c34cc12d3e2b5e04b4ed019fdae1f23e3b3f3

data/lib/multiple_man.rb

@@ -14,6 +14,7 @@ module MultipleMan
   require 'multiple_man/connection'
   require 'multiple_man/routing_key'
   require 'multiple_man/listener'
+  require 'multiple_man/seeder_listener'
   require 'multiple_man/model_populator'
   require 'multiple_man/identity'
 
@@ -36,4 +37,4 @@ module MultipleMan
       raise ex
     end
   end
-end
+end

data/lib/multiple_man/listener.rb

@@ -25,8 +25,8 @@ module MultipleMan
     attr_accessor :subscription
 
     def listen
-      MultipleMan.logger.info "Listening for #{subscription.klass} with routing key #{subscription.routing_key}."
-      queue.bind(connection.topic, routing_key: subscription.routing_key).subscribe do |delivery_info, meta_data, payload|
+      MultipleMan.logger.info "Listening for #{subscription.klass} with routing key #{routing_key}."
+      queue.bind(connection.topic, routing_key: routing_key).subscribe do |delivery_info, meta_data, payload|
         process_message(delivery_info, payload)
       end
     end
@@ -34,8 +34,7 @@ module MultipleMan
     def process_message(delivery_info, payload)
       MultipleMan.logger.info "Processing message for #{delivery_info.routing_key}."
       begin
-        operation = delivery_info.routing_key.split(".").last
-        subscription.send(operation, JSON.parse(payload).with_indifferent_access)
+        subscription.send(operation(delivery_info), JSON.parse(payload).with_indifferent_access)
       rescue Exception => ex
         MultipleMan.logger.error "   Error - #{ex.message}\n\n#{ex.backtrace}"
         MultipleMan.error(ex)
@@ -44,10 +43,18 @@ module MultipleMan
       end
     end
 
+    def operation(delivery_info)
+      delivery_info.routing_key.split(".").last
+    end
+
     def queue
       connection.queue(subscription.queue_name, durable: true, auto_delete: false)
     end
 
+    def routing_key
+      subscription.routing_key
+    end
+
   private
 
     def connection
@@ -55,4 +62,4 @@ module MultipleMan
     end
 
   end
-end
+end

data/lib/multiple_man/mixins/publisher.rb

@@ -2,13 +2,25 @@ module MultipleMan
   module Publisher
     def Publisher.included(base)
       base.extend(ClassMethods)
+      base.after_commit(on: :create) { |r| r.multiple_man_publish(:create) }
+      base.after_commit(on: :update) { |r| r.multiple_man_publish(:update) }
+      base.after_commit(on: :destroy) { |r| r.multiple_man_publish(:destroy) }
+    end
+
+    def multiple_man_publish(operation=:create)
+      self.class.multiple_man_publisher.publish(self, operation) 
     end
 
     module ClassMethods
+      
+      def multiple_man_publish(operation=:create)
+        multiple_man_publisher.publish(self, operation) 
+      end
+
+      attr_accessor :multiple_man_publisher
+
       def publish(options = {})
-        after_commit ModelPublisher.new(:create, options), on: :create
-        after_commit ModelPublisher.new(:update, options), on: :update
-        after_commit ModelPublisher.new(:destroy, options), on: :destroy
+        self.multiple_man_publisher = ModelPublisher.new(options)
       end
     end
   end

data/lib/multiple_man/model_publisher.rb

@@ -1,25 +1,25 @@
 module MultipleMan
   class ModelPublisher
 
-    def initialize(operation, options = {})
-      self.operation = operation
+    def initialize(options = {})
       self.options = options
     end
 
-    def after_commit(record)
+    def publish(records, operation=:create)
       return unless MultipleMan.configuration.enabled
 
-      MultipleMan.logger.info("Publishing #{record}")
       Connection.connect do |connection|
-        push_record(connection, record)
+        all_records(records) do |record|
+          push_record(connection, record, operation)
+        end
       end
     end
 
   private
 
-    attr_accessor :operation, :options
+    attr_accessor :options
 
-    def push_record(connection, record)
+    def push_record(connection, record, operation)
       data = record_data(record)
       routing_key = RoutingKey.new(record_type(record), operation).to_s
       MultipleMan.logger.debug("  Record Data: #{data} | Routing Key: #{routing_key}")
@@ -28,12 +28,23 @@ module MultipleMan
       MultipleMan.error(ex)
     end
 
+    def all_records(records, &block)
+      if records.respond_to?(:find_each)
+        records.find_each(&block)
+      elsif records.respond_to?(:each)
+        records.each(&block)
+      else
+        yield records
+      end
+    end
+
+
     def record_type(record)
       options[:as] || record.class.name
     end
 
     def record_data(record)
-      { 
+      {
         id: Identity.new(record, identifier).value,
         data: serializer(record).as_json
       }.to_json
@@ -56,6 +67,6 @@ module MultipleMan
     def identifier
       options[:identifier]
     end
-    
+
   end
-end
+end

data/lib/multiple_man/routing_key.rb

@@ -1,6 +1,6 @@
 module MultipleMan
   class RoutingKey
-    ALLOWED_OPERATIONS = [:create, :update, :destroy, :"#"]
+    ALLOWED_OPERATIONS = [:create, :update, :destroy, :seed, :"#"]
 
     def initialize(klass, operation = :"#")
       self.klass = klass
@@ -8,7 +8,11 @@ module MultipleMan
     end
 
     def to_s
-      "#{topic_name}.#{klass}.#{operation}"
+      if operation == :seed
+        "#{topic_name}.#{operation}.#{klass}"
+      else
+        "#{topic_name}.#{klass}.#{operation}"
+      end
     end
 
     attr_reader :operation
@@ -25,4 +29,4 @@ module MultipleMan
     end
 
   end
-end
+end

data/lib/multiple_man/seeder_listener.rb

@@ -0,0 +1,12 @@
+module MultipleMan
+  class SeederListener < Listener
+    def routing_key
+      subscription.routing_key(:seed)
+    end
+
+    # seeds should only ever be a create
+    def operation(delivery_info)
+      "create"
+    end
+  end
+end

data/lib/multiple_man/subscribers/base.rb

@@ -19,8 +19,12 @@ module MultipleMan::Subscribers
       # noop
     end
 
-    def routing_key
-      MultipleMan::RoutingKey.new(klass).to_s
+    def seed
+      # noop
+    end
+
+    def routing_key(operation = :'#')
+      MultipleMan::RoutingKey.new(klass, operation).to_s
     end
 
     def queue_name
@@ -31,4 +35,4 @@ module MultipleMan::Subscribers
 
     attr_writer :klass
   end
-end
+end

data/lib/multiple_man/subscribers/model_subscriber.rb

@@ -1,4 +1,4 @@
-module MultipleMan::Subscribers 
+module MultipleMan::Subscribers
   class ModelSubscriber < Base
 
     def initialize(klass, options)
@@ -15,6 +15,7 @@ module MultipleMan::Subscribers
     end
 
     alias_method :update, :create
+    alias_method :seed, :create
 
     def destroy(payload)
       model = find_model(payload[:id])
@@ -30,4 +31,4 @@ module MultipleMan::Subscribers
     attr_writer :klass
 
   end
-end
+end

data/lib/multiple_man/tasks/worker.rake

@@ -1,14 +1,23 @@
 namespace :multiple_man do
   desc "Run multiple man listeners"
   task :worker => :environment do
+    run_listener(MultipleMan::Listener)
+  end
+
+  desc 'Run a seeding listener'
+  task seed: :environment do
+    run_listener(MultipleMan::SeederListener)
+  end
+
+  def run_listener(listener)
     Rails.application.eager_load!
 
     MultipleMan::Connection.connect do |connection|
-      MultipleMan::Listener.start(connection)
+      listener.start(connection)
 
       while(true)
         sleep 10
       end
     end
   end
-end
+end

data/lib/multiple_man/version.rb

@@ -1,3 +1,3 @@
 module MultipleMan
-  VERSION = "0.3.2"
+  VERSION = "0.4.0"
 end

data/spec/connection_spec.rb

@@ -9,7 +9,7 @@ describe MultipleMan::Connection do
     it "should open a connection and a channel" do
       mock_bunny.should_receive(:start)
       Bunny.should_receive(:new).and_return(mock_bunny)
-      mock_bunny.should_receive(:create_channel).exactly(25).times
+      mock_bunny.should_receive(:create_channel)
 
       described_class.connect
     end
@@ -19,4 +19,4 @@ describe MultipleMan::Connection do
 
   its(:topic_name) { should == MultipleMan.configuration.topic_name }
 
-end
+end

data/spec/model_publisher_spec.rb

@@ -25,20 +25,20 @@ describe MultipleMan::ModelPublisher do
     end
   end
 
-  subject { described_class.new(:create, fields: [:foo]) }
+  subject { described_class.new(fields: [:foo]) }
 
-  describe "after_commit" do
+  describe "publish" do
     it "should send the jsonified version of the model to the correct routing key" do
       MultipleMan::AttributeExtractor.any_instance.should_receive(:as_json).and_return({foo: "bar"})
       topic_stub.should_receive(:publish).with('{"id":"10","data":{"foo":"bar"}}', routing_key: "app.MockObject.create")
-      described_class.new(:create, fields: [:foo]).after_commit(MockObject.new)
+      described_class.new(fields: [:foo]).publish(MockObject.new)
     end
 
     it "should call the error handler on error" do
       ex = Exception.new("Bad stuff happened")
       topic_stub.stub(:publish) { raise ex }
       MultipleMan.should_receive(:error).with(ex)
-      described_class.new(:create, fields: [:foo]).after_commit(MockObject.new)
+      described_class.new(fields: [:foo]).publish(MockObject.new)
     end
   end
 
@@ -51,12 +51,12 @@ describe MultipleMan::ModelPublisher do
       end
     end
 
-    subject { described_class.new(:create, with: MySerializer) }
+    subject { described_class.new(with: MySerializer) }
 
     it "should get its data from the serializer" do
       obj = MockObject.new
       topic_stub.should_receive(:publish).with('{"id":"10","data":{"a":"yes"}}', routing_key: "app.MockObject.create")
-      subject.after_commit(obj)
+      subject.publish(obj)
     end
   end
 

data/spec/publisher_spec.rb

@@ -5,7 +5,7 @@ describe MultipleMan::Publisher do
     class << self
       attr_accessor :subscriber
 
-      def after_commit(subscriber, operation)
+      def after_commit(subscriber)
         self.subscriber = subscriber
       end
     end
@@ -13,15 +13,15 @@ describe MultipleMan::Publisher do
     include MultipleMan::Publisher
 
     def save
-      self.class.subscriber.after_commit(self)
+      self.multiple_man_publish
     end
   end
 
   describe "including" do
     it "should add an after commit hook" do
       # once for each operation
-      MockClass.should_receive(:after_commit).exactly(3).times
       MockClass.publish
+      MockClass.multiple_man_publisher.should be_kind_of MultipleMan::ModelPublisher
     end
   end
 
@@ -30,7 +30,7 @@ describe MultipleMan::Publisher do
     it "should tell ModelPublisher to publish" do
       my_mock = MockClass.new
       mock_publisher = double(MultipleMan::ModelPublisher)
-      MultipleMan::ModelPublisher.any_instance.should_receive(:after_commit).with(my_mock)
+      MultipleMan::ModelPublisher.any_instance.should_receive(:publish).with(my_mock, :create)
       my_mock.save
     end 
   end

data/spec/routing_key_spec.rb

@@ -1,6 +1,6 @@
 require 'spec_helper'
 
-describe MultipleMan::RoutingKey do 
+describe MultipleMan::RoutingKey do
   class MockObject
   end
 
@@ -28,6 +28,11 @@ describe MultipleMan::RoutingKey do
       it { should == "app.MockObject.destroy" }
     end
 
+    context "seeding" do
+      let(:operation) { :seed }
+      it { should == "app.seed.MockObject" }
+    end
+
     context "not specified" do
       subject { described_class.new("MockObject").to_s }
       it { should == "app.MockObject.#" }
@@ -51,4 +56,4 @@ describe MultipleMan::RoutingKey do
     end
 
   end
-end
+end

data/spec/seeder_listener_spec.rb

@@ -0,0 +1,21 @@
+require 'spec_helper'
+
+describe MultipleMan::SeederListener do
+  class MockClass1; end
+
+  let(:connection_stub) { double(MultipleMan::Connection, queue: queue_stub, topic: 'app') }
+  let(:queue_stub) { double(Bunny::Queue, bind: bind_stub) }
+  let(:bind_stub) { double(:bind, subscribe: nil)}
+
+  before { MultipleMan::Listener.stub(:connection).and_return(connection_stub) }
+
+  it 'listens to seed events' do
+    listener = described_class.new(double(MultipleMan::Subscribers::ModelSubscriber,
+                                   klass: MockClass1,
+                                   routing_key: "seed.MockClass1",
+                                   queue_name: "MockClass1"))
+
+    queue_stub.should_receive(:bind).with('app', routing_key: "seed.MockClass1")
+    listener.listen
+  end
+end

data/spec/subscribers/base_spec.rb

@@ -14,4 +14,4 @@ describe MultipleMan::Subscribers::Base do
     end
     described_class.new(MockClass).queue_name.should == "app.test.MockClass"
   end
-end
+end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: multiple_man
 version: !ruby/object:Gem::Version
-  version: 0.3.2
+  version: 0.4.0
 platform: ruby
 authors:
 - Ryan Brunner
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2014-01-24 00:00:00.000000000 Z
+date: 2014-03-21 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler
@@ -119,6 +119,7 @@ files:
 - lib/multiple_man/model_publisher.rb
 - lib/multiple_man/railtie.rb
 - lib/multiple_man/routing_key.rb
+- lib/multiple_man/seeder_listener.rb
 - lib/multiple_man/subscribers/base.rb
 - lib/multiple_man/subscribers/model_subscriber.rb
 - lib/multiple_man/subscribers/registry.rb
@@ -134,6 +135,7 @@ files:
 - spec/model_publisher_spec.rb
 - spec/publisher_spec.rb
 - spec/routing_key_spec.rb
+- spec/seeder_listener_spec.rb
 - spec/spec_helper.rb
 - spec/subscriber_spec.rb
 - spec/subscribers/base_spec.rb
@@ -173,6 +175,7 @@ test_files:
 - spec/model_publisher_spec.rb
 - spec/publisher_spec.rb
 - spec/routing_key_spec.rb
+- spec/seeder_listener_spec.rb
 - spec/spec_helper.rb
 - spec/subscriber_spec.rb
 - spec/subscribers/base_spec.rb