Module: PreloadCounts::ClassMethods

Defined in:
lib/preload_counts/ar.rb

Instance Method Summary collapse

Instance Method Details

#preload_counts(options) ⇒ Object 



20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
 
# File 'lib/preload_counts/ar.rb', line 20

def preload_counts(options)
  options = Array(options).inject({}) {|h, v| h[v] = []; h}  unless options.is_a?(Hash)
  options.each do |association, scopes|
    scopes = scopes + [nil]

    # Define singleton metho to load all counts
    name = "preload_#{association.to_s.singularize}_counts"
    singleton = class << self; self end
    singleton.send :define_method, name do
      sql = ["#{table_name}.*"] + scopes_to_select(association, scopes)
      sql = sql.join(', ')
      all.select(sql)
    end

    scopes.each do |scope|
      # Define accessor for each count
      accessor_name = find_accessor_name(association, scope)
      define_method accessor_name do
        result = public_send(association)
        result = result.public_send(scope) if scope
        (self[accessor_name] || result.size).to_i
      end
    end

  end
end