debian-mirror-gitlab/app/models/preloaders/commit_status_preloader.rb

# frozen_string_literal: true

module Preloaders
  class CommitStatusPreloader
    CLASSES = [::Ci::Build, ::Ci::Bridge, ::GenericCommitStatus].freeze

    def initialize(statuses)
      @statuses = statuses
    end

    def execute(relations)
      CLASSES.each do |klass|
        ActiveRecord::Associations::Preloader.new(
          records: objects(klass),
          associations: associations(klass, relations)
        ).call
      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