Module: ActiveRecord::MassLoad

Defined in:
lib/vex/active_record/mass_load.rb

Defined Under Namespace

Classes: MissingAssociation

Class Method Summary collapse

Instance Method Summary collapse

Class Method Details

.get_id_condition(ids) ⇒ Object


60
61
62
# File 'lib/vex/active_record/mass_load.rb', line 60

def self.get_id_condition(ids)
  id_condition = { :id => ids.sort }
end

Instance Method Details

#mass_load_associations!(models, *associations) ⇒ Object


9
10
11
12
13
14
15
16
17
18
19
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
46
47
48
49
50
51
52
53
54
55
56
57
58
# File 'lib/vex/active_record/mass_load.rb', line 9

def mass_load_associations!(models, *associations)
  if defined?(Rails) && Rails.env.development? && models.detect { |rec| !rec.is_a?(self) }
    raise "Invalid models for mass_load_associations!"
  end

  return models if models.length < 2

  associations.flatten!

  models_by_id = models.inject({}) do |hash, model|
    hash.update model.id => model
  end
  ids = models_by_id.keys
  id_condition = ActiveRecord::MassLoad.get_id_condition(ids)

  associations = associations.collect do |association|
    case association
    when Hash then association.to_a.collect do |k,v| { k => v } end
    else association
    end
  end.flatten

  associations.each do |association|
    if association.is_a?(Hash)
      association = association.to_a.first
      other_assocs = association
      association = other_assocs.shift
    end

    varname = "@#{association}"

    base_assocs = []

    find(:all, :include => association, :conditions => id_condition).each do |model|
      target = models_by_id[model.id]
      proxy = model.instance_variable_get(varname)
      target.instance_variable_set(varname, proxy || MissingAssociation.instance)
      base_assocs << proxy if proxy
    end

    next if !other_assocs || other_assocs.empty?
    next if base_assocs.empty?

    base_assocs.flatten!

    base_assocs.first.class.mass_load_associations! base_assocs, *other_assocs
  end

  models
end