lib/gitlab/github_import/importer/pull_requests_reviews_importer.rb


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

# frozen_string_literal: true

module Gitlab
  module GithubImport
    module Importer
      class PullRequestsReviewsImporter
        include ParallelScheduling

        def importer_class
          PullRequestReviewImporter
        end

        def representation_class
          Gitlab::GithubImport::Representation::PullRequestReview
        end

        def sidekiq_worker_class
          ImportPullRequestReviewWorker
        end

        def collection_method
          :pull_request_reviews
        end

        def id_for_already_imported_cache(review)
          review.github_id
        end

        def each_object_to_import
          project.merge_requests.find_each do |merge_request|
            reviews = client.pull_request_reviews(project.import_source, merge_request.iid)
            reviews.each do |review|
              review.merge_request_id = merge_request.id
              yield(review)
            end
          end
        end
      end
    end
  end
end