module Bullet::Mongoid

Public Class Methods

enable() click to toggle source
# File lib/bullet/mongoid4x.rb, line 5
def self.enable
  require 'mongoid'
  ::Mongoid::Contextual::Mongo.class_eval do
    alias_method :origin_first, :first
    alias_method :origin_last, :last
    alias_method :origin_each, :each
    alias_method :origin_eager_load, :eager_load

    def first
      result = origin_first
      Bullet::Detector::NPlusOneQuery.add_impossible_object(result) if result
      result
    end

    def last
      result = origin_last
      Bullet::Detector::NPlusOneQuery.add_impossible_object(result) if result
      result
    end

    def each(&block)
      return to_enum unless block

      records = []
      origin_each { |record| records << record }
      if records.length > 1
        Bullet::Detector::NPlusOneQuery.add_possible_objects(records)
      elsif records.size == 1
        Bullet::Detector::NPlusOneQuery.add_impossible_object(records.first)
      end
      records.each(&block)
    end

    def eager_load(docs)
      associations = criteria.inclusions.map(&:name)
      docs.each { |doc| Bullet::Detector::NPlusOneQuery.add_object_associations(doc, associations) }
      Bullet::Detector::UnusedEagerLoading.add_eager_loadings(docs, associations)
      origin_eager_load(docs)
    end
  end

  ::Mongoid::Relations::Accessors.class_eval do
    alias_method :origin_get_relation, :get_relation

    def get_relation(name, metadata, object, reload = false)
      result = origin_get_relation(name, metadata, object, reload)
      Bullet::Detector::NPlusOneQuery.call_association(self, name) if metadata.macro !~ /embed/
      result
    end
  end
end

Public Instance Methods

each(&block) click to toggle source
# File lib/bullet/mongoid4x.rb, line 25
def each(&block)
  return to_enum unless block

  records = []
  origin_each { |record| records << record }
  if records.length > 1
    Bullet::Detector::NPlusOneQuery.add_possible_objects(records)
  elsif records.size == 1
    Bullet::Detector::NPlusOneQuery.add_impossible_object(records.first)
  end
  records.each(&block)
end
eager_load(docs) click to toggle source
# File lib/bullet/mongoid4x.rb, line 38
def eager_load(docs)
  associations = criteria.inclusions.map(&:name)
  docs.each { |doc| Bullet::Detector::NPlusOneQuery.add_object_associations(doc, associations) }
  Bullet::Detector::UnusedEagerLoading.add_eager_loadings(docs, associations)
  origin_eager_load(docs)
end
first() click to toggle source
# File lib/bullet/mongoid4x.rb, line 13
def first
  result = origin_first
  Bullet::Detector::NPlusOneQuery.add_impossible_object(result) if result
  result
end
get_relation(name, metadata, object, reload = false) click to toggle source
# File lib/bullet/mongoid4x.rb, line 49
def get_relation(name, metadata, object, reload = false)
  result = origin_get_relation(name, metadata, object, reload)
  Bullet::Detector::NPlusOneQuery.call_association(self, name) if metadata.macro !~ /embed/
  result
end
last() click to toggle source
# File lib/bullet/mongoid4x.rb, line 19
def last
  result = origin_last
  Bullet::Detector::NPlusOneQuery.add_impossible_object(result) if result
  result
end