lib/seymour/distributable.rb in seymour-0.0.3 vs lib/seymour/distributable.rb in seymour-0.0.4

@@ -5,47 +5,71 @@
 
     included do
     end
 
     module ClassMethods
+      DEFAULT_BATCH_SIZE = 500
+
       attr_accessor :audience_names, :feed_class_names
 
       def audience(*names)
         options = names.extract_options!
         names.each do |name|
-          feed_name = options[:feed] || "#{name.downcase.to_s.singularize}_feed".camelize
-          audience_to_feed_classes[name] = feed_name
+          feed_name = options.delete(:feed) || "#{name.downcase.to_s.singularize}_feed".camelize
+          audience_mappings[name] = [feed_name, options]
         end
       end
 
       def audience_names
-        audience_to_feed_classes.keys
+        audience_mappings.keys
       end
 
       def feed_class_names
-        audience_to_feed_classes.values
+        audience_mappings.values.map(&:first)
       end
 
       def feeds_for(activity)
-        audience_to_feed_classes.map do |audience_name, feed_class_name|
-          activity.send(audience_name).map do |member|
-            feed_class_name.constantize.new(member)
-          end
-        end.flatten
+        tap_feeds_for(activity)
       end
 
+      def distribute(activity)
+        tap_feeds_for(activity) { |feed| feed.push(activity) }
+      end
+
       private
 
-      def audience_to_feed_classes
-        @audience_to_feed_classes ||= {}
+      def tap_feeds_for(activity, &block)
+        [].tap do |feeds|
+          audience_mappings.each do |audience_name, mapping|
+            feed_class_name, options = mapping
+            try_find_each(activity.send(audience_name), options) do |member|
+              feed = feed_class_name.constantize.new(member)
+              yield feed if block_given?
+              feeds << feed
+            end
+          end
+        end
       end
 
+      def audience_mappings
+        @audience_mappings ||= {}
+      end
+
+      def try_find_each(activity_audience, options = {}, &block)
+        if defined? activity_audience.find_each
+          options[:batch_size] ||= DEFAULT_BATCH_SIZE
+          activity_audience.find_each(options) &block
+        else
+          activity_audience.each &block
+        end
+      end
+
     end
 
     module InstanceMethods
 
       def distribute
-        feeds.map { |feed| feed.push(self) }
+        self.class.distribute(self)
       end
 
       def feeds
         self.class.feeds_for(self)
       end
\ No newline at end of file