1 files changed, 37 insertions, 0 deletions
diff --git a/app/workers/gitlab/bitbucket_import/advance_stage_worker.rb b/app/workers/gitlab/bitbucket_import/advance_stage_worker.rb
new file mode 100644
index 00000000000..7f281352a1b
--- /dev/null
+++ b/app/workers/gitlab/bitbucket_import/advance_stage_worker.rb
@@ -0,0 +1,37 @@
+# frozen_string_literal: true
+
+module Gitlab
+  module BitbucketImport
+    # AdvanceStageWorker is a worker used by the BitBucket Importer to wait for a
+    # number of jobs to complete, without blocking a thread. Once all jobs have
+    # been completed this worker will advance the import process to the next
+    # stage.
+    class AdvanceStageWorker # rubocop:disable Scalability/IdempotentWorker
+      include ApplicationWorker
+      include ::Gitlab::Import::AdvanceStage
+
+      data_consistency :delayed
+
+      sidekiq_options dead: false, retry: 3
+
+      feature_category :importers
+
+      loggable_arguments 1, 2
+
+      # The known importer stages and their corresponding Sidekiq workers.
+      STAGES = {
+        finish: Stage::FinishImportWorker
+      }.freeze
+
+      def find_import_state(project_id)
+        ProjectImportState.jid_by(project_id: project_id, status: :started)
+      end
+
+      private
+
+      def next_stage_worker(next_stage)
+        STAGES.fetch(next_stage.to_sym)
+      end
+    end
+  end
+end