Method: Aggrobot::Aggregator#collection

Defined in:
lib/aggrobot/aggregator.rb

#collection(values = nil) ⇒ Object

returns collection if it is ActiveRecord::Relation or ActiveRecord::Base raises error when collection is none of the above returns @collection otherwise (which is nil)



30
31
32
33
34
35
36
37
38
39
40
# File 'lib/aggrobot/aggregator.rb', line 30

def collection(values = nil)
  if values
    if !values.is_a?(ActiveRecord::Relation) && values < ActiveRecord::Base
      values = values.unscoped
    end
    raise_error 'Collection should be an ActiveRecord::Relation or ActiveRecord::Base' unless values.is_a?(ActiveRecord::Relation)
    @collection = values
  else
    @collection
  end
end