clear changes after successful update

Author: freddiescadding-pennylane

Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    activerecord_batch_update (1.0.0)
+    activerecord_batch_update (1.0.1)
       activerecord (>= 7.0, < 8.1)
       activesupport (>= 7.0, < 8.1)
 

activerecord_batch_update.gemspec

@@ -2,7 +2,7 @@
 
 Gem::Specification.new do |s|
   s.name        = 'activerecord_batch_update'
-  s.version     = '1.0.0'
+  s.version     = '1.0.1'
   s.summary     = 'Update multiple records with different values in a small number of queries'
   s.description = ''
   s.authors     = ['Quentin de Metz']

lib/activerecord_batch_update.rb

@@ -35,6 +35,7 @@ def batch_update(entries, columns:, batch_size: 100, validate: true)
       end
 
       connection.clear_query_cache if connection.query_cache_enabled
+      entries.each(&:changes_applied)
 
       updated_count
     end

spec/activerecord_batch_update_spec.rb

@@ -180,6 +180,13 @@
                                                                          .and(change { cat2.reload.birthday })
       end
 
+      it 'clears the changes after the update' do
+        cat1.name = 'Nala'
+        Cat.batch_update([cat1], columns: :all)
+
+        expect(cat1.changes_to_save).to be_empty
+      end
+
       context 'when some fields are encrypted' do
         let!(:cat1) { Cat.create!(name: 'Felix', birthday: Date.new(2010, 1, 1)) }