71 lines
2.1 KiB
Ruby
71 lines
2.1 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
module Gitlab
|
|
module GithubImport
|
|
module Importer
|
|
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
|
|
end
|
|
# rubocop: enable CodeReuse/ActiveRecord
|
|
|
|
def execute
|
|
# We insert records in bulk, by-passing any standard model callbacks.
|
|
# The pre_hook here makes sure we track internal ids consistently.
|
|
# Note this has to be called before performing an insert of a batch
|
|
# because we're outside a transaction scope here.
|
|
bulk_insert(Milestone, build_milestones, pre_hook: method(:track_greatest_iid))
|
|
build_milestones_cache
|
|
end
|
|
|
|
def track_greatest_iid(slice)
|
|
greatest_iid = slice.max { |e| e[:iid] }[:iid]
|
|
|
|
InternalId.track_greatest(nil, { project: project }, :milestones, greatest_iid, ->(_) { project.milestones.maximum(:iid) })
|
|
end
|
|
|
|
def build_milestones
|
|
build_database_rows(each_milestone)
|
|
end
|
|
|
|
def already_imported?(milestone)
|
|
existing_milestones.include?(milestone.number)
|
|
end
|
|
|
|
def build_milestones_cache
|
|
MilestoneFinder.new(project).build_cache
|
|
end
|
|
|
|
def build(milestone)
|
|
{
|
|
iid: milestone.number,
|
|
title: milestone.title,
|
|
description: milestone.description,
|
|
project_id: project.id,
|
|
state: state_for(milestone),
|
|
created_at: milestone.created_at,
|
|
updated_at: milestone.updated_at
|
|
}
|
|
end
|
|
|
|
def state_for(milestone)
|
|
milestone.state == 'open' ? :active : :closed
|
|
end
|
|
|
|
def each_milestone
|
|
client.milestones(project.import_source, state: 'all')
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|