Welcome to mirror list, hosted at ThFree Co, Russian Federation.

group_root_ancestor_preloader.rb « preloaders « models « app - gitlab.com/gitlab-org/gitlab-foss.git - Unnamed repository; edit this file 'description' to name the repository.
summaryrefslogtreecommitdiff
blob: 29c60e90964d1118d43ad38570a38a0ff8dafac1 (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
# frozen_string_literal: true

module Preloaders
  class GroupRootAncestorPreloader
    def initialize(groups, root_ancestor_preloads = [])
      @groups = groups
      @root_ancestor_preloads = root_ancestor_preloads
    end

    def execute
      return unless ::Feature.enabled?(:use_traversal_ids)

      # type == 'Group' condition located on subquery to prevent a filter in the query
      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)

      @groups.each do |group|
        group.root_ancestor = root_ancestors_by_id[group.id].first
      end
    end

    private

    def join_sql
      Group.select('id, traversal_ids[1] as root_id').where(id: @groups.map(&:id)).to_sql
    end
  end
end