Contents

module ActiveRecord
  module Precount
    module ReflectionExtension
      def self.prepended(base)
        class << base
          prepend ClassMethods
        end
      end

      module ClassMethods
        def create(macro, name, scope, options, ar)
          case macro
          when :count_loader
            if ActiveRecord::VERSION::MAJOR >= 4 && ActiveRecord::VERSION::MINOR >= 2
              Reflection::CountLoaderReflection.new(name, scope, options, ar)
            else
              Reflection::AssociationReflection.new(macro, name, scope, options, ar)
            end
          else
            super(macro, name, scope, options, ar)
          end
        end
      end
    end

    module AssociationReflectionExtension
      def klass
        case macro
        when :count_loader
          @klass ||= active_record.send(:compute_type, options[:class_name] || name_without_count.singularize.classify)
        else
          super
        end
      end

      def name_without_count
        name.to_s.sub(/_count$/, "")
      end

      def association_class
        case macro
        when :count_loader
          ActiveRecord::Associations::CountLoader
        else
          super
        end
      end
    end
  end
end

Version data entries

6 entries across 6 versions & 1 rubygems

Version	Path
activerecord-precount-0.5.1	lib/active_record/precount/reflection_extension.rb
activerecord-precount-0.5.0	lib/active_record/precount/reflection_extension.rb
activerecord-precount-0.4.3	lib/active_record/precount/reflection_extension.rb
activerecord-precount-0.4.2	lib/active_record/precount/reflection_extension.rb
activerecord-precount-0.4.1	lib/active_record/precount/reflection_extension.rb
activerecord-precount-0.4.0	lib/active_record/precount/reflection_extension.rb