Contents

module CassandraObject
  module Batches
    extend ActiveSupport::Concern

    module ClassMethods
      def find_each
        connection.each(column_family, {:count => 500}) do |k, v|
          yield instantiate(k, v)
        end
      end

      def find_in_batches(options = {})
        batch_size = options.delete(:batch_size) || 1000

        batch = []

        find_each do |record|
          batch << record
          if batch.size == batch_size
            yield(batch)
            batch = []
          end
        end

        if batch.size > 0
          yield batch
        end
      end

      def batch(&block)
        connection.batch(&block)
      end
    end
  end
end

Version data entries

24 entries across 24 versions & 1 rubygems

Version	Path
gotime-cassandra_object-2.9.0	lib/cassandra_object/batches.rb
gotime-cassandra_object-2.8.6	lib/cassandra_object/batches.rb
gotime-cassandra_object-2.8.5	lib/cassandra_object/batches.rb
gotime-cassandra_object-2.8.4	lib/cassandra_object/batches.rb