Contents

require_relative 'batchifier/exceptions'
require_relative 'batchifier/interface'
require_relative 'batchifier/strategy'
require_relative 'batchifier/add'
require_relative 'batchifier/no_response'
require_relative 'batchifier/array_merge'
require_relative 'batchifier/maintain_unique'

module Wor
  module Batchifier

    # This module exports the function execute_in_batches, that needs a collections and
    # => optionaly a batch_size and a merge strategy. It will slice the collection and
    # => apply the chozen strategy to all chunks and merge the results. It expects the responses
    # => to be Hash. It can ignore them if the given strategy is no_response
    def execute_in_batches(collection, batch_size: 100, strategy: :add)
      strategy_class = classify_strategy(strategy)
      collection.lazy.each_slice(batch_size).inject(strategy_class.new.base_case) do |rec, chunk|
        response = yield(chunk)
        merge(response, rec, strategy)
      end
    end

    private

    def merge(response, rec, strategy)
      return Wor::Batchifier.classify_strategy(strategy).new.merge_strategy(response,rec)
    end

    def classify_strategy(strategy)
      strategy_class_name = strategy.to_s.split('_').collect(&:capitalize).join
      Kernel.const_get(strategy_class_name)
    rescue NameError => e
      raise Wor::Batchifier::Exceptions::StrategyNotFound
    end
  end
end

Version data entries

1 entries across 1 versions & 1 rubygems

Version	Path
wor-batchifier-0.0.1	lib/wor/batchifier.rb