diff options
Diffstat (limited to 'app/services/export_csv/base_service.rb')
-rw-r--r-- | app/services/export_csv/base_service.rb | 56 |
1 files changed, 56 insertions, 0 deletions
diff --git a/app/services/export_csv/base_service.rb b/app/services/export_csv/base_service.rb new file mode 100644 index 00000000000..84d44fd75fc --- /dev/null +++ b/app/services/export_csv/base_service.rb @@ -0,0 +1,56 @@ +# frozen_string_literal: true + +module ExportCsv + class BaseService + # Target attachment size before base64 encoding + TARGET_FILESIZE = 15.megabytes + + def initialize(relation, resource_parent, fields = []) + @objects = relation + @resource_parent = resource_parent + @fields = fields + end + + def csv_data + csv_builder.render(TARGET_FILESIZE) + end + + def email(user) + raise NotImplementedError + end + + def invalid_fields + ::ExportCsv::MapExportFieldsService.new(fields, header_to_value_hash).invalid_fields + end + + private + + attr_reader :resource_parent, :objects, :fields + + # rubocop: disable CodeReuse/ActiveRecord + def csv_builder + @csv_builder ||= begin + data_hash = MapExportFieldsService.new(fields, header_to_value_hash).execute + + if preload_associations_in_batches? + CsvBuilder.new(objects, data_hash, associations_to_preload) + else + CsvBuilder.new(objects.preload(associations_to_preload), data_hash, []) + end + end + end + # rubocop: enable CodeReuse/ActiveRecord + + def associations_to_preload + [] + end + + def header_to_value_hash + raise NotImplementedError + end + + def preload_associations_in_batches? + false + end + end +end |