summaryrefslogtreecommitdiff
path: root/app/services/export_csv/base_service.rb
diff options
context:
space:
mode:
Diffstat (limited to 'app/services/export_csv/base_service.rb')
-rw-r--r--app/services/export_csv/base_service.rb56
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