File: merge_request_diff_preloader.rb

package info (click to toggle)
gitlab 17.6.5-19
  • links: PTS, VCS
  • area: main
  • in suites: sid
  • size: 629,368 kB
  • sloc: ruby: 1,915,304; javascript: 557,307; sql: 60,639; xml: 6,509; sh: 4,567; makefile: 1,239; python: 406
file content (27 lines) | stat: -rw-r--r-- 934 bytes parent folder | download
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
# frozen_string_literal: true

module Preloaders
  # This class preloads the `merge_request_diff` association for the given merge request models.
  #
  # Usage:
  #   merge_requests = MergeRequest.where(...)
  #   Preloaders::MergeRequestDiffPreloader.new(merge_requests).preload_all
  #   merge_requests.first.merge_request_diff # won't fire any query
  class MergeRequestDiffPreloader
    def initialize(merge_requests)
      @merge_requests = merge_requests
    end

    def preload_all
      merge_request_diffs = MergeRequestDiff.latest_diff_for_merge_requests(@merge_requests)
      cache = merge_request_diffs.index_by { |diff| diff.merge_request_id }

      @merge_requests.each do |merge_request|
        merge_request_diff = cache[merge_request.id]

        merge_request.association(:merge_request_diff).target = merge_request_diff
        merge_request.association(:merge_request_diff).loaded!
      end
    end
  end
end