Contents

module ActiveRecord
  module Precount
    module CollectionProxyExtension
      def count(*args)
        return super(*args) if args.present?

        counter_name = :"#{@association.reflection.name}_count"
        owner        = @association.owner

        if owner.class.has_reflection?(counter_name) && owner.association(counter_name).loaded?
          owner.association(counter_name).target
        else
          super(*args)
        end
      end
    end
  end

  Associations::CollectionProxy.prepend(Precount::CollectionProxyExtension)
end

Version data entries

1 entries across 1 versions & 1 rubygems

Version	Path
activerecord-precount-0.6.0	lib/active_record/precount/collection_proxy_extension.rb