Add latest changes from gitlab-org/gitlab@14-3-stable-eev14.3.0-rc42

1 files changed, 29 insertions, 0 deletions

diff --git a/app/models/preloaders/commit_status_preloader.rb b/app/models/preloaders/commit_status_preloader.rb
new file mode 100644
index 00000000000..535dd24ba6b
--- /dev/null
+++ b/app/models/preloaders/commit_status_preloader.rb
@@ -0,0 +1,29 @@
+# frozen_string_literal: true
+
+module Preloaders
+  class CommitStatusPreloader
+    CLASSES = [::Ci::Build, ::Ci::Bridge, ::GenericCommitStatus].freeze
+
+    def initialize(statuses)
+      @statuses = statuses
+    end
+
+    def execute(relations)
+      preloader = ActiveRecord::Associations::Preloader.new
+
+      CLASSES.each do |klass|
+        preloader.preload(objects(klass), associations(klass, relations))
+      end
+    end
+
+    private
+
+    def objects(klass)
+      @statuses.select { |job| job.is_a?(klass) }
+    end
+
+    def associations(klass, relations)
+      klass.reflections.keys.map(&:to_sym) & relations.map(&:to_sym)
+    end
+  end
+end