Add latest changes from gitlab-org/gitlab@13-11-stable-eev13.11.0-rc43

2 files changed, 59 insertions, 0 deletions

diff --git a/app/models/preloaders/labels_preloader.rb b/app/models/preloaders/labels_preloader.rb
new file mode 100644
index 00000000000..427f2869aac
--- /dev/null
+++ b/app/models/preloaders/labels_preloader.rb
@@ -0,0 +1,34 @@
+# frozen_string_literal: true
+
+module Preloaders
+  # This class preloads the `project`, `group`, and subscription associations for the given
+  # labels, user, and project (if provided). A Label can be of type ProjectLabel or GroupLabel
+  # and the preloader supports both.
+  #
+  # Usage:
+  #   labels = Label.where(...)
+  #   Preloaders::LabelsPreloader.new(labels, current_user, @project).preload_all
+  #   labels.first.project # won't fire any query
+  class LabelsPreloader
+    attr_reader :labels, :user, :project
+
+    def initialize(labels, user, project = nil)
+      @labels = labels
+      @user = user
+      @project = project
+    end
+
+    def preload_all
+      preloader = ActiveRecord::Associations::Preloader.new
+
+      preloader.preload(labels.select {|l| l.is_a? ProjectLabel }, { project: [:project_feature, namespace: :route] })
+      preloader.preload(labels.select {|l| l.is_a? GroupLabel }, { group: :route })
+      labels.each do |label|
+        label.lazy_subscription(user)
+        label.lazy_subscription(user, project) if project.present?
+      end
+    end
+  end
+end
+
+Preloaders::LabelsPreloader.prepend_if_ee('EE::Preloaders::LabelsPreloader')
diff --git a/app/models/preloaders/user_max_access_level_in_projects_preloader.rb b/app/models/preloaders/user_max_access_level_in_projects_preloader.rb
new file mode 100644
index 00000000000..671091480ee
--- /dev/null
+++ b/app/models/preloaders/user_max_access_level_in_projects_preloader.rb
@@ -0,0 +1,25 @@
+# frozen_string_literal: true
+
+module Preloaders
+  # This class preloads the max access level for the user within the given projects and
+  # stores the values in requests store via the ProjectTeam class.
+  class UserMaxAccessLevelInProjectsPreloader
+    def initialize(projects, user)
+      @projects = projects
+      @user = user
+    end
+
+    def execute
+      access_levels = @user
+        .project_authorizations
+        .where(project_id: @projects)
+        .group(:project_id)
+        .maximum(:access_level)
+
+      @projects.each do |project|
+        access_level = access_levels[project.id] || Gitlab::Access::NO_ACCESS
+        ProjectTeam.new(project).write_member_access_for_user_id(@user.id, access_level)
+      end
+    end
+  end
+end