diff options
Diffstat (limited to 'lib/gitlab/github_import/importer/milestones_importer.rb')
-rw-r--r-- | lib/gitlab/github_import/importer/milestones_importer.rb | 14 |
1 files changed, 6 insertions, 8 deletions
diff --git a/lib/gitlab/github_import/importer/milestones_importer.rb b/lib/gitlab/github_import/importer/milestones_importer.rb index 71ff7465d9b..d11b151bbe2 100644 --- a/lib/gitlab/github_import/importer/milestones_importer.rb +++ b/lib/gitlab/github_import/importer/milestones_importer.rb @@ -6,15 +6,9 @@ module Gitlab class MilestonesImporter include BulkImporting - attr_reader :project, :client, :existing_milestones - - # project - An instance of `Project` - # client - An instance of `Gitlab::GithubImport::Client` # rubocop: disable CodeReuse/ActiveRecord - def initialize(project, client) - @project = project - @client = client - @existing_milestones = project.milestones.pluck(:iid).to_set + def existing_milestones + @existing_milestones ||= project.milestones.pluck(:iid).to_set end # rubocop: enable CodeReuse/ActiveRecord @@ -55,6 +49,10 @@ module Gitlab def each_milestone client.milestones(project.import_source, state: 'all') end + + def object_type + :milestone + end end end end |