has_many_association.rb 4.3 KB
Newer Older
D
Initial  
David Heinemeier Hansson 已提交
1
module ActiveRecord
2
  # = Active Record Has Many Association
D
Initial  
David Heinemeier Hansson 已提交
3
  module Associations
P
Pratik Naik 已提交
4 5 6 7
    # This is the proxy that handles a has many association.
    #
    # If the association has a <tt>:through</tt> option further specialization
    # is provided by its child HasManyThroughAssociation.
8
    class HasManyAssociation < CollectionAssociation #:nodoc:
9

10
      def insert_record(record, validate = true, raise = false)
11
        set_owner_attributes(record)
12 13 14 15 16 17

        if raise
          record.save!(:validate => validate)
        else
          record.save(:validate => validate)
        end
18
      end
19 20 21

      private

P
Pratik Naik 已提交
22 23 24
        # Returns the number of records in this collection.
        #
        # If the association has a counter cache it gets that value. Otherwise
25
        # it will attempt to do a count via SQL, bounded to <tt>:limit</tt> if
26
        # there's one. Some configuration options like :group make it impossible
27
        # to do an SQL count, in those cases the array count will be used.
28
        #
P
Pratik Naik 已提交
29 30 31 32 33 34
        # That does not depend on whether the collection has already been loaded
        # or not. The +size+ method is the one that takes the loaded flag into
        # account and delegates to +count_records+ if needed.
        #
        # If the collection is empty the target is set to an empty array and
        # the loaded flag is set to true as well.
D
Initial  
David Heinemeier Hansson 已提交
35
        def count_records
36
          count = if has_cached_counter?
37
            owner.send(:read_attribute, cached_counter_attribute_name)
38
          elsif options[:counter_sql] || options[:finder_sql]
39
            reflection.klass.count_by_sql(custom_counter_sql)
D
Initial  
David Heinemeier Hansson 已提交
40
          else
41
            scoped.count
D
Initial  
David Heinemeier Hansson 已提交
42
          end
43 44 45 46

          # If there's nothing in the database and @target has no new records
          # we are certain the current target is an empty array. This is a
          # documented side-effect of the method that may avoid an extra SELECT.
47
          @target ||= [] and loaded! if count == 0
48

49
          [options[:limit], count].compact.min
D
Initial  
David Heinemeier Hansson 已提交
50
        end
51

52 53
        def has_cached_counter?(reflection = reflection)
          owner.attribute_present?(cached_counter_attribute_name(reflection))
D
Initial  
David Heinemeier Hansson 已提交
54
        end
55

56
        def cached_counter_attribute_name(reflection = reflection)
57 58 59
          "#{reflection.name}_count"
        end

60
        def update_counter(difference, reflection = reflection)
61 62
          if has_cached_counter?(reflection)
            counter = cached_counter_attribute_name(reflection)
63 64 65
            owner.class.update_counters(owner.id, counter => difference)
            owner[counter] += difference
            owner.changed_attributes.delete(counter) # eww
66 67 68 69 70 71 72
          end
        end

        # This shit is nasty. We need to avoid the following situation:
        #
        #   * An associated record is deleted via record.destroy
        #   * Hence the callbacks run, and they find a belongs_to on the record with a
73
        #     :counter_cache options which points back at our owner. So they update the
74 75 76 77 78
        #     counter cache.
        #   * In which case, we must make sure to *not* update the counter cache, or else
        #     it will be decremented twice.
        #
        # Hence this method.
79
        def inverse_updates_counter_cache?(reflection = reflection)
80 81 82 83
          counter_name = cached_counter_attribute_name(reflection)
          reflection.klass.reflect_on_all_associations(:belongs_to).any? { |inverse_reflection|
            inverse_reflection.counter_cache_column == counter_name
          }
D
Initial  
David Heinemeier Hansson 已提交
84 85
        end

86
        # Deletes the records according to the <tt>:dependent</tt> option.
87
        def delete_records(records, method)
88
          if method == :destroy
89
            records.each { |r| r.destroy }
90
            update_counter(-records.length) unless inverse_updates_counter_cache?
91
          else
J
Jon Leighton 已提交
92 93 94 95 96 97
            scope = scoped

            unless records == load_target
              keys  = records.map { |r| r[reflection.association_primary_key] }
              scope = scoped.where(reflection.association_primary_key => keys)
            end
98

99 100 101
            if method == :delete_all
              update_counter(-scope.delete_all)
            else
102
              update_counter(-scope.update_all(reflection.foreign_key => nil))
103
            end
104
          end
D
Initial  
David Heinemeier Hansson 已提交
105
        end
106 107 108 109
        
        def foreign_key_present?
          owner.attribute_present?(reflection.association_primary_key)
        end
D
Initial  
David Heinemeier Hansson 已提交
110 111 112
    end
  end
end