debian-mirror-gitlab/app/workers/issuable_export_csv_worker.rb

55 lines
1.5 KiB
Ruby
Raw Normal View History

2021-01-03 14:25:43 +05:30
# frozen_string_literal: true
class IssuableExportCsvWorker # rubocop:disable Scalability/IdempotentWorker
include ApplicationWorker
2021-10-27 15:23:28 +05:30
data_consistency :always
2021-06-08 01:23:25 +05:30
sidekiq_options retry: 3
2021-12-11 22:18:48 +05:30
feature_category :team_planning
2021-01-03 14:25:43 +05:30
worker_resource_boundary :cpu
loggable_arguments 2
def perform(type, current_user_id, project_id, params)
2021-03-08 18:12:59 +05:30
user = User.find(current_user_id)
project = Project.find(project_id)
2021-01-03 14:25:43 +05:30
2021-03-11 19:13:27 +05:30
export_service(type, user, project, params).email(user)
2021-03-08 18:12:59 +05:30
rescue ActiveRecord::RecordNotFound => error
logger.error("Failed to export CSV (current_user_id:#{current_user_id}, project_id:#{project_id}): #{error.message}")
2021-01-03 14:25:43 +05:30
end
private
2021-03-08 18:12:59 +05:30
def export_service(type, user, project, params)
2021-03-11 19:13:27 +05:30
issuable_classes = issuable_classes_for(type.to_sym)
issuables = issuable_classes[:finder].new(user, parse_params(params, project.id)).execute
issuable_classes[:service].new(issuables, project)
2021-03-08 18:12:59 +05:30
end
2021-03-11 19:13:27 +05:30
def issuable_classes_for(type)
2021-03-08 18:12:59 +05:30
case type
2021-01-03 14:25:43 +05:30
when :issue
2021-03-08 18:12:59 +05:30
{ finder: IssuesFinder, service: Issues::ExportCsvService }
2021-01-03 14:25:43 +05:30
when :merge_request
2021-03-08 18:12:59 +05:30
{ finder: MergeRequestsFinder, service: MergeRequests::ExportCsvService }
else
raise ArgumentError, type_error_message(type)
2021-01-03 14:25:43 +05:30
end
end
2021-03-11 19:13:27 +05:30
def parse_params(params, project_id)
params
2022-01-26 12:08:38 +05:30
.with_indifferent_access
2021-03-11 19:13:27 +05:30
.except(:sort)
.merge(project_id: project_id)
end
2021-03-08 18:12:59 +05:30
def type_error_message(type)
"Type parameter must be :issue or :merge_request, it was #{type}"
2021-01-03 14:25:43 +05:30
end
end
2021-03-08 18:12:59 +05:30
2021-06-08 01:23:25 +05:30
IssuableExportCsvWorker.prepend_mod_with('IssuableExportCsvWorker')