Add latest changes from gitlab-org/gitlab@16-5-stable-eev16.5.0-rc42

1 files changed, 57 insertions, 0 deletions

diff --git a/lib/gitlab/bitbucket_import/importers/lfs_objects_importer.rb b/lib/gitlab/bitbucket_import/importers/lfs_objects_importer.rb
new file mode 100644
index 00000000000..aa9ff7000f1
--- /dev/null
+++ b/lib/gitlab/bitbucket_import/importers/lfs_objects_importer.rb
@@ -0,0 +1,57 @@
+# frozen_string_literal: true
+
+module Gitlab
+  module BitbucketImport
+    module Importers
+      class LfsObjectsImporter
+        include ParallelScheduling
+
+        def execute
+          log_info(import_stage: 'import_lfs_objects', message: 'starting')
+
+          download_service = Projects::LfsPointers::LfsObjectDownloadListService.new(project)
+
+          begin
+            queue_workers(download_service) if project.lfs_enabled?
+          rescue StandardError => e
+            track_import_failure!(project, exception: e)
+          end
+
+          log_info(import_stage: 'import_lfs_objects', message: 'finished')
+
+          job_waiter
+        end
+
+        private
+
+        def sidekiq_worker_class
+          ImportLfsObjectWorker
+        end
+
+        def collection_method
+          :lfs_objects
+        end
+
+        def id_for_already_enqueued_cache(object)
+          object.oid
+        end
+
+        def queue_workers(download_service)
+          download_service.each_list_item do |lfs_download_object|
+            # Needs to come before `already_enqueued?` as `jobs_remaining` resets to zero when the job restarts and
+            # jobs_remaining needs to be the total amount of enqueued jobs
+            job_waiter.jobs_remaining += 1
+
+            next if already_enqueued?(lfs_download_object)
+
+            job_delay = calculate_job_delay(job_waiter.jobs_remaining)
+
+            sidekiq_worker_class.perform_in(job_delay, project.id, lfs_download_object.to_hash, job_waiter.key)
+
+            mark_as_enqueued(lfs_download_object)
+          end
+        end
+      end
+    end
+  end
+end