summaryrefslogtreecommitdiff
path: root/lib/gitlab/github_import/parallel_importer.rb
blob: 9d81441d96e2699c96e46ab58186a613c7aa4f46 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
# frozen_string_literal: true

module Gitlab
  module GithubImport
    # The ParallelImporter schedules the importing of a GitHub project using
    # Sidekiq.
    class ParallelImporter
      attr_reader :project

      def self.async?
        true
      end

      def self.imports_repository?
        true
      end

      # This is a workaround for a Ruby 2.3.7 bug. rspec-mocks cannot restore
      # the visibility of prepended modules. See
      # https://github.com/rspec/rspec-mocks/issues/1231 for more details.
      if Rails.env.test?
        def self.requires_ci_cd_setup?
          raise NotImplementedError
        end
      end

      def initialize(project)
        @project = project
      end

      def execute
        jid = generate_jid

        # The original import JID is the JID of the RepositoryImportWorker job,
        # which will be removed once that job completes. Reusing that JID could
        # result in StuckImportJobsWorker marking the job as stuck before we get
        # to running Stage::ImportRepositoryWorker.
        #
        # We work around this by setting the JID to a custom generated one, then
        # refreshing it in the various stages whenever necessary.
        Gitlab::SidekiqStatus
          .set(jid, StuckImportJobsWorker::IMPORT_JOBS_EXPIRATION)

        project.import_state.update_column(:jid, jid)

        Stage::ImportRepositoryWorker
          .perform_async(project.id)

        true
      end

      def generate_jid
        "github-importer/#{project.id}"
      end
    end
  end
end