module BatchLoaderActiveRecord::ClassMethods

Public Instance Methods

association_accessor(name) click to toggle source
# File lib/batch_loader_active_record.rb, line 13
def association_accessor(name)
  reflection = reflect_on_association(name) or raise "Can't find association #{name.inspect}"
  manager = AssociationManager.new(model: self, reflection: reflection)
  case reflection.macro
  when :belongs_to
    if reflection.polymorphic?
      define_method(manager.accessor_name) { manager.polymorphic_belongs_to_batch_loader(self) }
    else
      define_method(manager.accessor_name) { manager.belongs_to_batch_loader(self) }
    end
  when :has_one
    define_method(manager.accessor_name) { manager.has_one_to_batch_loader(self) }
  when :has_many
    define_method(manager.accessor_name) do |instance_scope = nil|
      manager.has_many_to_batch_loader(self, instance_scope)
    end
  when :has_and_belongs_to_many
    define_method(manager.accessor_name) do |instance_scope = nil|
      manager.has_and_belongs_to_many_to_batch_loader(self, instance_scope)
    end
  else
    raise NotImplementedError, "association kind #{reflection.macro.inspect} is not yet supported"
  end
end
belongs_to_lazy(*args) click to toggle source
# File lib/batch_loader_active_record.rb, line 38
def belongs_to_lazy(*args)
  belongs_to(*args).tap do
    reflection = reflect_on_all_associations.last
    manager = AssociationManager.new(model: self, reflection: reflection)
    if reflection.polymorphic?
      define_method(manager.accessor_name) { manager.polymorphic_belongs_to_batch_loader(self) }
    else
      define_method(manager.accessor_name) { manager.belongs_to_batch_loader(self) }
    end
  end
end
has_and_belongs_to_many_lazy(*args) click to toggle source
# File lib/batch_loader_active_record.rb, line 66
def has_and_belongs_to_many_lazy(*args)
  has_and_belongs_to_many(*args).tap do
    manager = AssociationManager.new(model: self, reflection: reflect_on_all_associations.last)
    define_method(manager.accessor_name) do |instance_scope = nil|
      manager.has_and_belongs_to_many_to_batch_loader(self, instance_scope)
    end
  end
end
has_many_lazy(*args) click to toggle source
# File lib/batch_loader_active_record.rb, line 57
def has_many_lazy(*args)
  has_many(*args).tap do
    manager = AssociationManager.new(model: self, reflection: reflect_on_all_associations.last)
    define_method(manager.accessor_name) do |instance_scope = nil|
      manager.has_many_to_batch_loader(self, instance_scope)
    end
  end
end
has_one_lazy(*args) click to toggle source
# File lib/batch_loader_active_record.rb, line 50
def has_one_lazy(*args)
  has_one(*args).tap do
    manager = AssociationManager.new(model: self, reflection: reflect_on_all_associations.last)
    define_method(manager.accessor_name) { manager.has_one_to_batch_loader(self) }
  end
end