30 lines
849 B
Ruby
30 lines
849 B
Ruby
|
# frozen_string_literal: true
|
||
|
|
|
||
|
|
module Gitlab
|
||
|
|
module Graphql
|
||
|
|
module Loaders
|
||
|
|
class BatchModelLoader
|
||
|
|
attr_reader :model_class, :model_id
|
||
|
|
|
||
|
|
def initialize(model_class, model_id)
|
||
|
|
@model_class, @model_id = model_class, model_id
|
||
|
|
end
|
||
|
|
|
||
|
|
# rubocop: disable CodeReuse/ActiveRecord
|
||
|
|
def find
|
||
|
|
BatchLoader.for({ model: model_class, id: model_id }).batch do |loader_info, loader|
|
||
|
|
per_model = loader_info.group_by { |info| info[:model] }
|
||
|
|
per_model.each do |model, info|
|
||
|
|
ids = info.map { |i| i[:id] }
|
||
|
|
results = model.where(id: ids)
|
||
|
|
|
||
|
|
results.each { |record| loader.call({ model: model, id: record.id }, record) }
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|
||
|
|
# rubocop: enable CodeReuse/ActiveRecord
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|