diff options
| author | Ben Woosley <ben.woosley@gmail.com> | 2013-06-18 02:21:52 -0700 |
|---|---|---|
| committer | Ben Woosley <ben.woosley@gmail.com> | 2013-06-19 04:04:13 -0700 |
| commit | 25042359b388a73bae798e023276df59e53c74e2 (patch) | |
| tree | 8128c43c74df54dc60a1c8e4d755af179a830ab7 /activerecord/lib/active_record/relation/batches.rb | |
| parent | 7324624f748b9717a5f9757ca89d791563ac5f95 (diff) | |
| download | rails-25042359b388a73bae798e023276df59e53c74e2.tar.gz rails-25042359b388a73bae798e023276df59e53c74e2.tar.bz2 rails-25042359b388a73bae798e023276df59e53c74e2.zip | |
When .find_each is called without a block, return an Enumerator.
This lets us do things like call: .find_each.with_index
Diffstat (limited to 'activerecord/lib/active_record/relation/batches.rb')
| -rw-r--r-- | activerecord/lib/active_record/relation/batches.rb | 15 |
1 files changed, 13 insertions, 2 deletions
diff --git a/activerecord/lib/active_record/relation/batches.rb b/activerecord/lib/active_record/relation/batches.rb index 41291844fc..91ea1756c4 100644 --- a/activerecord/lib/active_record/relation/batches.rb +++ b/activerecord/lib/active_record/relation/batches.rb @@ -19,6 +19,13 @@ module ActiveRecord # person.party_all_night! # end # + # If you do not provide a block to #find_each, it will return an Enumerator + # for chaining with other methods: + # + # Person.find_each.with_index do |person, index| + # person.award_trophy(index + 1) + # end + # # ==== Options # * <tt>:batch_size</tt> - Specifies the size of the batch. Default to 1000. # * <tt>:start</tt> - Specifies the starting point for the batch processing. @@ -40,8 +47,12 @@ module ActiveRecord # NOTE: You can't set the limit either, that's used to control # the batch sizes. def find_each(options = {}) - find_in_batches(options) do |records| - records.each { |record| yield record } + if block_given? + find_in_batches(options) do |records| + records.each { |record| yield record } + end + else + enum_for :find_each, options end end |