54 lines
1.3 KiB
Ruby
54 lines
1.3 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
module BulkImports
|
|
class ExportStatus
|
|
include Gitlab::Utils::StrongMemoize
|
|
|
|
def initialize(pipeline_tracker, relation)
|
|
@pipeline_tracker = pipeline_tracker
|
|
@relation = relation
|
|
@entity = @pipeline_tracker.entity
|
|
@configuration = @entity.bulk_import.configuration
|
|
@client = Clients::HTTP.new(url: @configuration.url, token: @configuration.access_token)
|
|
end
|
|
|
|
def started?
|
|
export_status['status'] == Export::STARTED
|
|
end
|
|
|
|
def failed?
|
|
export_status['status'] == Export::FAILED
|
|
end
|
|
|
|
def error
|
|
export_status['error']
|
|
end
|
|
|
|
private
|
|
|
|
attr_reader :client, :entity, :relation
|
|
|
|
def export_status
|
|
strong_memoize(:export_status) do
|
|
status = fetch_export_status
|
|
|
|
relation_export_status = status&.find { |item| item['relation'] == relation }
|
|
|
|
# Consider empty response as failed export
|
|
raise StandardError, 'Empty relation export status' unless relation_export_status&.present?
|
|
|
|
relation_export_status
|
|
end
|
|
rescue StandardError => e
|
|
{ 'status' => Export::FAILED, 'error' => e.message }
|
|
end
|
|
|
|
def fetch_export_status
|
|
client.get(status_endpoint).parsed_response
|
|
end
|
|
|
|
def status_endpoint
|
|
File.join(entity.export_relations_url_path, 'status')
|
|
end
|
|
end
|
|
end
|