Module: ActiveRecord::Calculations

Included in:
Relation
Defined in:
lib/active_record/relation/calculations.rb

Instance Method Summary collapse

Instance Method Details

#average(column_name, options = {}) ⇒ Object

Calculates the average value on a given column. Returns nil if there’s no row. See calculate for examples with options.

Person.average('age') # => 35.8


65
66
67
# File 'lib/active_record/relation/calculations.rb', line 65

def average(column_name, options = {})
  calculate(:average, column_name, options)
end

#calculate(operation, column_name, options = {}) ⇒ Object

This calculates aggregate values in the given column. Methods for count, sum, average, minimum, and maximum have been added as shortcuts. Options such as :conditions, :order, :group, :having, and :joins can be passed to customize the query.

There are two basic forms of output:

* Single aggregate value: The single value is type cast to Fixnum for COUNT, Float
  for AVG, and the given column's type for everything else.
* Grouped values: This returns an ordered hash of the values and groups them by the
  <tt>:group</tt> option. It takes either a column name, or the name of a belongs_to association.

    values = Person.maximum(:age, :group => 'last_name')
    puts values["Drake"]
    => 43

    drake  = Family.find_by_last_name('Drake')
    values = Person.maximum(:age, :group => :family) # Person belongs_to :family
    puts values[drake]
    => 43

    values.each do |family, max_age|
    ...
    end

Options:

  • :conditions - An SQL fragment like “administrator = 1” or [ “user_name = ?”, username ]. See conditions in the intro to ActiveRecord::Base.

  • :include: Eager loading, see Associations for details. Since calculations don’t load anything, the purpose of this is to access fields on joined tables in your conditions, order, or group clauses.

  • :joins - An SQL fragment for additional joins like “LEFT JOIN comments ON comments.post_id = id”. (Rarely needed). The records will be returned read-only since they will have attributes that do not correspond to the table’s columns.

  • :order - An SQL fragment like “created_at DESC, name” (really only used with GROUP BY calculations).

  • :group - An attribute name by which the result should be grouped. Uses the GROUP BY SQL-clause.

  • :select - By default, this is * as in SELECT * FROM, but can be changed if you for example want to do a join, but not include the joined columns.

  • :distinct - Set this to true to make this a distinct calculation, such as SELECT COUNT(DISTINCT posts.id) …

Examples:

Person.calculate(:count, :all) # The same as Person.count
Person.average(:age) # SELECT AVG(age) FROM people...
Person.minimum(:age, :conditions => ['last_name != ?', 'Drake']) # Selects the minimum age for
                                                                 # everyone with a last name other than 'Drake'

# Selects the minimum age for any family without any minors
Person.minimum(:age, :having => 'min(age) > 17', :group => :last_name)

Person.sum("2 * age")


149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
# File 'lib/active_record/relation/calculations.rb', line 149

def calculate(operation, column_name, options = {})
  if options.except(:distinct).present?
    apply_finder_options(options.except(:distinct)).calculate(operation, column_name, :distinct => options[:distinct])
  else
    relation = with_default_scope

    if relation.equal?(self)
      if eager_loading? || (includes_values.present? && references_eager_loaded_tables?)
        construct_relation_for_association_calculations.calculate(operation, column_name, options)
      else
        perform_calculation(operation, column_name, options)
      end
    else
      relation.calculate(operation, column_name, options)
    end
  end
rescue ThrowResult
  0
end

#count(column_name = nil, options = {}) ⇒ Object

Count operates using three different approaches.

  • Count all: By not passing any parameters to count, it will return a count of all the rows for the model.

  • Count using column: By passing a column name to count, it will return a count of all the rows for the model with supplied column present.

  • Count using options will find the row count matched by the options used.

The third approach, count using options, accepts an option hash as the only parameter. The options are:

  • :conditions: An SQL fragment like “administrator = 1” or [ “user_name = ?”, username ]. See conditions in the intro to ActiveRecord::Base.

  • :joins: Either an SQL fragment for additional joins like “LEFT JOIN comments ON comments.post_id = id” (rarely needed) or named associations in the same form used for the :include option, which will perform an INNER JOIN on the associated table(s). If the value is a string, then the records will be returned read-only since they will have attributes that do not correspond to the table’s columns. Pass :readonly => false to override.

  • :include: Named associations that should be loaded alongside using LEFT OUTER JOINs. The symbols named refer to already defined associations. When using named associations, count returns the number of DISTINCT items for the model you’re counting. See eager loading under Associations.

  • :order: An SQL fragment like “created_at DESC, name” (really only used with GROUP BY calculations).

  • :group: An attribute name by which the result should be grouped. Uses the GROUP BY SQL-clause.

  • :select: By default, this is * as in SELECT * FROM, but can be changed if you, for example, want to do a join but not include the joined columns.

  • :distinct: Set this to true to make this a distinct calculation, such as SELECT COUNT(DISTINCT posts.id) …

  • :from - By default, this is the table name of the class, but can be changed to an alternate table name (or even the name of a database view).

Examples for counting all:

Person.count         # returns the total count of all people

Examples for counting by column:

Person.count(:age)  # returns the total count of all people whose age is present in database

Examples for count with options:

Person.count(:conditions => "age > 26")

# because of the named association, it finds the DISTINCT count using LEFT OUTER JOIN.
Person.count(:conditions => "age > 26 AND job.salary > 60000", :include => :job)

# finds the number of rows matching the conditions and joins.
Person.count(:conditions => "age > 26 AND job.salary > 60000",
             :joins => "LEFT JOIN jobs on jobs.person_id = person.id")

Person.count('id', :conditions => "age > 26") # Performs a COUNT(id)
Person.count(:all, :conditions => "age > 26") # Performs a COUNT(*) (:all is an alias for '*')

Note: Person.count(:all) will not work because it will use :all as the condition. Use Person.count instead.



56
57
58
59
# File 'lib/active_record/relation/calculations.rb', line 56

def count(column_name = nil, options = {})
  column_name, options = nil, column_name if column_name.is_a?(Hash)
  calculate(:count, column_name, options)
end

#maximum(column_name, options = {}) ⇒ Object

Calculates the maximum value on a given column. The value is returned with the same data type of the column, or nil if there’s no row. See calculate for examples with options.

Person.maximum('age') # => 93


83
84
85
# File 'lib/active_record/relation/calculations.rb', line 83

def maximum(column_name, options = {})
  calculate(:maximum, column_name, options)
end

#minimum(column_name, options = {}) ⇒ Object

Calculates the minimum value on a given column. The value is returned with the same data type of the column, or nil if there’s no row. See calculate for examples with options.

Person.minimum('age') # => 7


74
75
76
# File 'lib/active_record/relation/calculations.rb', line 74

def minimum(column_name, options = {})
  calculate(:minimum, column_name, options)
end

#pluck(column_name) ⇒ Object

This method is designed to perform select by a single column as direct SQL query Returns Array with values of the specified column name The values has same data type as column.

Examples:

Person.pluck(:id) # SELECT people.id FROM people
Person.uniq.pluck(:role) # SELECT DISTINCT role FROM people
Person.where(:confirmed => true).limit(5).pluck(:id)


179
180
181
182
183
184
# File 'lib/active_record/relation/calculations.rb', line 179

def pluck(column_name)
  column_name = column_name.to_s
  klass.connection.select_all(select(column_name).arel).map! do |attributes|
    klass.type_cast_attribute(attributes.keys.first, klass.initialize_attributes(attributes))
  end
end

#sum(*args) ⇒ Object

Calculates the sum of values on a given column. The value is returned with the same data type of the column, 0 if there’s no row. See calculate for examples with options.

Person.sum('age') # => 4562


92
93
94
95
96
97
98
# File 'lib/active_record/relation/calculations.rb', line 92

def sum(*args)
  if block_given?
    self.to_a.sum(*args) {|*block_args| yield(*block_args)}
  else
    calculate(:sum, *args)
  end
end