Merge pull request #138 from doctolib/batch-over-select-id

BatchRecord: select primary keys instead of full records
doctolib · Feb 26, 2024 · fe0be8d · fe0be8d
2 parents 15bac4a + 427fe31
commit fe0be8d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/lib/safe-pg-migrations/helpers/batch_over.rb b/lib/safe-pg-migrations/helpers/batch_over.rb
@@ -19,11 +19,11 @@ def each_batch
       def next_batch
         return if endless?
 
-        first = next_scope.take
+        first = next_scope.select(primary_key).take
 
         return unless first
 
-        last = next_scope.offset(@of).take
+        last = next_scope.select(primary_key).offset(@of).take
 
         first_key = first[primary_key]
         last_key = last.nil? ? nil : last[primary_key]