refactor(dossier loader): simplify all the things

This commit is contained in:
Paul Chavard 2024-08-22 17:40:36 +02:00
parent 2ba356bc68
commit 33e502234e
No known key found for this signature in database

View file

@ -37,18 +37,12 @@ class DossierPreloader
private private
def revisions def revisions(pj_template: false)
@revisions ||= ProcedureRevision.where(id: @dossiers.pluck(:revision_id).uniq) @revisions ||= ProcedureRevision.where(id: @dossiers.pluck(:revision_id).uniq)
.includes(types_de_champ: { piece_justificative_template_attachment: :blob }) .includes(types_de_champ: pj_template ? { piece_justificative_template_attachment: :blob } : [])
.index_by(&:id) .index_by(&:id)
end end
# returns: { revision_id : { stable_id : position } }
def positions
@positions ||= revisions
.transform_values { |revision| revision.revision_types_de_champ.map { [_1.stable_id, _1.position] }.to_h }
end
def load_dossiers(dossiers, pj_template: false) def load_dossiers(dossiers, pj_template: false)
to_include = @includes_for_champ.dup to_include = @includes_for_champ.dup
to_include << [piece_justificative_file_attachments: :blob] to_include << [piece_justificative_file_attachments: :blob]
@ -58,16 +52,10 @@ class DossierPreloader
.where(dossier_id: dossiers) .where(dossier_id: dossiers)
.to_a .to_a
children_champs, root_champs = all_champs.partition(&:child?) champs_by_dossier = all_champs.group_by(&:dossier_id)
champs_by_dossier = root_champs.group_by(&:dossier_id)
champs_by_dossier_by_parent = children_champs
.group_by(&:dossier_id)
.transform_values do |champs|
champs.group_by(&:parent_id)
end
dossiers.each do |dossier| dossiers.each do |dossier|
load_dossier(dossier, champs_by_dossier[dossier.id] || [], champs_by_dossier_by_parent[dossier.id] || {}) load_dossier(dossier, champs_by_dossier[dossier.id] || [], pj_template:)
end end
load_etablissements(all_champs) load_etablissements(all_champs)
@ -86,17 +74,30 @@ class DossierPreloader
end end
end end
def load_dossier(dossier, champs, children_by_parent = {}) def load_dossier(dossier, champs, pj_template: false)
revision = revisions[dossier.revision_id] revision = revisions(pj_template:)[dossier.revision_id]
if revision.present? if revision.present?
dossier.association(:revision).target = revision dossier.association(:revision).target = revision
end end
dossier.association(:champs).target = champs
dossier.association(:champs_public).target = dossier.champs_for_revision(scope: :public, root: true)
dossier.association(:champs_private).target = dossier.champs_for_revision(scope: :private, root: true)
champs_public, champs_private = champs.partition(&:public?) # remove once parent_id is deprecated
champs_by_parent_id = champs.group_by(&:parent_id)
dossier.association(:champs).target = [] champs.each do |champ|
load_champs(dossier, :champs_public, champs_public, dossier, children_by_parent) champ.association(:dossier).target = dossier
load_champs(dossier, :champs_private, champs_private, dossier, children_by_parent)
# remove once parent_id is deprecated
if champ.repetition?
children = champs_by_parent_id.fetch(champ.id, [])
children.each do |child|
child.association(:parent).target = champ
end
champ.association(:champs).target = children
end
end
# We need to do this because of the check on `Etablissement#champ` in # We need to do this because of the check on `Etablissement#champ` in
# `Etablissement#libelle_for_export`. By assigning `nil` to `target` we mark association # `Etablissement#libelle_for_export`. By assigning `nil` to `target` we mark association
@ -105,33 +106,4 @@ class DossierPreloader
dossier.etablissement.association(:champ).target = nil dossier.etablissement.association(:champ).target = nil
end end
end end
def load_champs(parent, name, champs, dossier, children_by_parent)
if champs.empty?
parent.association(name).target = [] # tells to Rails association has been loaded
return
end
champs.each do |champ|
champ.association(:dossier).target = dossier
if parent.is_a?(Champ)
champ.association(:parent).target = parent
end
end
dossier.association(:champs).target += champs
parent.association(name).target = champs
.filter { positions[dossier.revision_id][_1.stable_id].present? }
.sort_by { [_1.row_id, positions[dossier.revision_id][_1.stable_id]] }
# Load children champs
champs.filter(&:block?).each do |parent_champ|
champs = children_by_parent[parent_champ.id] || []
parent_champ.association(:dossier).target = dossier
load_champs(parent_champ, :champs, champs, dossier, children_by_parent)
end
end
end end