Merge pull request rails#51785 from fatkodima/touch_all-in-batches

Support `touch_all` in batches

Author: byroot

activerecord/CHANGELOG.md

@@ -1,3 +1,11 @@
+*   Support `touch_all` in batches.
+
+    ```ruby
+    Post.in_batches.touch_all
+    ```
+
+    *fatkodima*
+
 *   Add support for `:if_not_exists` and `:force` options to `create_schema`
 
     *fatkodima*

activerecord/lib/active_record/relation/batches/batch_enumerator.rb

@@ -77,6 +77,17 @@ def update_all(updates)
         end
       end
 
+      # Touches records in batches. Returns the total number of rows affected.
+      #
+      #   Person.in_batches.touch_all
+      #
+      # See Relation#touch_all for details of how each batch is touched.
+      def touch_all(...)
+        sum do |relation|
+          relation.touch_all(...)
+        end
+      end
+
       # Destroys records in batches.
       #
       #   Person.where("age < 10").in_batches.destroy_all

activerecord/test/cases/batches_test.rb

@@ -4,10 +4,11 @@
 require "models/comment"
 require "models/post"
 require "models/subscriber"
+require "models/developer"
 require "models/cpk"
 
 class EachTest < ActiveRecord::TestCase
-  fixtures :posts, :subscribers, :cpk_orders
+  fixtures :posts, :subscribers, :developers, :cpk_orders
 
   def setup
     @posts = Post.order("id asc")
@@ -384,6 +385,22 @@ def test_in_batches_update_all_returns_zero_when_no_batches
     assert_equal 0, Post.where("1=0").in_batches(of: 2).update_all(title: "updated-title")
   end
 
+  def test_in_batches_touch_all_affect_all_records
+    time = Time.new(2000, 1, 1, 0, 0, 0)
+    assert_queries_count(6 + 6) do # 6 selects, 6 updates
+      Developer.in_batches(of: 2).touch_all(time: time)
+    end
+    assert_equal Developer.all.pluck(:updated_at), [time] * Developer.count
+  end
+
+  def test_in_batches_touch_all_returns_rows_affected
+    assert_equal 11, Developer.in_batches(of: 2).touch_all
+  end
+
+  def test_in_batches_touch_all_returns_zero_when_no_batches
+    assert_equal 0, Developer.where("1=0").in_batches(of: 2).touch_all
+  end
+
   def test_in_batches_delete_all_should_not_delete_records_in_other_batches
     not_deleted_count = Post.where("id <= 2").count
     Post.where("id > 2").in_batches(of: 2).delete_all