Class: ActiveRecord::Batches::BatchEnumerator

Inherits:
Object
  • Object
show all
Includes:
Enumerable
Defined in:
lib/active_record/relation/batches/batch_enumerator.rb

Instance Attribute Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(of: 1000, start: nil, finish: nil, relation:, order: :asc, use_ranges: nil) ⇒ BatchEnumerator

:nodoc:



8
9
10
11
12
13
14
15
 
# File 'lib/active_record/relation/batches/batch_enumerator.rb', line 8

def initialize(of: 1000, start: nil, finish: nil, relation:, order: :asc, use_ranges: nil) # :nodoc:
  @of       = of
  @relation = relation
  @start = start
  @finish = finish
  @order = order
  @use_ranges = use_ranges
end

Instance Attribute Details

#finish ⇒ Object (readonly)

The primary key value at which the BatchEnumerator ends, inclusive of the value.



21
22
23
 
# File 'lib/active_record/relation/batches/batch_enumerator.rb', line 21

def finish
  @finish
end

#relation ⇒ Object (readonly)

The relation from which the BatchEnumerator yields batches.



24
25
26
 
# File 'lib/active_record/relation/batches/batch_enumerator.rb', line 24

def relation
  @relation
end

#start ⇒ Object (readonly)

The primary key value from which the BatchEnumerator starts, inclusive of the value.



18
19
20
 
# File 'lib/active_record/relation/batches/batch_enumerator.rb', line 18

def start
  @start
end

Instance Method Details

#batch_size ⇒ Object

The size of the batches yielded by the BatchEnumerator.



27
28
29
 
# File 'lib/active_record/relation/batches/batch_enumerator.rb', line 27

def batch_size
  @of
end

#delete_all ⇒ Object

Deletes records in batches. Returns the total number of rows affected.

Person.in_batches.delete_all

See Relation#delete_all for details of how each batch is deleted.



65
66
67
 
# File 'lib/active_record/relation/batches/batch_enumerator.rb', line 65

def delete_all
  sum(&:delete_all)
end

#destroy_all ⇒ Object

Destroys records in batches.

Person.where("age < 10").in_batches.destroy_all

See Relation#destroy_all for details of how each batch is destroyed.



85
86
87
 
# File 'lib/active_record/relation/batches/batch_enumerator.rb', line 85

def destroy_all
  each(&:destroy_all)
end

#each(&block) ⇒ Object

Yields an ActiveRecord::Relation object for each batch of records.

Person.in_batches.each do |relation|
  relation.update_all(awesome: true)
end



94
95
96
97
98
 
# File 'lib/active_record/relation/batches/batch_enumerator.rb', line 94

def each(&block)
  enum = @relation.to_enum(:in_batches, of: @of, start: @start, finish: @finish, load: false, order: @order, use_ranges: @use_ranges)
  return enum.each(&block) if block_given?
  enum
end

#each_record(&block) ⇒ Object

Looping through a collection of records from the database (using the all method, for example) is very inefficient since it will try to instantiate all the objects at once.

In that case, batch processing methods allow you to work with the records in batches, thereby greatly reducing memory consumption.

Person.in_batches.each_record do |person|
  person.do_awesome_stuff
end

Person.where("age > 21").in_batches(of: 10).each_record do |person|
  person.party_all_night!
end

If you do not provide a block to #each_record, it will return an Enumerator for chaining with other methods:

Person.in_batches.each_record.with_index do |person, index|
  person.award_trophy(index + 1)
end



52
53
54
55
56
57
58
 
# File 'lib/active_record/relation/batches/batch_enumerator.rb', line 52

def each_record(&block)
  return to_enum(:each_record) unless block_given?

  @relation.to_enum(:in_batches, of: @of, start: @start, finish: @finish, load: true, order: @order).each do |relation|
    relation.records.each(&block)
  end
end

#update_all(updates) ⇒ Object

Updates records in batches. Returns the total number of rows affected.

Person.in_batches.update_all("age = age + 1")

See Relation#update_all for details of how each batch is updated.



74
75
76
77
78
 
# File 'lib/active_record/relation/batches/batch_enumerator.rb', line 74

def update_all(updates)
  sum do |relation|
    relation.update_all(updates)
  end
end