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

BatchRecord: Revert selecting only PK
doctolib · May 16, 2024 · fbcf746 · fbcf746
2 parents 2475fb4 + 5b69b55
commit fbcf746
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.select(primary_key).take
+        first = next_scope.take
 
         return unless first
 
-        last = next_scope.select(primary_key).offset(@of).take
+        last = next_scope.offset(@of).take
 
         first_key = first[primary_key]
         last_key = last.nil? ? nil : last[primary_key]