Contents

module BetterAr
  module Calculations
    # Does a count on the query
    #
    # example:
    #  User.count(:age => 20)
    #
    # is the same as:
    #  User.where(:age => 20).count
    #
    # if the key :conditions is present it will fall back to legacy
    #
    # @param [Hash]
    #   Optional follows same convention as {#all}
    # @return [Integer]
    def count(column_name = nil, count_options = {})
      if column_name.is_a?(Hash)
        count_options = column_name
        column_name = nil
      end

      if count_options.empty?
       super
      elsif count_options.key?(:conditions)
        super(column_name, count_options)
      else
        all(count_options).count
      end
    end
  end
end

ActiveSupport.on_load(:active_record) do
  class << ActiveRecord::Base
    prepend BetterAr::Calculations
  end
end

Version data entries

2 entries across 2 versions & 1 rubygems

Version	Path
better_ar-0.2.1	lib/better_ar/calculations.rb
better_ar-0.2.0	lib/better_ar/calculations.rb