class GraphQL::Dataloader::ActiveRecordSource

# File lib/graphql/dataloader/active_record_source.rb, line 8
def initialize(model_class, find_by: model_class.primary_key)
  @model_class = model_class
  @find_by = find_by
  @find_by_many = find_by.is_a?(Array)
  if @find_by_many
    @type_for_column = @find_by.map { |fb| @model_class.type_for_attribute(fb) }
  else
    @type_for_column = @model_class.type_for_attribute(@find_by)
  end
end

# File lib/graphql/dataloader/active_record_source.rb, line 33
def fetch(record_ids)
  records = @model_class.where(@find_by => record_ids)
  record_lookup = {}
  if @find_by_many
    records.each do |r|
      key = @find_by.map { |fb| r.public_send(fb) }
      record_lookup[key] = r
    end
  else
    records.each { |r| record_lookup[r.public_send(@find_by)] = r }
  end
  record_ids.map { |id| record_lookup[id] }
end

# File lib/graphql/dataloader/active_record_source.rb, line 23
def normalize_fetch_key(requested_key)
  if @find_by_many
    requested_key.each_with_index.map do |k, idx|
      @type_for_column[idx].cast(k)
    end
  else
    @type_for_column.cast(requested_key)
  end
end

# File lib/graphql/dataloader/active_record_source.rb, line 19
def result_key_for(requested_key)
  normalize_fetch_key(requested_key)
end