Merge branch 'georgekoltsov/relation_batch_export_in_batches' into 'master'

Update creation of Direct Transfer batch exports to use `in_batches`

See merge request https://gitlab.com/gitlab-org/gitlab/-/merge_requests/138815



Merged-by: Rodrigo Tomonari <rtomonari@gitlab.com>
Approved-by: Carla Drago <cdrago@gitlab.com>
Approved-by: Rodrigo Tomonari <rtomonari@gitlab.com>
Co-authored-by: George Koltsov <gkoltsov@gitlab.com>

app/services/bulk_imports/batched_relation_export_service.rb

@@ -65,19 +65,27 @@ def update_export!(event)
       )
     end
+    # rubocop:disable Cop/InBatches
+    # rubocop:disable CodeReuse/ActiveRecord
     def enqueue_batch_exports
-      resolved_relation.each_batch(of: BATCH_SIZE) do |batch, batch_number|
+      batch_number = 0
+
+      resolved_relation.in_batches(of: BATCH_SIZE) do |batch|
+        batch_number += 1
+
         batch_id = find_or_create_batch(batch_number).id
-        ids = batch.pluck(batch.model.primary_key) # rubocop:disable CodeReuse/ActiveRecord
+        ids = batch.pluck(batch.model.primary_key)
         Gitlab::Cache::Import::Caching.set_add(self.class.cache_key(export.id, batch_id), ids, timeout: CACHE_DURATION)
         RelationBatchExportWorker.perform_async(user.id, batch_id)
       end
     end
+    # rubocop:enable Cop/InBatches
     def find_or_create_batch(batch_number)
-      export.batches.find_or_create_by!(batch_number: batch_number) # rubocop:disable CodeReuse/ActiveRecord
+      export.batches.find_or_create_by!(batch_number: batch_number)
     end
+    # rubocop:enable CodeReuse/ActiveRecord
   end
 end