2018-11-27 15:10:40 +00:00
|
|
|
# frozen_string_literal: true
|
|
|
|
|
|
|
|
module Gitlab
|
|
|
|
module Graphql
|
|
|
|
module Loaders
|
|
|
|
class BatchModelLoader
|
2022-05-30 12:08:23 +00:00
|
|
|
attr_reader :model_class, :model_id, :preloads
|
2018-11-27 15:10:40 +00:00
|
|
|
|
2022-05-30 12:08:23 +00:00
|
|
|
def initialize(model_class, model_id, preloads = nil)
|
2021-04-19 21:09:27 +00:00
|
|
|
@model_class = model_class
|
|
|
|
@model_id = model_id
|
2022-05-30 12:08:23 +00:00
|
|
|
@preloads = preloads || []
|
2018-11-27 15:10:40 +00:00
|
|
|
end
|
|
|
|
|
|
|
|
# rubocop: disable CodeReuse/ActiveRecord
|
|
|
|
def find
|
2022-05-30 12:08:23 +00:00
|
|
|
BatchLoader::GraphQL.for([model_id.to_i, preloads]).batch(key: model_class) do |for_params, loader, args|
|
2020-11-06 18:09:07 +00:00
|
|
|
model = args[:key]
|
2022-05-30 12:08:23 +00:00
|
|
|
keys_by_id = for_params.group_by(&:first)
|
|
|
|
ids = for_params.map(&:first)
|
|
|
|
preloads = for_params.flat_map(&:second).uniq
|
2020-11-06 18:09:07 +00:00
|
|
|
results = model.where(id: ids)
|
2022-05-30 12:08:23 +00:00
|
|
|
results = results.preload(*preloads) unless preloads.empty?
|
2018-11-27 15:10:40 +00:00
|
|
|
|
2022-05-30 12:08:23 +00:00
|
|
|
results.each do |record|
|
|
|
|
keys_by_id.fetch(record.id, []).each { |k| loader.call(k, record) }
|
|
|
|
end
|
2018-11-27 15:10:40 +00:00
|
|
|
end
|
|
|
|
end
|
|
|
|
# rubocop: enable CodeReuse/ActiveRecord
|
|
|
|
end
|
|
|
|
end
|
|
|
|
end
|
|
|
|
end
|