Commit 320d6793 authored by Adam Hegyi's avatar Adam Hegyi Committed by Heinrich Lee Yu

Fix performance of GroupsWithTemplatesFinder query

- Introduce OptimizedGroupsWithTemplatesFinder that queries group
templates more efficiently.
- Introduce a feature flag to enable the optimization:
`optimized_groups_with_templates_finder`
parent a7813e74
...@@ -21,15 +21,30 @@ class GroupsWithTemplatesFinder ...@@ -21,15 +21,30 @@ class GroupsWithTemplatesFinder
attr_reader :group_id attr_reader :group_id
# Cleanup issue: https://gitlab.com/gitlab-org/gitlab/issues/35733
def extended_group_search def extended_group_search
if ::Feature.enabled?(:optimized_groups_with_templates_finder)
groups = Group.with_project_templates_optimized
groups_with_plan = Gitlab::ObjectHierarchy
.new(groups)
.base_and_ancestors
.with_feature_available_in_plan(:group_project_templates)
Gitlab::ObjectHierarchy.new(groups_with_plan).base_and_descendants
else
groups = Group.with_feature_available_in_plan(:group_project_templates) groups = Group.with_feature_available_in_plan(:group_project_templates)
Gitlab::ObjectHierarchy.new(groups).base_and_descendants Gitlab::ObjectHierarchy.new(groups).base_and_descendants
end end
end
def simple_group_search(groups) def simple_group_search(groups)
groups = group_id ? groups.find_by(id: group_id)&.self_and_ancestors : groups # rubocop: disable CodeReuse/ActiveRecord groups = group_id ? groups.find_by(id: group_id)&.self_and_ancestors : groups # rubocop: disable CodeReuse/ActiveRecord
return Group.none unless groups return Group.none unless groups
if ::Feature.enabled?(:optimized_groups_with_templates_finder)
groups.with_project_templates_optimized
else
groups.with_project_templates groups.with_project_templates
end end
end
end end
...@@ -62,6 +62,8 @@ module EE ...@@ -62,6 +62,8 @@ module EE
.distinct .distinct
end end
scope :with_project_templates_optimized, -> { where.not(custom_project_templates_group_id: nil) }
scope :with_custom_file_templates, -> do scope :with_custom_file_templates, -> do
preload( preload(
file_template_project: :route, file_template_project: :route,
......
...@@ -207,15 +207,25 @@ module EE ...@@ -207,15 +207,25 @@ module EE
end end
def available_subgroups_with_custom_project_templates(group_id = nil) def available_subgroups_with_custom_project_templates(group_id = nil)
groups = GroupsWithTemplatesFinder.new(group_id).execute found_groups = GroupsWithTemplatesFinder.new(group_id).execute
if ::Feature.enabled?(:optimized_groups_with_templates_finder)
GroupsFinder.new(self, min_access_level: ::Gitlab::Access::DEVELOPER) GroupsFinder.new(self, min_access_level: ::Gitlab::Access::DEVELOPER)
.execute .execute
.where(id: groups.select(:custom_project_templates_group_id)) .where(id: found_groups.select(:custom_project_templates_group_id))
.preload(:projects)
.joins(:projects)
.reorder(nil)
.distinct
else
GroupsFinder.new(self, min_access_level: ::Gitlab::Access::DEVELOPER)
.execute
.where(id: found_groups.select(:custom_project_templates_group_id))
.includes(:projects) .includes(:projects)
.reorder(nil) .reorder(nil)
.distinct .distinct
end end
end
def roadmap_layout def roadmap_layout
super || DEFAULT_ROADMAP_LAYOUT super || DEFAULT_ROADMAP_LAYOUT
......
---
title: Fix new project page load performance
merge_request: 18180
author:
type: performance
...@@ -269,7 +269,7 @@ describe 'New project' do ...@@ -269,7 +269,7 @@ describe 'New project' do
end end
end end
context 'when custom project group template is set' do shared_context 'when custom project group template is set' do
let(:group1) { create(:group) } let(:group1) { create(:group) }
let(:group2) { create(:group) } let(:group2) { create(:group) }
let(:group3) { create(:group) } let(:group3) { create(:group) }
...@@ -429,6 +429,23 @@ describe 'New project' do ...@@ -429,6 +429,23 @@ describe 'New project' do
end end
end end
# Cleanup issue: https://gitlab.com/gitlab-org/gitlab/issues/35733
context 'when `optimized_groups_with_templates_finder` feature flag is enabled' do
before do
stub_feature_flags(optimized_groups_with_templates_finder: true)
end
include_context 'when custom project group template is set'
end
context 'when `optimized_groups_with_templates_finder` feature flag is disabled' do
before do
stub_feature_flags(optimized_groups_with_templates_finder: false)
end
include_context 'when custom project group template is set'
end
context 'when group template is not set' do context 'when group template is not set' do
it_behaves_like 'group templates displayed' do it_behaves_like 'group templates displayed' do
let(:template_number) { 0 } let(:template_number) { 0 }
......
...@@ -26,6 +26,7 @@ describe GroupsWithTemplatesFinder do ...@@ -26,6 +26,7 @@ describe GroupsWithTemplatesFinder do
create(:gitlab_subscription, :silver, namespace: group_2) create(:gitlab_subscription, :silver, namespace: group_2)
end end
shared_examples 'GroupsWithTemplatesFinder examples' do
describe 'without group id' do describe 'without group id' do
it 'returns all groups' do it 'returns all groups' do
expect(described_class.new.execute).to contain_exactly(group_1, group_2, group_3) expect(described_class.new.execute).to contain_exactly(group_1, group_2, group_3)
...@@ -116,4 +117,21 @@ describe GroupsWithTemplatesFinder do ...@@ -116,4 +117,21 @@ describe GroupsWithTemplatesFinder do
end end
end end
end end
end
context 'when `optimized_groups_with_templates_finder` feature flag is enabled' do
before do
stub_feature_flags(optimized_groups_with_templates_finder: true)
end
include_examples 'GroupsWithTemplatesFinder examples'
end
context 'when `optimized_groups_with_templates_finder` feature flag is disabled' do
before do
stub_feature_flags(optimized_groups_with_templates_finder: false)
end
include_examples 'GroupsWithTemplatesFinder examples'
end
end end
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment