Module: Mongoload::ContextWithAutoInclude

Defined in:
lib/mongoload/patches/contextual_mongo.rb

Instance Method Summary collapse

Instance Method Details

#documents_for_iteration ⇒ Object 



4
5
6
7
8
9
10
11
12
 
# File 'lib/mongoload/patches/contextual_mongo.rb', line 4

def documents_for_iteration
  results = super
  if results.is_a?(Mongo::Collection::View)
    results = results.map { |doc| Mongoid::Factory.from_db(klass, doc, criteria.options[:fields]) }
  end
  results.tap do |documents|
    Mongoload::AutoIncludeContext.register_docs(documents.to_a)
  end
end