28 lines
543 B
Ruby
28 lines
543 B
Ruby
|
require 'banzai'
|
||
|
|
|
||
|
|
module Banzai
|
||
|
|
class LazyReference
|
||
|
|
def self.load(refs)
|
||
|
|
lazy_references, values = refs.partition { |ref| ref.is_a?(self) }
|
||
|
|
|
||
|
|
lazy_values = lazy_references.group_by(&:klass).flat_map do |klass, refs|
|
||
|
|
ids = refs.flat_map(&:ids)
|
||
|
|
klass.where(id: ids)
|
||
|
|
end
|
||
|
|
|
||
|
|
values + lazy_values
|
||
|
|
end
|
||
|
|
|
||
|
|
attr_reader :klass, :ids
|
||
|
|
|
||
|
|
def initialize(klass, ids)
|
||
|
|
@klass = klass
|
||
|
|
@ids = Array.wrap(ids).map(&:to_i)
|
||
|
|
end
|
||
|
|
|
||
|
|
def load
|
||
|
|
self.klass.where(id: self.ids)
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|