blob: 1e935249407be10c06712385d8cb0e06e42329ec (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
|
# frozen_string_literal: true
module Preloaders
class ProjectRootAncestorPreloader
def initialize(projects, namespace_sti_name = :namespace, root_ancestor_preloads = [])
@projects = projects
@namespace_sti_name = namespace_sti_name
@root_ancestor_preloads = root_ancestor_preloads
end
def execute
return if @projects.is_a?(ActiveRecord::NullRelation)
return unless ::Feature.enabled?(:use_traversal_ids)
root_query = Namespace.joins("INNER JOIN (#{join_sql}) as root_query ON root_query.root_id = namespaces.id")
.select('namespaces.*, root_query.id as source_id')
root_query = root_query.preload(*@root_ancestor_preloads) if @root_ancestor_preloads.any?
root_ancestors_by_id = root_query.group_by(&:source_id)
ActiveRecord::Associations::Preloader.new.preload(@projects, :namespace)
@projects.each do |project|
root_ancestor = root_ancestors_by_id[project.id]&.first
project.namespace.root_ancestor = root_ancestor if root_ancestor.present?
end
end
private
def join_sql
@projects
.joins(@namespace_sti_name)
.select('projects.id, namespaces.traversal_ids[1] as root_id')
.to_sql
end
end
end
|