class AWS::S3::ObjectCollection::BatchHelper

def add item
  @batch << item
  if @batch.size == @batch_size
    process_batch
    @batch = []
  end
  item
end

def after_batch &block
  @after_batch = block
end

def complete!
  process_batch unless @batch.empty?
end

def initialize batch_size, &block
  @batch_size = batch_size
  @block = block
  @batch = []
end

def process_batch
  response = @block.call(@batch)
  @after_batch.call(response) if @after_batch
end