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

loaded_in_group_list.rb « concerns « models « app - gitlab.com/gitlab-org/gitlab-foss.git - Unnamed repository; edit this file 'description' to name the repository.
summaryrefslogtreecommitdiff
blob: 59e0ed75d2d88af9a927fd68ad16093c1c8d48be (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
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
# frozen_string_literal: true

module LoadedInGroupList
  extend ActiveSupport::Concern

  class_methods do
    def with_counts(archived:)
      selects_including_counts = [
        'namespaces.*',
        "(#{project_count_sql(archived).to_sql}) AS preloaded_project_count",
        "(#{member_count_sql.to_sql}) AS preloaded_member_count",
        "(#{subgroup_count_sql.to_sql}) AS preloaded_subgroup_count"
      ]

      select(selects_including_counts)
    end

    def with_selects_for_list(archived: nil)
      with_route.with_counts(archived: archived)
    end

    private

    def project_count_sql(archived = nil)
      projects = Project.arel_table
      namespaces = Namespace.arel_table

      base_count = projects.project(Arel.star.count.as('preloaded_project_count'))
                     .where(projects[:namespace_id].eq(namespaces[:id]))

      if archived == 'only'
        base_count.where(projects[:archived].eq(true))
      elsif Gitlab::Utils.to_boolean(archived)
        base_count
      else
        base_count.where(projects[:archived].not_eq(true))
      end
    end

    def subgroup_count_sql
      namespaces = Namespace.arel_table
      children = namespaces.alias('children')

      namespaces.project(Arel.star.count.as('preloaded_subgroup_count'))
        .from(children)
        .where(children[:parent_id].eq(namespaces[:id]))
    end

    def member_count_sql
      members = Member.arel_table
      namespaces = Namespace.arel_table

      members.project(Arel.star.count.as('preloaded_member_count'))
        .where(members[:source_type].eq(Namespace.name))
        .where(members[:source_id].eq(namespaces[:id]))
        .where(members[:requested_at].eq(nil))
        .where(members[:access_level].gt(Gitlab::Access::MINIMAL_ACCESS))
    end
  end

  def children_count
    @children_count ||= project_count + subgroup_count
  end

  def project_count
    @project_count ||= try(:preloaded_project_count) || projects.non_archived.count
  end

  def subgroup_count
    @subgroup_count ||= try(:preloaded_subgroup_count) || children.count
  end

  def member_count
    @member_count ||= try(:preloaded_member_count) || members.count
  end

  def guest_count
    @guest_count ||= members.guests.count
  end
end

LoadedInGroupList::ClassMethods.prepend_if_ee('EE::LoadedInGroupList::ClassMethods')