associations.rb 113.4 KB
Newer Older
1
require 'active_support/core_ext/array/wrap'
2
require 'active_support/core_ext/enumerable'
3
require 'active_support/core_ext/module/delegation'
4
require 'active_support/core_ext/object/blank'
5
require 'active_support/core_ext/string/conversions'
6
require 'active_support/core_ext/module/remove_method'
J
Jeremy Kemper 已提交
7

D
Initial  
David Heinemeier Hansson 已提交
8
module ActiveRecord
9
  class InverseOfAssociationNotFoundError < ActiveRecordError #:nodoc:
10 11
    def initialize(reflection, associated_class = nil)
      super("Could not find the inverse association for #{reflection.name} (#{reflection.options[:inverse_of].inspect} in #{associated_class.nil? ? reflection.class_name : associated_class.name})")
12 13 14
    end
  end

D
David Heinemeier Hansson 已提交
15
  class HasManyThroughAssociationNotFoundError < ActiveRecordError #:nodoc:
16 17
    def initialize(owner_class_name, reflection)
      super("Could not find the association #{reflection.options[:through].inspect} in model #{owner_class_name}")
18 19 20
    end
  end

D
David Heinemeier Hansson 已提交
21
  class HasManyThroughAssociationPolymorphicError < ActiveRecordError #:nodoc:
22
    def initialize(owner_class_name, reflection, source_reflection)
23
      super("Cannot have a has_many :through association '#{owner_class_name}##{reflection.name}' on the polymorphic object '#{source_reflection.class_name}##{source_reflection.name}'.")
24 25
    end
  end
26

27 28 29 30 31
  class HasManyThroughAssociationPointlessSourceTypeError < ActiveRecordError #:nodoc:
    def initialize(owner_class_name, reflection, source_reflection)
      super("Cannot have a has_many :through association '#{owner_class_name}##{reflection.name}' with a :source_type option if the '#{reflection.through_reflection.class_name}##{source_reflection.name}' is not polymorphic.  Try removing :source_type on your association.")
    end
  end
32

D
David Heinemeier Hansson 已提交
33
  class HasManyThroughSourceAssociationNotFoundError < ActiveRecordError #:nodoc:
34
    def initialize(reflection)
35 36 37
      through_reflection      = reflection.through_reflection
      source_reflection_names = reflection.source_reflection_names
      source_associations     = reflection.through_reflection.klass.reflect_on_all_associations.collect { |a| a.name.inspect }
38
      super("Could not find the source association(s) #{source_reflection_names.collect{ |a| a.inspect }.to_sentence(:two_words_connector => ' or ', :last_word_connector => ', or ', :locale => :en)} in model #{through_reflection.klass}.  Try 'has_many #{reflection.name.inspect}, :through => #{through_reflection.name.inspect}, :source => <name>'.  Is it one of #{source_associations.to_sentence(:two_words_connector => ' or ', :last_word_connector => ', or ', :locale => :en)}?")
39 40 41
    end
  end

42
  class HasManyThroughSourceAssociationMacroError < ActiveRecordError #:nodoc:
43
    def initialize(reflection)
44 45 46
      through_reflection = reflection.through_reflection
      source_reflection  = reflection.source_reflection
      super("Invalid source reflection macro :#{source_reflection.macro}#{" :through" if source_reflection.options[:through]} for has_many #{reflection.name.inspect}, :through => #{through_reflection.name.inspect}.  Use :source to specify the source reflection.")
47 48 49
    end
  end

50
  class HasManyThroughCantAssociateThroughHasOneOrManyReflection < ActiveRecordError #:nodoc:
51 52 53 54
    def initialize(owner, reflection)
      super("Cannot modify association '#{owner.class.name}##{reflection.name}' because the source reflection class '#{reflection.source_reflection.class_name}' is associated to '#{reflection.through_reflection.class_name}' via :#{reflection.source_reflection.macro}.")
    end
  end
55

56 57 58 59 60 61
  class HasManyThroughCantAssociateNewRecords < ActiveRecordError #:nodoc:
    def initialize(owner, reflection)
      super("Cannot associate new records through '#{owner.class.name}##{reflection.name}' on '#{reflection.source_reflection.class_name rescue nil}##{reflection.source_reflection.name rescue nil}'. Both records must have an id in order to create the has_many :through record associating them.")
    end
  end

62 63 64 65 66 67
  class HasManyThroughCantDissociateNewRecords < ActiveRecordError #:nodoc:
    def initialize(owner, reflection)
      super("Cannot dissociate new records through '#{owner.class.name}##{reflection.name}' on '#{reflection.source_reflection.class_name rescue nil}##{reflection.source_reflection.name rescue nil}'. Both records must have an id in order to delete the has_many :through record associating them.")
    end
  end

68 69 70 71 72 73
  class HasAndBelongsToManyAssociationWithPrimaryKeyError < ActiveRecordError #:nodoc:
    def initialize(reflection)
      super("Primary key is not allowed in a has_and_belongs_to_many join table (#{reflection.options[:join_table]}).")
    end
  end

74 75 76 77 78 79
  class HasAndBelongsToManyAssociationForeignKeyNeeded < ActiveRecordError #:nodoc:
    def initialize(reflection)
      super("Cannot create self referential has_and_belongs_to_many association on '#{reflection.class_name rescue nil}##{reflection.name rescue nil}'. :association_foreign_key cannot be the same as the :foreign_key.")
    end
  end

D
David Heinemeier Hansson 已提交
80
  class EagerLoadPolymorphicError < ActiveRecordError #:nodoc:
81
    def initialize(reflection)
82
      super("Can not eagerly load the polymorphic association #{reflection.name.inspect}")
83 84 85
    end
  end

86 87
  class ReadOnlyAssociation < ActiveRecordError #:nodoc:
    def initialize(reflection)
88
      super("Can not add to a has_many :through association.  Try adding to #{reflection.through_reflection.name.inspect}.")
89 90 91
    end
  end

92
  # This error is raised when trying to destroy a parent instance in N:1 or 1:1 associations
93
  # (has_many, has_one) when there is at least 1 child associated instance.
94 95 96 97 98 99 100
  # ex: if @project.tasks.size > 0, DeleteRestrictionError will be raised when trying to destroy @project
  class DeleteRestrictionError < ActiveRecordError #:nodoc:
    def initialize(reflection)
      super("Cannot delete record because of dependent #{reflection.name}")
    end
  end

P
Pratik Naik 已提交
101
  # See ActiveRecord::Associations::ClassMethods for documentation.
D
Initial  
David Heinemeier Hansson 已提交
102
  module Associations # :nodoc:
103
    extend ActiveSupport::Concern
104

105
    # These classes will be loaded when associations are created.
J
Joshua Peek 已提交
106 107 108 109 110 111 112 113 114 115 116
    # So there is no need to eager load them.
    autoload :AssociationCollection, 'active_record/associations/association_collection'
    autoload :AssociationProxy, 'active_record/associations/association_proxy'
    autoload :BelongsToAssociation, 'active_record/associations/belongs_to_association'
    autoload :BelongsToPolymorphicAssociation, 'active_record/associations/belongs_to_polymorphic_association'
    autoload :HasAndBelongsToManyAssociation, 'active_record/associations/has_and_belongs_to_many_association'
    autoload :HasManyAssociation, 'active_record/associations/has_many_association'
    autoload :HasManyThroughAssociation, 'active_record/associations/has_many_through_association'
    autoload :HasOneAssociation, 'active_record/associations/has_one_association'
    autoload :HasOneThroughAssociation, 'active_record/associations/has_one_through_association'

117
    # Clears out the association cache.
118 119 120
    def clear_association_cache #:nodoc:
      self.class.reflect_on_all_associations.to_a.each do |assoc|
        instance_variable_set "@#{assoc.name}", nil
121
      end unless self.new_record?
122
    end
123

124
    private
125
      # Returns the specified association instance if it responds to :loaded?, nil otherwise.
126
      def association_instance_get(name)
A
Aaron Patterson 已提交
127 128 129 130 131
        ivar = "@#{name}"
        if instance_variable_defined?(ivar)
          association = instance_variable_get(ivar)
          association if association.respond_to?(:loaded?)
        end
132 133 134 135 136 137 138
      end

      # Set the specified association instance.
      def association_instance_set(name, association)
        instance_variable_set("@#{name}", association)
      end

139 140 141 142
    # Associations are a set of macro-like class methods for tying objects together through
    # foreign keys. They express relationships like "Project has one Project Manager"
    # or "Project belongs to a Portfolio". Each macro adds a number of methods to the
    # class which are specialized according to the collection or association symbol and the
143
    # options hash. It works much the same way as Ruby's own <tt>attr*</tt>
144
    # methods.
D
Initial  
David Heinemeier Hansson 已提交
145 146 147
    #
    #   class Project < ActiveRecord::Base
    #     belongs_to              :portfolio
148
    #     has_one                 :project_manager
D
Initial  
David Heinemeier Hansson 已提交
149 150 151 152
    #     has_many                :milestones
    #     has_and_belongs_to_many :categories
    #   end
    #
153
    # The project class now has the following methods (and more) to ease the traversal and
154
    # manipulation of its relationships:
155
    # * <tt>Project#portfolio, Project#portfolio=(portfolio), Project#portfolio.nil?</tt>
D
Initial  
David Heinemeier Hansson 已提交
156 157
    # * <tt>Project#project_manager, Project#project_manager=(project_manager), Project#project_manager.nil?,</tt>
    # * <tt>Project#milestones.empty?, Project#milestones.size, Project#milestones, Project#milestones<<(milestone),</tt>
158
    #   <tt>Project#milestones.delete(milestone), Project#milestones.find(milestone_id), Project#milestones.find(:all, options),</tt>
D
Initial  
David Heinemeier Hansson 已提交
159 160 161 162
    #   <tt>Project#milestones.build, Project#milestones.create</tt>
    # * <tt>Project#categories.empty?, Project#categories.size, Project#categories, Project#categories<<(category1),</tt>
    #   <tt>Project#categories.delete(category1)</tt>
    #
163 164
    # === A word of warning
    #
165 166
    # Don't create associations that have the same name as instance methods of
    # <tt>ActiveRecord::Base</tt>. Since the association adds a method with that name to
167
    # its model, it will override the inherited method and break things.
P
Pratik Naik 已提交
168
    # For instance, +attributes+ and +connection+ would be bad choices for association names.
169
    #
170
    # == Auto-generated methods
D
Initial  
David Heinemeier Hansson 已提交
171
    #
172
    # === Singular associations (one-to-one)
173 174 175
    #                                     |            |  belongs_to  |
    #   generated methods                 | belongs_to | :polymorphic | has_one
    #   ----------------------------------+------------+--------------+---------
P
Pratik Naik 已提交
176 177 178 179 180
    #   other                             |     X      |      X       |    X
    #   other=(other)                     |     X      |      X       |    X
    #   build_other(attributes={})        |     X      |              |    X
    #   create_other(attributes={})       |     X      |              |    X
    #   other.create!(attributes={})      |            |              |    X
181 182 183
    #
    # ===Collection associations (one-to-many / many-to-many)
    #                                     |       |          | has_many
184
    #   generated methods                 | habtm | has_many | :through
185
    #   ----------------------------------+-------+----------+----------
P
Pratik Naik 已提交
186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209
    #   others                            |   X   |    X     |    X
    #   others=(other,other,...)          |   X   |    X     |    X
    #   other_ids                         |   X   |    X     |    X
    #   other_ids=(id,id,...)             |   X   |    X     |    X
    #   others<<                          |   X   |    X     |    X
    #   others.push                       |   X   |    X     |    X
    #   others.concat                     |   X   |    X     |    X
    #   others.build(attributes={})       |   X   |    X     |    X
    #   others.create(attributes={})      |   X   |    X     |    X
    #   others.create!(attributes={})     |   X   |    X     |    X
    #   others.size                       |   X   |    X     |    X
    #   others.length                     |   X   |    X     |    X
    #   others.count                      |   X   |    X     |    X
    #   others.sum(args*,&block)          |   X   |    X     |    X
    #   others.empty?                     |   X   |    X     |    X
    #   others.clear                      |   X   |    X     |    X
    #   others.delete(other,other,...)    |   X   |    X     |    X
    #   others.delete_all                 |   X   |    X     |
    #   others.destroy_all                |   X   |    X     |    X
    #   others.find(*args)                |   X   |    X     |    X
    #   others.find_first                 |   X   |          |
    #   others.exists?                    |   X   |    X     |    X
    #   others.uniq                       |   X   |    X     |    X
    #   others.reset                      |   X   |    X     |    X
D
Initial  
David Heinemeier Hansson 已提交
210
    #
211
    # == Cardinality and associations
212
    #
P
Pratik Naik 已提交
213
    # Active Record associations can be used to describe one-to-one, one-to-many and many-to-many
P
Pratik Naik 已提交
214 215
    # relationships between models. Each model uses an association to describe its role in
    # the relation. The +belongs_to+ association is always used in the model that has
216
    # the foreign key.
217 218 219
    #
    # === One-to-one
    #
220
    # Use +has_one+ in the base, and +belongs_to+ in the associated model.
221 222 223 224 225 226 227 228 229 230
    #
    #   class Employee < ActiveRecord::Base
    #     has_one :office
    #   end
    #   class Office < ActiveRecord::Base
    #     belongs_to :employee    # foreign key - employee_id
    #   end
    #
    # === One-to-many
    #
231
    # Use +has_many+ in the base, and +belongs_to+ in the associated model.
232 233 234 235 236
    #
    #   class Manager < ActiveRecord::Base
    #     has_many :employees
    #   end
    #   class Employee < ActiveRecord::Base
237
    #     belongs_to :manager     # foreign key - manager_id
238 239 240 241 242 243
    #   end
    #
    # === Many-to-many
    #
    # There are two ways to build a many-to-many relationship.
    #
244
    # The first way uses a +has_many+ association with the <tt>:through</tt> option and a join model, so
245 246 247 248 249 250 251 252 253 254 255 256 257 258 259
    # there are two stages of associations.
    #
    #   class Assignment < ActiveRecord::Base
    #     belongs_to :programmer  # foreign key - programmer_id
    #     belongs_to :project     # foreign key - project_id
    #   end
    #   class Programmer < ActiveRecord::Base
    #     has_many :assignments
    #     has_many :projects, :through => :assignments
    #   end
    #   class Project < ActiveRecord::Base
    #     has_many :assignments
    #     has_many :programmers, :through => :assignments
    #   end
    #
260
    # For the second way, use +has_and_belongs_to_many+ in both models. This requires a join table
261 262 263 264 265 266 267 268 269
    # that has no corresponding model or primary key.
    #
    #   class Programmer < ActiveRecord::Base
    #     has_and_belongs_to_many :projects       # foreign keys in the join table
    #   end
    #   class Project < ActiveRecord::Base
    #     has_and_belongs_to_many :programmers    # foreign keys in the join table
    #   end
    #
270
    # Choosing which way to build a many-to-many relationship is not always simple.
271
    # If you need to work with the relationship model as its own entity,
272
    # use <tt>has_many :through</tt>. Use +has_and_belongs_to_many+ when working with legacy schemas or when
273 274
    # you never work directly with the relationship itself.
    #
275
    # == Is it a +belongs_to+ or +has_one+ association?
D
Initial  
David Heinemeier Hansson 已提交
276
    #
277
    # Both express a 1-1 relationship. The difference is mostly where to place the foreign
278
    # key, which goes on the table for the class declaring the +belongs_to+ relationship.
D
Initial  
David Heinemeier Hansson 已提交
279
    #
280 281 282
    #   class User < ActiveRecord::Base
    #     # I reference an account.
    #     belongs_to :account
D
Initial  
David Heinemeier Hansson 已提交
283 284
    #   end
    #
285 286 287
    #   class Account < ActiveRecord::Base
    #     # One user references me.
    #     has_one :user
D
Initial  
David Heinemeier Hansson 已提交
288 289 290 291
    #   end
    #
    # The tables for these classes could look something like:
    #
292
    #   CREATE TABLE users (
D
Initial  
David Heinemeier Hansson 已提交
293
    #     id int(11) NOT NULL auto_increment,
294 295
    #     account_id int(11) default NULL,
    #     name varchar default NULL,
D
Initial  
David Heinemeier Hansson 已提交
296 297 298
    #     PRIMARY KEY  (id)
    #   )
    #
299
    #   CREATE TABLE accounts (
D
Initial  
David Heinemeier Hansson 已提交
300 301 302 303 304
    #     id int(11) NOT NULL auto_increment,
    #     name varchar default NULL,
    #     PRIMARY KEY  (id)
    #   )
    #
305 306
    # == Unsaved objects and associations
    #
307 308
    # You can manipulate objects and associations before they are saved to the database, but
    # there is some special behavior you should be aware of, mostly involving the saving of
309
    # associated objects.
310
    #
311
    # You can set the :autosave option on a <tt>has_one</tt>, <tt>belongs_to</tt>,
312 313
    # <tt>has_many</tt>, or <tt>has_and_belongs_to_many</tt> association. Setting it
    # to +true+ will _always_ save the members, whereas setting it to +false+ will
314
    # _never_ save the members. More details about :autosave option is available at
315
    # autosave_association.rb .
316
    #
317 318
    # === One-to-one associations
    #
319 320
    # * Assigning an object to a +has_one+ association automatically saves that object and
    # the object being replaced (if there is one), in order to update their primary
321
    # keys - except if the parent object is unsaved (<tt>new_record? == true</tt>).
322
    # * If either of these saves fail (due to one of the objects being invalid) the assignment
323
    # statement returns +false+ and the assignment is cancelled.
324
    # * If you wish to assign an object to a +has_one+ association without saving it,
325
    # use the <tt>association.build</tt> method (documented below).
326
    # * Assigning an object to a +belongs_to+ association does not save the object, since
327
    # the foreign key field belongs on the parent. It does not save the parent either.
328 329 330
    #
    # === Collections
    #
331 332
    # * Adding an object to a collection (+has_many+ or +has_and_belongs_to_many+) automatically
    # saves that object, except if the parent object (the owner of the collection) is not yet
333
    # stored in the database.
334
    # * If saving any of the objects being added to a collection (via <tt>push</tt> or similar)
335
    # fails, then <tt>push</tt> returns +false+.
336
    # * You can add an object to a collection without automatically saving it by using the
337
    # <tt>collection.build</tt> method (documented below).
338
    # * All unsaved (<tt>new_record? == true</tt>) members of the collection are automatically
339
    # saved when the parent is saved.
340
    #
341
    # === Association callbacks
342
    #
343
    # Similar to the normal callbacks that hook into the life cycle of an Active Record object,
344
    # you can also define callbacks that get triggered when you add an object to or remove an
345
    # object from an association collection.
346 347 348 349 350 351 352
    #
    #   class Project
    #     has_and_belongs_to_many :developers, :after_add => :evaluate_velocity
    #
    #     def evaluate_velocity(developer)
    #       ...
    #     end
353
    #   end
354 355
    #
    # It's possible to stack callbacks by passing them as an array. Example:
356
    #
357
    #   class Project
358
    #     has_and_belongs_to_many :developers,
359
    #                             :after_add => [:evaluate_velocity, Proc.new { |p, d| p.shipping_date = Time.now}]
360 361
    #   end
    #
362
    # Possible callbacks are: +before_add+, +after_add+, +before_remove+ and +after_remove+.
363
    #
364 365
    # Should any of the +before_add+ callbacks throw an exception, the object does not get
    # added to the collection. Same with the +before_remove+ callbacks; if an exception is
366
    # thrown the object doesn't get removed.
367
    #
368 369
    # === Association extensions
    #
370 371
    # The proxy objects that control the access to associations can be extended through anonymous
    # modules. This is especially beneficial for adding new finders, creators, and other
372
    # factory-type methods that are only used as part of this association.
373 374
    #
    #   class Account < ActiveRecord::Base
375
    #     has_many :people do
376
    #       def find_or_create_by_name(name)
377
    #         first_name, last_name = name.split(" ", 2)
D
David Heinemeier Hansson 已提交
378
    #         find_or_create_by_first_name_and_last_name(first_name, last_name)
379
    #       end
380
    #     end
381 382 383 384 385 386
    #   end
    #
    #   person = Account.find(:first).people.find_or_create_by_name("David Heinemeier Hansson")
    #   person.first_name # => "David"
    #   person.last_name  # => "Heinemeier Hansson"
    #
387 388
    # If you need to share the same extensions between many associations, you can use a named
    # extension module.
389 390 391
    #
    #   module FindOrCreateByNameExtension
    #     def find_or_create_by_name(name)
392
    #       first_name, last_name = name.split(" ", 2)
D
David Heinemeier Hansson 已提交
393
    #       find_or_create_by_first_name_and_last_name(first_name, last_name)
394 395 396 397 398 399 400 401 402 403
    #     end
    #   end
    #
    #   class Account < ActiveRecord::Base
    #     has_many :people, :extend => FindOrCreateByNameExtension
    #   end
    #
    #   class Company < ActiveRecord::Base
    #     has_many :people, :extend => FindOrCreateByNameExtension
    #   end
404
    #
405
    # If you need to use multiple named extension modules, you can specify an array of modules
406
    # with the <tt>:extend</tt> option.
407 408
    # In the case of name conflicts between methods in the modules, methods in modules later
    # in the array supercede those earlier in the array.
409 410 411 412 413
    #
    #   class Account < ActiveRecord::Base
    #     has_many :people, :extend => [FindOrCreateByNameExtension, FindRecentExtension]
    #   end
    #
414 415
    # Some extensions can only be made to work with knowledge of the association proxy's internals.
    # Extensions can access relevant state using accessors on the association proxy:
416
    #
417 418
    # * +proxy_owner+ - Returns the object the association is part of.
    # * +proxy_reflection+ - Returns the reflection object that describes the association.
419
    # * +proxy_target+ - Returns the associated object for +belongs_to+ and +has_one+, or
420
    #   the collection of associated objects for +has_many+ and +has_and_belongs_to_many+.
421
    #
R
Rick Olson 已提交
422
    # === Association Join Models
423
    #
424 425
    # Has Many associations can be configured with the <tt>:through</tt> option to use an
    # explicit join model to retrieve the data.  This operates similarly to a
426
    # +has_and_belongs_to_many+ association.  The advantage is that you're able to add validations,
427
    # callbacks, and extra attributes on the join model.  Consider the following schema:
428
    #
429 430 431 432
    #   class Author < ActiveRecord::Base
    #     has_many :authorships
    #     has_many :books, :through => :authorships
    #   end
433
    #
434 435 436 437
    #   class Authorship < ActiveRecord::Base
    #     belongs_to :author
    #     belongs_to :book
    #   end
438
    #
439
    #   @author = Author.find :first
440
    #   @author.authorships.collect { |a| a.book } # selects all books that the author's authorships belong to
441
    #   @author.books                              # selects all books by using the Authorship join model
442
    #
443
    # You can also go through a +has_many+ association on the join model:
444
    #
445 446 447 448
    #   class Firm < ActiveRecord::Base
    #     has_many   :clients
    #     has_many   :invoices, :through => :clients
    #   end
449
    #
450 451 452 453
    #   class Client < ActiveRecord::Base
    #     belongs_to :firm
    #     has_many   :invoices
    #   end
454
    #
455 456 457 458 459 460
    #   class Invoice < ActiveRecord::Base
    #     belongs_to :client
    #   end
    #
    #   @firm = Firm.find :first
    #   @firm.clients.collect { |c| c.invoices }.flatten # select all invoices for all clients of the firm
461
    #   @firm.invoices                                   # selects all invoices by going through the Client join model
462
    #
463 464 465 466 467 468 469 470 471 472 473 474 475 476 477 478 479 480 481 482
    # Similarly you can go through a +has_one+ association on the join model:
    #
    #   class Group < ActiveRecord::Base
    #     has_many   :users
    #     has_many   :avatars, :through => :users
    #   end
    #
    #   class User < ActiveRecord::Base
    #     belongs_to :group
    #     has_one    :avatar
    #   end
    #
    #   class Avatar < ActiveRecord::Base
    #     belongs_to :user
    #   end
    #
    #   @group = Group.first
    #   @group.users.collect { |u| u.avatar }.flatten # select all avatars for all users in the group
    #   @group.avatars                                # selects all avatars by going through the User join model.
    #
483 484
    # An important caveat with going through +has_one+ or +has_many+ associations on the
    # join model is that these associations are *read-only*.  For example, the following
485
    # would not work following the previous example:
486
    #
487
    #   @group.avatars << Avatar.new   # this would work if User belonged_to Avatar rather than the other way around
488 489
    #   @group.avatars.delete(@group.avatars.last)  # so would this
    #
R
Rick Olson 已提交
490
    # === Polymorphic Associations
491
    #
492 493
    # Polymorphic associations on models are not restricted on what types of models they
    # can be associated with.  Rather, they specify an interface that a +has_many+ association
494
    # must adhere to.
495
    #
R
Rick Olson 已提交
496 497 498
    #   class Asset < ActiveRecord::Base
    #     belongs_to :attachable, :polymorphic => true
    #   end
499
    #
R
Rick Olson 已提交
500
    #   class Post < ActiveRecord::Base
501
    #     has_many :assets, :as => :attachable         # The :as option specifies the polymorphic interface to use.
R
Rick Olson 已提交
502 503 504
    #   end
    #
    #   @asset.attachable = @post
505
    #
506 507
    # This works by using a type column in addition to a foreign key to specify the associated
    # record.  In the Asset example, you'd need an +attachable_id+ integer column and an
508
    # +attachable_type+ string column.
R
Rick Olson 已提交
509
    #
510 511 512
    # Using polymorphic associations in combination with single table inheritance (STI) is
    # a little tricky. In order for the associations to work as expected, ensure that you
    # store the base model for the STI models in the type column of the polymorphic
513
    # association. To continue with the asset example above, suppose there are guest posts
514
    # and member posts that use the posts table for STI. In this case, there must be a +type+
515
    # column in the posts table.
D
David Heinemeier Hansson 已提交
516 517 518
    #
    #   class Asset < ActiveRecord::Base
    #     belongs_to :attachable, :polymorphic => true
519
    #
D
David Heinemeier Hansson 已提交
520 521 522 523
    #     def attachable_type=(sType)
    #        super(sType.to_s.classify.constantize.base_class.to_s)
    #     end
    #   end
524
    #
D
David Heinemeier Hansson 已提交
525 526 527 528 529
    #   class Post < ActiveRecord::Base
    #     # because we store "Post" in attachable_type now :dependent => :destroy will work
    #     has_many :assets, :as => :attachable, :dependent => :destroy
    #   end
    #
530
    #   class GuestPost < Post
D
David Heinemeier Hansson 已提交
531 532
    #   end
    #
533
    #   class MemberPost < Post
D
David Heinemeier Hansson 已提交
534 535
    #   end
    #
D
Initial  
David Heinemeier Hansson 已提交
536 537
    # == Caching
    #
538 539
    # All of the methods are built on a simple caching principle that will keep the result
    # of the last query around unless specifically instructed not to. The cache is even
540 541
    # shared across methods to make it even cheaper to use the macro-added methods without
    # worrying too much about performance at the first go.
D
Initial  
David Heinemeier Hansson 已提交
542 543 544 545 546 547 548
    #
    #   project.milestones             # fetches milestones from the database
    #   project.milestones.size        # uses the milestone cache
    #   project.milestones.empty?      # uses the milestone cache
    #   project.milestones(true).size  # fetches milestones from the database
    #   project.milestones             # uses the milestone cache
    #
549 550
    # == Eager loading of associations
    #
551 552 553
    # Eager loading is a way to find objects of a certain class and a number of named associations.
    # This is one of the easiest ways of to prevent the dreaded 1+N problem in which fetching 100
    # posts that each need to display their author triggers 101 database queries. Through the
554
    # use of eager loading, the 101 queries can be reduced to 2.
555 556 557 558 559 560 561 562
    #
    #   class Post < ActiveRecord::Base
    #     belongs_to :author
    #     has_many   :comments
    #   end
    #
    # Consider the following loop using the class above:
    #
P
Pratik Naik 已提交
563
    #   for post in Post.all
564 565 566
    #     puts "Post:            " + post.title
    #     puts "Written by:      " + post.author.name
    #     puts "Last comment on: " + post.comments.first.created_on
567
    #   end
568
    #
569
    # To iterate over these one hundred posts, we'll generate 201 database queries. Let's
570
    # first just optimize it for retrieving the author:
571
    #
572
    #   for post in Post.find(:all, :include => :author)
573
    #
574 575 576
    # This references the name of the +belongs_to+ association that also used the <tt>:author</tt>
    # symbol. After loading the posts, find will collect the +author_id+ from each one and load
    # all the referenced authors with one query. Doing so will cut down the number of queries
577
    # from 201 to 102.
578 579 580
    #
    # We can improve upon the situation further by referencing both associations in the finder with:
    #
581
    #   for post in Post.find(:all, :include => [ :author, :comments ])
582
    #
583 584
    # This will load all comments with a single query. This reduces the total number of queries
    # to 3. More generally the number of queries will be 1 plus the number of associations
585
    # named (except if some of the associations are polymorphic +belongs_to+ - see below).
586
    #
587
    # To include a deep hierarchy of associations, use a hash:
588 589 590
    #
    #   for post in Post.find(:all, :include => [ :author, { :comments => { :author => :gravatar } } ])
    #
591 592
    # That'll grab not only all the comments but all their authors and gravatar pictures.
    # You can mix and match symbols, arrays and hashes in any combination to describe the
593
    # associations you want to load.
594
    #
595 596 597 598
    # All of this power shouldn't fool you into thinking that you can pull out huge amounts
    # of data with no performance penalty just because you've reduced the number of queries.
    # The database still needs to send all the data to Active Record and it still needs to
    # be processed. So it's no catch-all for performance problems, but it's a great way to
599
    # cut down on the number of queries in a situation as the one described above.
600
    #
601 602
    # Since only one table is loaded at a time, conditions or orders cannot reference tables
    # other than the main one. If this is the case Active Record falls back to the previously
603
    # used LEFT OUTER JOIN based strategy. For example
604
    #
E
Emilio Tagua 已提交
605
    #   Post.includes([:author, :comments]).where(['comments.approved = ?', true]).all
606
    #
607
    # This will result in a single SQL query with joins along the lines of:
608
    # <tt>LEFT OUTER JOIN comments ON comments.post_id = posts.id</tt> and
609
    # <tt>LEFT OUTER JOIN authors ON authors.id = posts.author_id</tt>. Note that using conditions
610
    # like this can have unintended consequences.
611 612
    # In the above example posts with no approved comments are not returned at all, because
    # the conditions apply to the SQL statement as a whole and not just to the association.
613
    # You must disambiguate column references for this fallback to happen, for example
614
    # <tt>:order => "author.name DESC"</tt> will work but <tt>:order => "name DESC"</tt> will not.
P
Pratik Naik 已提交
615
    #
616
    # If you do want eager load only some members of an association it is usually more natural
617
    # to <tt>:include</tt> an association which has conditions defined on it:
P
Pratik Naik 已提交
618 619 620 621 622 623 624
    #
    #   class Post < ActiveRecord::Base
    #     has_many :approved_comments, :class_name => 'Comment', :conditions => ['approved = ?', true]
    #   end
    #
    #   Post.find(:all, :include => :approved_comments)
    #
625
    # This will load posts and eager load the +approved_comments+ association, which contains
626
    # only those comments that have been approved.
627
    #
628
    # If you eager load an association with a specified <tt>:limit</tt> option, it will be ignored,
629
    # returning all the associated objects:
630 631 632 633 634 635 636
    #
    #   class Picture < ActiveRecord::Base
    #     has_many :most_recent_comments, :class_name => 'Comment', :order => 'id DESC', :limit => 10
    #   end
    #
    #   Picture.find(:first, :include => :most_recent_comments).most_recent_comments # => returns all associated comments.
    #
637
    # When eager loaded, conditions are interpolated in the context of the model class, not
638
    # the model instance.  Conditions are lazily interpolated before the actual model exists.
639
    #
P
Pratik Naik 已提交
640
    # Eager loading is supported with polymorphic associations.
641 642 643 644 645
    #
    #   class Address < ActiveRecord::Base
    #     belongs_to :addressable, :polymorphic => true
    #   end
    #
P
Pratik Naik 已提交
646
    # A call that tries to eager load the addressable model
647
    #
P
Pratik Naik 已提交
648
    #   Address.find(:all, :include => :addressable)
649
    #
650
    # This will execute one query to load the addresses and load the addressables with one
651
    # query per addressable type.
652 653
    # For example if all the addressables are either of class Person or Company then a total
    # of 3 queries will be executed. The list of addressable types to load is determined on
654
    # the back of the addresses loaded. This is not supported if Active Record has to fallback
655 656
    # to the previous implementation of eager loading and will raise ActiveRecord::EagerLoadPolymorphicError.
    # The reason is that the parent model's type is a column value so its corresponding table
657
    # name cannot be put in the +FROM+/+JOIN+ clauses of that query.
658
    #
659 660
    # == Table Aliasing
    #
661 662 663
    # Active Record uses table aliasing in the case that a table is referenced multiple times
    # in a join.  If a table is referenced only once, the standard table name is used.  The
    # second time, the table is aliased as <tt>#{reflection_name}_#{parent_table_name}</tt>.
664
    # Indexes are appended for any more successive uses of the table name.
665
    #
P
Pratik Naik 已提交
666 667 668 669 670 671
    #   Post.find :all, :joins => :comments
    #   # => SELECT ... FROM posts INNER JOIN comments ON ...
    #   Post.find :all, :joins => :special_comments # STI
    #   # => SELECT ... FROM posts INNER JOIN comments ON ... AND comments.type = 'SpecialComment'
    #   Post.find :all, :joins => [:comments, :special_comments] # special_comments is the reflection name, posts is the parent table name
    #   # => SELECT ... FROM posts INNER JOIN comments ON ... INNER JOIN comments special_comments_posts
672
    #
673
    # Acts as tree example:
674
    #
P
Pratik Naik 已提交
675 676 677 678 679 680 681 682 683
    #   TreeMixin.find :all, :joins => :children
    #   # => SELECT ... FROM mixins INNER JOIN mixins childrens_mixins ...
    #   TreeMixin.find :all, :joins => {:children => :parent}
    #   # => SELECT ... FROM mixins INNER JOIN mixins childrens_mixins ...
    #                               INNER JOIN parents_mixins ...
    #   TreeMixin.find :all, :joins => {:children => {:parent => :children}}
    #   # => SELECT ... FROM mixins INNER JOIN mixins childrens_mixins ...
    #                               INNER JOIN parents_mixins ...
    #                               INNER JOIN mixins childrens_mixins_2
684
    #
685
    # Has and Belongs to Many join tables use the same idea, but add a <tt>_join</tt> suffix:
686
    #
P
Pratik Naik 已提交
687 688 689 690 691 692 693 694 695
    #   Post.find :all, :joins => :categories
    #   # => SELECT ... FROM posts INNER JOIN categories_posts ... INNER JOIN categories ...
    #   Post.find :all, :joins => {:categories => :posts}
    #   # => SELECT ... FROM posts INNER JOIN categories_posts ... INNER JOIN categories ...
    #                              INNER JOIN categories_posts posts_categories_join INNER JOIN posts posts_categories
    #   Post.find :all, :joins => {:categories => {:posts => :categories}}
    #   # => SELECT ... FROM posts INNER JOIN categories_posts ... INNER JOIN categories ...
    #                              INNER JOIN categories_posts posts_categories_join INNER JOIN posts posts_categories
    #                              INNER JOIN categories_posts categories_posts_join INNER JOIN categories categories_posts_2
696
    #
697
    # If you wish to specify your own custom joins using a <tt>:joins</tt> option, those table
698
    # names will take precedence over the eager associations:
699
    #
P
Pratik Naik 已提交
700 701 702 703 704
    #   Post.find :all, :joins => :comments, :joins => "inner join comments ..."
    #   # => SELECT ... FROM posts INNER JOIN comments_posts ON ... INNER JOIN comments ...
    #   Post.find :all, :joins => [:comments, :special_comments], :joins => "inner join comments ..."
    #   # => SELECT ... FROM posts INNER JOIN comments comments_posts ON ...
    #                              INNER JOIN comments special_comments_posts ...
705
    #                              INNER JOIN comments ...
706
    #
707
    # Table aliases are automatically truncated according to the maximum length of table identifiers
708
    # according to the specific database.
709
    #
D
Initial  
David Heinemeier Hansson 已提交
710 711 712 713 714 715 716 717 718 719
    # == Modules
    #
    # By default, associations will look for objects within the current module scope. Consider:
    #
    #   module MyApplication
    #     module Business
    #       class Firm < ActiveRecord::Base
    #          has_many :clients
    #        end
    #
P
Pratik Naik 已提交
720
    #       class Client < ActiveRecord::Base; end
D
Initial  
David Heinemeier Hansson 已提交
721 722 723
    #     end
    #   end
    #
724
    # When <tt>Firm#clients</tt> is called, it will in turn call
725
    # <tt>MyApplication::Business::Client.find_all_by_firm_id(firm.id)</tt>.
726
    # If you want to associate with a class in another module scope, this can be done by
727
    # specifying the complete class name.
D
Initial  
David Heinemeier Hansson 已提交
728 729 730 731 732 733 734 735 736 737 738 739 740
    #
    #   module MyApplication
    #     module Business
    #       class Firm < ActiveRecord::Base; end
    #     end
    #
    #     module Billing
    #       class Account < ActiveRecord::Base
    #         belongs_to :firm, :class_name => "MyApplication::Business::Firm"
    #       end
    #     end
    #   end
    #
P
Pratik Naik 已提交
741 742
    # == Bi-directional associations
    #
743
    # When you specify an association there is usually an association on the associated model
744
    # that specifies the same relationship in reverse.  For example, with the following models:
P
Pratik Naik 已提交
745 746 747 748 749 750 751 752 753 754 755 756 757 758
    #
    #    class Dungeon < ActiveRecord::Base
    #      has_many :traps
    #      has_one :evil_wizard
    #    end
    #
    #    class Trap < ActiveRecord::Base
    #      belongs_to :dungeon
    #    end
    #
    #    class EvilWizard < ActiveRecord::Base
    #      belongs_to :dungeon
    #    end
    #
759 760
    # The +traps+ association on +Dungeon+ and the the +dungeon+ association on +Trap+ are
    # the inverse of each other and the inverse of the +dungeon+ association on +EvilWizard+
761
    # is the +evil_wizard+ association on +Dungeon+ (and vice-versa).  By default,
762
    # Active Record doesn't know anything about these inverse relationships and so no object
763
    # loading optimisation is possible.  For example:
P
Pratik Naik 已提交
764 765 766 767 768 769 770
    #
    #    d = Dungeon.first
    #    t = d.traps.first
    #    d.level == t.dungeon.level # => true
    #    d.level = 10
    #    d.level == t.dungeon.level # => false
    #
771 772
    # The +Dungeon+ instances +d+ and <tt>t.dungeon</tt> in the above example refer to
    # the same object data from the database, but are actually different in-memory copies
773
    # of that data.  Specifying the <tt>:inverse_of</tt> option on associations lets you tell
774
    # Active Record about inverse relationships and it will optimise object loading.  For
775
    # example, if we changed our model definitions to:
P
Pratik Naik 已提交
776 777 778 779 780 781 782 783 784 785 786 787 788 789
    #
    #    class Dungeon < ActiveRecord::Base
    #      has_many :traps, :inverse_of => :dungeon
    #      has_one :evil_wizard, :inverse_of => :dungeon
    #    end
    #
    #    class Trap < ActiveRecord::Base
    #      belongs_to :dungeon, :inverse_of => :traps
    #    end
    #
    #    class EvilWizard < ActiveRecord::Base
    #      belongs_to :dungeon, :inverse_of => :evil_wizard
    #    end
    #
790
    # Then, from our code snippet above, +d+ and <tt>t.dungeon</tt> are actually the same
791
    # in-memory instance and our final <tt>d.level == t.dungeon.level</tt> will return +true+.
P
Pratik Naik 已提交
792 793 794 795 796 797 798
    #
    # There are limitations to <tt>:inverse_of</tt> support:
    #
    # * does not work with <tt>:through</tt> associations.
    # * does not work with <tt>:polymorphic</tt> associations.
    # * for +belongs_to+ associations +has_many+ inverse associations are ignored.
    #
799
    # == Type safety with <tt>ActiveRecord::AssociationTypeMismatch</tt>
D
Initial  
David Heinemeier Hansson 已提交
800
    #
801
    # If you attempt to assign an object to an association that doesn't match the inferred
802
    # or specified <tt>:class_name</tt>, you'll get an <tt>ActiveRecord::AssociationTypeMismatch</tt>.
D
Initial  
David Heinemeier Hansson 已提交
803 804 805
    #
    # == Options
    #
806
    # All of the association macros can be specialized through options. This makes cases
807
    # more complex than the simple and guessable ones possible.
D
Initial  
David Heinemeier Hansson 已提交
808
    module ClassMethods
P
Pratik Naik 已提交
809 810 811 812 813
      # Specifies a one-to-many association. The following methods for retrieval and query of
      # collections of associated objects will be added:
      #
      # [collection(force_reload = false)]
      #   Returns an array of all the associated objects.
D
Initial  
David Heinemeier Hansson 已提交
814
      #   An empty array is returned if none are found.
P
Pratik Naik 已提交
815 816
      # [collection<<(object, ...)]
      #   Adds one or more objects to the collection by setting their foreign keys to the collection's primary key.
817
      #   Note that this operation instantly fires update sql without waiting for the save or update call on the
818
      #   parent object.
P
Pratik Naik 已提交
819 820
      # [collection.delete(object, ...)]
      #   Removes one or more objects from the collection by setting their foreign keys to +NULL+.
821 822
      #   Objects will be in addition destroyed if they're associated with <tt>:dependent => :destroy</tt>,
      #   and deleted if they're associated with <tt>:dependent => :delete_all</tt>.
P
Pratik Naik 已提交
823
      # [collection=objects]
824 825 826
      #   Replaces the collections content by deleting and adding objects as appropriate. If the <tt>:through</tt>
      #   option is true callbacks in the join models are triggered except destroy callbacks, since deletion is
      #   direct.
P
Pratik Naik 已提交
827 828 829
      # [collection_singular_ids]
      #   Returns an array of the associated objects' ids
      # [collection_singular_ids=ids]
830 831
      #   Replace the collection with the objects identified by the primary keys in +ids+. This
      #   method loads the models and calls <tt>collection=</tt>. See above.
P
Pratik Naik 已提交
832 833 834 835
      # [collection.clear]
      #   Removes every object from the collection. This destroys the associated objects if they
      #   are associated with <tt>:dependent => :destroy</tt>, deletes them directly from the
      #   database if <tt>:dependent => :delete_all</tt>, otherwise sets their foreign keys to +NULL+.
836 837
      #   If the <tt>:through</tt> option is true no destroy callbacks are invoked on the join models.
      #   Join models are directly deleted.
P
Pratik Naik 已提交
838 839 840 841 842 843
      # [collection.empty?]
      #   Returns +true+ if there are no associated objects.
      # [collection.size]
      #   Returns the number of associated objects.
      # [collection.find(...)]
      #   Finds an associated object according to the same rules as ActiveRecord::Base.find.
P
Pratik Naik 已提交
844
      # [collection.exists?(...)]
P
Pratik Naik 已提交
845 846 847 848 849
      #   Checks whether an associated object with the given conditions exists.
      #   Uses the same rules as ActiveRecord::Base.exists?.
      # [collection.build(attributes = {}, ...)]
      #   Returns one or more new objects of the collection type that have been instantiated
      #   with +attributes+ and linked to this object through a foreign key, but have not yet
P
Pratik Naik 已提交
850
      #   been saved.
P
Pratik Naik 已提交
851 852 853
      # [collection.create(attributes = {})]
      #   Returns a new object of the collection type that has been instantiated
      #   with +attributes+, linked to this object through a foreign key, and that has already
P
Pratik Naik 已提交
854 855
      #   been saved (if it passed the validation). *Note*: This only works if the base model
      #   already exists in the DB, not if it is a new (unsaved) record!
P
Pratik Naik 已提交
856 857 858 859 860
      #
      # (*Note*: +collection+ is replaced with the symbol passed as the first argument, so
      # <tt>has_many :clients</tt> would add among others <tt>clients.empty?</tt>.)
      #
      # === Example
D
Initial  
David Heinemeier Hansson 已提交
861
      #
P
Pratik Naik 已提交
862 863
      # Example: A Firm class declares <tt>has_many :clients</tt>, which will add:
      # * <tt>Firm#clients</tt> (similar to <tt>Clients.find :all, :conditions => ["firm_id = ?", id]</tt>)
D
Initial  
David Heinemeier Hansson 已提交
864 865
      # * <tt>Firm#clients<<</tt>
      # * <tt>Firm#clients.delete</tt>
866
      # * <tt>Firm#clients=</tt>
867
      # * <tt>Firm#client_ids</tt>
868
      # * <tt>Firm#client_ids=</tt>
D
Initial  
David Heinemeier Hansson 已提交
869 870 871
      # * <tt>Firm#clients.clear</tt>
      # * <tt>Firm#clients.empty?</tt> (similar to <tt>firm.clients.size == 0</tt>)
      # * <tt>Firm#clients.size</tt> (similar to <tt>Client.count "firm_id = #{id}"</tt>)
872
      # * <tt>Firm#clients.find</tt> (similar to <tt>Client.find(id, :conditions => "firm_id = #{id}")</tt>)
P
Pratik Naik 已提交
873
      # * <tt>Firm#clients.exists?(:name => 'ACME')</tt> (similar to <tt>Client.exists?(:name => 'ACME', :firm_id => firm.id)</tt>)
D
Initial  
David Heinemeier Hansson 已提交
874
      # * <tt>Firm#clients.build</tt> (similar to <tt>Client.new("firm_id" => id)</tt>)
D
David Heinemeier Hansson 已提交
875
      # * <tt>Firm#clients.create</tt> (similar to <tt>c = Client.new("firm_id" => id); c.save; c</tt>)
D
Initial  
David Heinemeier Hansson 已提交
876
      # The declaration can also include an options hash to specialize the behavior of the association.
877
      #
P
Pratik Naik 已提交
878 879 880
      # === Supported options
      # [:class_name]
      #   Specify the class name of the association. Use it only if that name can't be inferred
881 882
      #   from the association name. So <tt>has_many :products</tt> will by default be linked
      #   to the Product class, but if the real class name is SpecialProduct, you'll have to
883
      #   specify it with this option.
P
Pratik Naik 已提交
884 885
      # [:conditions]
      #   Specify the conditions that the associated objects must meet in order to be included as a +WHERE+
886 887 888
      #   SQL fragment, such as <tt>price > 5 AND name LIKE 'B%'</tt>.  Record creations from
      #   the association are scoped if a hash is used.
      #   <tt>has_many :posts, :conditions => {:published => true}</tt> will create published
889
      #   posts with <tt>@blog.posts.create</tt> or <tt>@blog.posts.build</tt>.
P
Pratik Naik 已提交
890 891
      # [:order]
      #   Specify the order in which the associated objects are returned as an <tt>ORDER BY</tt> SQL fragment,
P
Pratik Naik 已提交
892
      #   such as <tt>last_name, first_name DESC</tt>.
P
Pratik Naik 已提交
893 894
      # [:foreign_key]
      #   Specify the foreign key used for the association. By default this is guessed to be the name
895
      #   of this class in lower-case and "_id" suffixed. So a Person class that makes a +has_many+
896
      #   association will use "person_id" as the default <tt>:foreign_key</tt>.
P
Pratik Naik 已提交
897 898 899 900
      # [:primary_key]
      #   Specify the method that returns the primary key used for the association. By default this is +id+.
      # [:dependent]
      #   If set to <tt>:destroy</tt> all the associated objects are destroyed
P
Pratik Naik 已提交
901 902
      #   alongside this object by calling their +destroy+ method.  If set to <tt>:delete_all</tt> all associated
      #   objects are deleted *without* calling their +destroy+ method.  If set to <tt>:nullify</tt> all associated
903 904 905 906 907
      #   objects' foreign keys are set to +NULL+ *without* calling their +save+ callbacks. If set to
      #   <tt>:restrict</tt> this object cannot be deleted if it has any associated object.
      #
      #   *Warning:* This option is ignored when used with <tt>:through</tt> option.
      #
P
Pratik Naik 已提交
908 909
      # [:finder_sql]
      #   Specify a complete SQL statement to fetch the association. This is a good way to go for complex
910
      #   associations that depend on multiple tables. Note: When this option is used, +find_in_collection+
911
      #   is _not_ added.
P
Pratik Naik 已提交
912 913
      # [:counter_sql]
      #   Specify a complete SQL statement to fetch the size of the association. If <tt>:finder_sql</tt> is
914
      #   specified but not <tt>:counter_sql</tt>, <tt>:counter_sql</tt> will be generated by
915
      #   replacing <tt>SELECT ... FROM</tt> with <tt>SELECT COUNT(*) FROM</tt>.
P
Pratik Naik 已提交
916 917 918 919 920 921
      # [:extend]
      #   Specify a named module for extending the proxy. See "Association extensions".
      # [:include]
      #   Specify second-order associations that should be eager loaded when the collection is loaded.
      # [:group]
      #   An attribute name by which the result should be grouped. Uses the <tt>GROUP BY</tt> SQL-clause.
922
      # [:having]
923
      #   Combined with +:group+ this can be used to filter the records that a <tt>GROUP BY</tt>
924
      #   returns. Uses the <tt>HAVING</tt> SQL-clause.
P
Pratik Naik 已提交
925 926 927
      # [:limit]
      #   An integer determining the limit on the number of rows that should be returned.
      # [:offset]
928
      #   An integer determining the offset from where the rows should be fetched. So at 5,
929
      #   it would skip the first 4 rows.
P
Pratik Naik 已提交
930
      # [:select]
931 932
      #   By default, this is <tt>*</tt> as in <tt>SELECT * FROM</tt>, but can be changed if
      #   you, for example, want to do a join but not include the joined columns. Do not forget
933
      #   to include the primary and foreign keys, otherwise it will raise an error.
P
Pratik Naik 已提交
934 935 936
      # [:as]
      #   Specifies a polymorphic interface (See <tt>belongs_to</tt>).
      # [:through]
937 938 939 940 941
      #   Specifies a join model through which to perform the query.  Options for <tt>:class_name</tt>
      #   and <tt>:foreign_key</tt> are ignored, as the association uses the source reflection. You
      #   can only use a <tt>:through</tt> query through a <tt>belongs_to</tt>, <tt>has_one</tt>
      #   or <tt>has_many</tt> association on the join model. The collection of join models
      #   can be managed via the collection API. For example, new join models are created for
942
      #   newly associated objects, and if some are gone their rows are deleted (directly,
943
      #   no destroy callbacks are triggered).
P
Pratik Naik 已提交
944
      # [:source]
945 946
      #   Specifies the source association name used by <tt>has_many :through</tt> queries.
      #   Only use it if the name cannot be inferred from the association.
947
      #   <tt>has_many :subscribers, :through => :subscriptions</tt> will look for either <tt>:subscribers</tt> or
P
Pratik Naik 已提交
948
      #   <tt>:subscriber</tt> on Subscription, unless a <tt>:source</tt> is given.
P
Pratik Naik 已提交
949 950
      # [:source_type]
      #   Specifies type of the source association used by <tt>has_many :through</tt> queries where the source
951
      #   association is a polymorphic +belongs_to+.
P
Pratik Naik 已提交
952 953 954 955 956
      # [:uniq]
      #   If true, duplicates will be omitted from the collection. Useful in conjunction with <tt>:through</tt>.
      # [:readonly]
      #   If true, all the associated objects are readonly through the association.
      # [:validate]
957
      #   If +false+, don't validate the associated objects when saving the parent object. true by default.
958
      # [:autosave]
959
      #   If true, always save the associated objects or destroy them if marked for destruction,
960
      #   when saving the parent object. If false, never save or destroy the associated objects.
961
      #   By default, only save associated objects that are new records.
P
Pratik Naik 已提交
962
      # [:inverse_of]
963 964
      #   Specifies the name of the <tt>belongs_to</tt> association on the associated object
      #   that is the inverse of this <tt>has_many</tt> association. Does not work in combination
965
      #   with <tt>:through</tt> or <tt>:as</tt> options.
966
      #   See ActiveRecord::Associations::ClassMethods's overview on Bi-directional associations for more detail.
967
      #
D
Initial  
David Heinemeier Hansson 已提交
968 969
      # Option examples:
      #   has_many :comments, :order => "posted_on"
970
      #   has_many :comments, :include => :author
D
Initial  
David Heinemeier Hansson 已提交
971
      #   has_many :people, :class_name => "Person", :conditions => "deleted = 0", :order => "name"
972 973
      #   has_many :tracks, :order => "position", :dependent => :destroy
      #   has_many :comments, :dependent => :nullify
R
Rick Olson 已提交
974
      #   has_many :tags, :as => :taggable
975
      #   has_many :reports, :readonly => true
976
      #   has_many :subscribers, :through => :subscriptions, :source => :user
D
Initial  
David Heinemeier Hansson 已提交
977 978 979 980 981
      #   has_many :subscribers, :class_name => "Person", :finder_sql =>
      #       'SELECT DISTINCT people.* ' +
      #       'FROM people p, post_subscriptions ps ' +
      #       'WHERE ps.post_id = #{id} AND ps.person_id = p.id ' +
      #       'ORDER BY p.first_name'
982
      def has_many(association_id, options = {}, &extension)
983 984
        reflection = create_has_many_reflection(association_id, options, &extension)
        configure_dependency_for_has_many(reflection)
985 986
        add_association_callbacks(reflection.name, reflection.options)

987
        if options[:through]
988
          collection_accessor_methods(reflection, HasManyThroughAssociation)
989 990
        else
          collection_accessor_methods(reflection, HasManyAssociation)
D
Initial  
David Heinemeier Hansson 已提交
991 992 993
        end
      end

P
Pratik Naik 已提交
994 995 996 997 998 999 1000 1001 1002 1003 1004
      # Specifies a one-to-one association with another class. This method should only be used
      # if the other class contains the foreign key. If the current class contains the foreign key,
      # then you should use +belongs_to+ instead. See also ActiveRecord::Associations::ClassMethods's overview
      # on when to use has_one and when to use belongs_to.
      #
      # The following methods for retrieval and query of a single associated object will be added:
      #
      # [association(force_reload = false)]
      #   Returns the associated object. +nil+ is returned if none is found.
      # [association=(associate)]
      #   Assigns the associate object, extracts the primary key, sets it as the foreign key,
D
Initial  
David Heinemeier Hansson 已提交
1005
      #   and saves the associate object.
P
Pratik Naik 已提交
1006 1007 1008 1009 1010 1011 1012 1013 1014 1015 1016 1017
      # [build_association(attributes = {})]
      #   Returns a new object of the associated type that has been instantiated
      #   with +attributes+ and linked to this object through a foreign key, but has not
      #   yet been saved. <b>Note:</b> This ONLY works if an association already exists.
      #   It will NOT work if the association is +nil+.
      # [create_association(attributes = {})]
      #   Returns a new object of the associated type that has been instantiated
      #   with +attributes+, linked to this object through a foreign key, and that
      #   has already been saved (if it passed the validation).
      #
      # (+association+ is replaced with the symbol passed as the first argument, so
      # <tt>has_one :manager</tt> would add among others <tt>manager.nil?</tt>.)
D
Initial  
David Heinemeier Hansson 已提交
1018
      #
P
Pratik Naik 已提交
1019 1020 1021
      # === Example
      #
      # An Account class declares <tt>has_one :beneficiary</tt>, which will add:
1022
      # * <tt>Account#beneficiary</tt> (similar to <tt>Beneficiary.find(:first, :conditions => "account_id = #{id}")</tt>)
D
Initial  
David Heinemeier Hansson 已提交
1023
      # * <tt>Account#beneficiary=(beneficiary)</tt> (similar to <tt>beneficiary.account_id = account.id; beneficiary.save</tt>)
1024 1025 1026
      # * <tt>Account#build_beneficiary</tt> (similar to <tt>Beneficiary.new("account_id" => id)</tt>)
      # * <tt>Account#create_beneficiary</tt> (similar to <tt>b = Beneficiary.new("account_id" => id); b.save; b</tt>)
      #
P
Pratik Naik 已提交
1027 1028
      # === Options
      #
D
Initial  
David Heinemeier Hansson 已提交
1029
      # The declaration can also include an options hash to specialize the behavior of the association.
1030
      #
D
Initial  
David Heinemeier Hansson 已提交
1031
      # Options are:
P
Pratik Naik 已提交
1032 1033
      # [:class_name]
      #   Specify the class name of the association. Use it only if that name can't be inferred
P
Pratik Naik 已提交
1034 1035
      #   from the association name. So <tt>has_one :manager</tt> will by default be linked to the Manager class, but
      #   if the real class name is Person, you'll have to specify it with this option.
P
Pratik Naik 已提交
1036 1037
      # [:conditions]
      #   Specify the conditions that the associated object must meet in order to be included as a +WHERE+
1038
      #   SQL fragment, such as <tt>rank = 5</tt>. Record creation from the association is scoped if a hash
1039
      #   is used. <tt>has_one :account, :conditions => {:enabled => true}</tt> will create
1040
      #   an enabled account with <tt>@company.create_account</tt> or <tt>@company.build_account</tt>.
P
Pratik Naik 已提交
1041 1042
      # [:order]
      #   Specify the order in which the associated objects are returned as an <tt>ORDER BY</tt> SQL fragment,
P
Pratik Naik 已提交
1043
      #   such as <tt>last_name, first_name DESC</tt>.
P
Pratik Naik 已提交
1044 1045
      # [:dependent]
      #   If set to <tt>:destroy</tt>, the associated object is destroyed when this object is. If set to
1046 1047
      #   <tt>:delete</tt>, the associated object is deleted *without* calling its destroy method.
      #   If set to <tt>:nullify</tt>, the associated object's foreign key is set to +NULL+.
1048
      #   Also, association is assigned.
P
Pratik Naik 已提交
1049 1050
      # [:foreign_key]
      #   Specify the foreign key used for the association. By default this is guessed to be the name
1051
      #   of this class in lower-case and "_id" suffixed. So a Person class that makes a +has_one+ association
1052
      #   will use "person_id" as the default <tt>:foreign_key</tt>.
P
Pratik Naik 已提交
1053 1054 1055 1056 1057 1058 1059
      # [:primary_key]
      #   Specify the method that returns the primary key used for the association. By default this is +id+.
      # [:include]
      #   Specify second-order associations that should be eager loaded when this object is loaded.
      # [:as]
      #   Specifies a polymorphic interface (See <tt>belongs_to</tt>).
      # [:select]
1060 1061
      #   By default, this is <tt>*</tt> as in <tt>SELECT * FROM</tt>, but can be changed if, for example,
      #   you want to do a join but not include the joined columns. Do not forget to include the
1062
      #   primary and foreign keys, otherwise it will raise an error.
P
Pratik Naik 已提交
1063
      # [:through]
1064 1065 1066
      #   Specifies a Join Model through which to perform the query.  Options for <tt>:class_name</tt>
      #   and <tt>:foreign_key</tt> are ignored, as the association uses the source reflection. You
      #   can only use a <tt>:through</tt> query through a <tt>has_one</tt> or <tt>belongs_to</tt>
1067
      #   association on the join model.
P
Pratik Naik 已提交
1068
      # [:source]
1069 1070
      #   Specifies the source association name used by <tt>has_one :through</tt> queries.
      #   Only use it if the name cannot be inferred from the association.
1071
      #   <tt>has_one :favorite, :through => :favorites</tt> will look for a
1072
      #   <tt>:favorite</tt> on Favorite, unless a <tt>:source</tt> is given.
P
Pratik Naik 已提交
1073 1074
      # [:source_type]
      #   Specifies type of the source association used by <tt>has_one :through</tt> queries where the source
1075
      #   association is a polymorphic +belongs_to+.
P
Pratik Naik 已提交
1076 1077 1078
      # [:readonly]
      #   If true, the associated object is readonly through the association.
      # [:validate]
1079
      #   If +false+, don't validate the associated object when saving the parent object. +false+ by default.
1080
      # [:autosave]
1081
      #   If true, always save the associated object or destroy it if marked for destruction,
1082
      #   when saving the parent object. If false, never save or destroy the associated object.
1083
      #   By default, only save the associated object if it's a new record.
P
Pratik Naik 已提交
1084
      # [:inverse_of]
1085 1086
      #   Specifies the name of the <tt>belongs_to</tt> association on the associated object
      #   that is the inverse of this <tt>has_one</tt> association.  Does not work in combination
1087
      #   with <tt>:through</tt> or <tt>:as</tt> options.
1088
      #   See ActiveRecord::Associations::ClassMethods's overview on Bi-directional associations for more detail.
1089
      #
D
Initial  
David Heinemeier Hansson 已提交
1090
      # Option examples:
1091
      #   has_one :credit_card, :dependent => :destroy  # destroys the associated credit card
1092
      #   has_one :credit_card, :dependent => :nullify  # updates the associated records foreign
1093
      #                                                 # key value to NULL rather than destroying it
D
Initial  
David Heinemeier Hansson 已提交
1094 1095
      #   has_one :last_comment, :class_name => "Comment", :order => "posted_on"
      #   has_one :project_manager, :class_name => "Person", :conditions => "role = 'project_manager'"
1096
      #   has_one :attachment, :as => :attachable
1097
      #   has_one :boss, :readonly => :true
1098 1099
      #   has_one :club, :through => :membership
      #   has_one :primary_address, :through => :addressables, :conditions => ["addressable.primary = ?", true], :source => :addressable
D
Initial  
David Heinemeier Hansson 已提交
1100
      def has_one(association_id, options = {})
1101 1102 1103 1104 1105 1106 1107 1108 1109 1110
        if options[:through]
          reflection = create_has_one_through_reflection(association_id, options)
          association_accessor_methods(reflection, ActiveRecord::Associations::HasOneThroughAssociation)
        else
          reflection = create_has_one_reflection(association_id, options)
          association_accessor_methods(reflection, HasOneAssociation)
          association_constructor_method(:build,  reflection, HasOneAssociation)
          association_constructor_method(:create, reflection, HasOneAssociation)
          configure_dependency_for_has_one(reflection)
        end
D
Initial  
David Heinemeier Hansson 已提交
1111 1112
      end

P
Pratik Naik 已提交
1113 1114 1115 1116 1117 1118 1119 1120 1121 1122 1123 1124 1125 1126
      # Specifies a one-to-one association with another class. This method should only be used
      # if this class contains the foreign key. If the other class contains the foreign key,
      # then you should use +has_one+ instead. See also ActiveRecord::Associations::ClassMethods's overview
      # on when to use +has_one+ and when to use +belongs_to+.
      #
      # Methods will be added for retrieval and query for a single associated object, for which
      # this object holds an id:
      #
      # [association(force_reload = false)]
      #   Returns the associated object. +nil+ is returned if none is found.
      # [association=(associate)]
      #   Assigns the associate object, extracts the primary key, and sets it as the foreign key.
      # [build_association(attributes = {})]
      #   Returns a new object of the associated type that has been instantiated
1127
      #   with +attributes+ and linked to this object through a foreign key, but has not yet been saved.
P
Pratik Naik 已提交
1128 1129 1130 1131 1132 1133 1134
      # [create_association(attributes = {})]
      #   Returns a new object of the associated type that has been instantiated
      #   with +attributes+, linked to this object through a foreign key, and that
      #   has already been saved (if it passed the validation).
      #
      # (+association+ is replaced with the symbol passed as the first argument, so
      # <tt>belongs_to :author</tt> would add among others <tt>author.nil?</tt>.)
D
Initial  
David Heinemeier Hansson 已提交
1135
      #
P
Pratik Naik 已提交
1136 1137 1138
      # === Example
      #
      # A Post class declares <tt>belongs_to :author</tt>, which will add:
D
Initial  
David Heinemeier Hansson 已提交
1139 1140
      # * <tt>Post#author</tt> (similar to <tt>Author.find(author_id)</tt>)
      # * <tt>Post#author=(author)</tt> (similar to <tt>post.author_id = author.id</tt>)
D
David Heinemeier Hansson 已提交
1141 1142
      # * <tt>Post#build_author</tt> (similar to <tt>post.author = Author.new</tt>)
      # * <tt>Post#create_author</tt> (similar to <tt>post.author = Author.new; post.author.save; post.author</tt>)
D
Initial  
David Heinemeier Hansson 已提交
1143
      # The declaration can also include an options hash to specialize the behavior of the association.
1144
      #
P
Pratik Naik 已提交
1145 1146 1147 1148
      # === Options
      #
      # [:class_name]
      #   Specify the class name of the association. Use it only if that name can't be inferred
P
Pratik Naik 已提交
1149 1150
      #   from the association name. So <tt>has_one :author</tt> will by default be linked to the Author class, but
      #   if the real class name is Person, you'll have to specify it with this option.
P
Pratik Naik 已提交
1151 1152
      # [:conditions]
      #   Specify the conditions that the associated object must meet in order to be included as a +WHERE+
1153
      #   SQL fragment, such as <tt>authorized = 1</tt>.
P
Pratik Naik 已提交
1154
      # [:select]
1155 1156
      #   By default, this is <tt>*</tt> as in <tt>SELECT * FROM</tt>, but can be changed
      #   if, for example, you want to do a join but not include the joined columns. Do not
1157
      #   forget to include the primary and foreign keys, otherwise it will raise an error.
P
Pratik Naik 已提交
1158 1159
      # [:foreign_key]
      #   Specify the foreign key used for the association. By default this is guessed to be the name
1160 1161 1162
      #   of the association with an "_id" suffix. So a class that defines a <tt>belongs_to :person</tt>
      #   association will use "person_id" as the default <tt>:foreign_key</tt>. Similarly,
      #   <tt>belongs_to :favorite_person, :class_name => "Person"</tt> will use a foreign key
1163
      #   of "favorite_person_id".
1164
      # [:primary_key]
1165
      #   Specify the method that returns the primary key of associated object used for the association.
1166
      #   By default this is id.
P
Pratik Naik 已提交
1167 1168
      # [:dependent]
      #   If set to <tt>:destroy</tt>, the associated object is destroyed when this object is. If set to
1169 1170
      #   <tt>:delete</tt>, the associated object is deleted *without* calling its destroy method.
      #   This option should not be specified when <tt>belongs_to</tt> is used in conjunction with
1171
      #   a <tt>has_many</tt> relationship on another class because of the potential to leave
1172
      #   orphaned records behind.
P
Pratik Naik 已提交
1173 1174
      # [:counter_cache]
      #   Caches the number of belonging objects on the associate class through the use of +increment_counter+
1175 1176
      #   and +decrement_counter+. The counter cache is incremented when an object of this
      #   class is created and decremented when it's destroyed. This requires that a column
1177
      #   named <tt>#{table_name}_count</tt> (such as +comments_count+ for a belonging Comment class)
1178 1179
      #   is used on the associate class (such as a Post class). You can also specify a custom counter
      #   cache column by providing a column name instead of a +true+/+false+ value to this
1180
      #   option (e.g., <tt>:counter_cache => :my_custom_counter</tt>.)
1181
      #   Note: Specifying a counter cache will add it to that model's list of readonly attributes
1182
      #   using +attr_readonly+.
P
Pratik Naik 已提交
1183 1184 1185 1186
      # [:include]
      #   Specify second-order associations that should be eager loaded when this object is loaded.
      # [:polymorphic]
      #   Specify this association is a polymorphic association by passing +true+.
1187
      #   Note: If you've enabled the counter cache, then you may want to add the counter cache attribute
P
Pratik Naik 已提交
1188
      #   to the +attr_readonly+ list in the associated classes (e.g. <tt>class Post; attr_readonly :comments_count; end</tt>).
P
Pratik Naik 已提交
1189 1190 1191
      # [:readonly]
      #   If true, the associated object is readonly through the association.
      # [:validate]
1192
      #   If +false+, don't validate the associated objects when saving the parent object. +false+ by default.
1193
      # [:autosave]
1194
      #   If true, always save the associated object or destroy it if marked for destruction, when
1195
      #   saving the parent object.
1196 1197
      #   If false, never save or destroy the associated object.
      #   By default, only save the associated object if it's a new record.
1198
      # [:touch]
1199 1200
      #   If true, the associated object will be touched (the updated_at/on attributes set to now)
      #   when this record is either saved or destroyed. If you specify a symbol, that attribute
1201
      #   will be updated with the current time instead of the updated_at/on attribute.
P
Pratik Naik 已提交
1202
      # [:inverse_of]
1203 1204
      #   Specifies the name of the <tt>has_one</tt> or <tt>has_many</tt> association on the associated
      #   object that is the inverse of this <tt>belongs_to</tt> association.  Does not work in
1205
      #   combination with the <tt>:polymorphic</tt> options.
1206
      #   See ActiveRecord::Associations::ClassMethods's overview on Bi-directional associations for more detail.
D
Initial  
David Heinemeier Hansson 已提交
1207 1208 1209
      #
      # Option examples:
      #   belongs_to :firm, :foreign_key => "client_of"
1210
      #   belongs_to :person, :primary_key => "name", :foreign_key => "person_name"
D
Initial  
David Heinemeier Hansson 已提交
1211
      #   belongs_to :author, :class_name => "Person", :foreign_key => "author_id"
1212
      #   belongs_to :valid_coupon, :class_name => "Coupon", :foreign_key => "coupon_id",
D
Initial  
David Heinemeier Hansson 已提交
1213
      #              :conditions => 'discounts > #{payments_count}'
R
Rick Olson 已提交
1214
      #   belongs_to :attachable, :polymorphic => true
1215
      #   belongs_to :project, :readonly => true
1216
      #   belongs_to :post, :counter_cache => true
1217 1218
      #   belongs_to :company, :touch => true
      #   belongs_to :company, :touch => :employees_last_updated_at
1219
      def belongs_to(association_id, options = {})
1220
        reflection = create_belongs_to_reflection(association_id, options)
1221

1222 1223
        if reflection.options[:polymorphic]
          association_accessor_methods(reflection, BelongsToPolymorphicAssociation)
1224
        else
1225 1226 1227
          association_accessor_methods(reflection, BelongsToAssociation)
          association_constructor_method(:build,  reflection, BelongsToAssociation)
          association_constructor_method(:create, reflection, BelongsToAssociation)
1228
        end
1229

1230 1231
        add_counter_cache_callbacks(reflection)          if options[:counter_cache]
        add_touch_callbacks(reflection, options[:touch]) if options[:touch]
1232 1233

        configure_dependency_for_belongs_to(reflection)
1234 1235
      end

P
Pratik Naik 已提交
1236 1237 1238
      # Specifies a many-to-many relationship with another class. This associates two classes via an
      # intermediate join table.  Unless the join table is explicitly specified as an option, it is
      # guessed using the lexical order of the class names. So a join between Developer and Project
1239 1240 1241
      # will give the default join table name of "developers_projects" because "D" outranks "P".
      # Note that this precedence is calculated using the <tt><</tt> operator for String.  This
      # means that if the strings are of different lengths, and the strings are equal when compared
1242
      # up to the shortest length, then the longer string is considered of higher
P
Pratik Naik 已提交
1243 1244 1245 1246
      # lexical precedence than the shorter one.  For example, one would expect the tables "paper_boxes" and "papers"
      # to generate a join table name of "papers_paper_boxes" because of the length of the name "paper_boxes",
      # but it in fact generates a join table name of "paper_boxes_papers".  Be aware of this caveat, and use the
      # custom <tt>:join_table</tt> option if you need to.
D
Initial  
David Heinemeier Hansson 已提交
1247
      #
1248 1249 1250 1251 1252 1253 1254 1255 1256 1257 1258 1259 1260 1261 1262 1263
      # The join table should not have a primary key or a model associated with it. You must manually generate the
      # join table with a migration such as this:
      #
      #   class CreateDevelopersProjectsJoinTable < ActiveRecord::Migration
      #     def self.up
      #       create_table :developers_projects, :id => false do |t|
      #         t.integer :developer_id
      #         t.integer :project_id
      #       end
      #     end
      #
      #     def self.down
      #       drop_table :developers_projects
      #     end
      #   end
      #
1264 1265 1266
      # Deprecated: Any additional fields added to the join table will be placed as attributes when
      # pulling records out through +has_and_belongs_to_many+ associations. Records returned from join
      # tables with additional attributes will be marked as readonly (because we can't save changes
1267
      # to the additional attributes). It's strongly recommended that you upgrade any
1268
      # associations with attributes to a real join model (see introduction).
D
Initial  
David Heinemeier Hansson 已提交
1269
      #
1270
      # Adds the following methods for retrieval and query:
P
Pratik Naik 已提交
1271 1272 1273
      #
      # [collection(force_reload = false)]
      #   Returns an array of all the associated objects.
1274
      #   An empty array is returned if none are found.
P
Pratik Naik 已提交
1275 1276
      # [collection<<(object, ...)]
      #   Adds one or more objects to the collection by creating associations in the join table
1277
      #   (<tt>collection.push</tt> and <tt>collection.concat</tt> are aliases to this method).
1278
      #   Note that this operation instantly fires update sql without waiting for the save or update call on the
1279
      #   parent object.
P
Pratik Naik 已提交
1280 1281
      # [collection.delete(object, ...)]
      #   Removes one or more objects from the collection by removing their associations from the join table.
D
Initial  
David Heinemeier Hansson 已提交
1282
      #   This does not destroy the objects.
P
Pratik Naik 已提交
1283 1284 1285 1286 1287 1288 1289 1290 1291 1292 1293 1294 1295 1296
      # [collection=objects]
      #   Replaces the collection's content by deleting and adding objects as appropriate.
      # [collection_singular_ids]
      #   Returns an array of the associated objects' ids.
      # [collection_singular_ids=ids]
      #   Replace the collection by the objects identified by the primary keys in +ids+.
      # [collection.clear]
      #   Removes every object from the collection. This does not destroy the objects.
      # [collection.empty?]
      #   Returns +true+ if there are no associated objects.
      # [collection.size]
      #   Returns the number of associated objects.
      # [collection.find(id)]
      #   Finds an associated object responding to the +id+ and that
1297
      #   meets the condition that it has to be associated with this object.
P
Pratik Naik 已提交
1298
      #   Uses the same rules as ActiveRecord::Base.find.
P
Pratik Naik 已提交
1299
      # [collection.exists?(...)]
P
Pratik Naik 已提交
1300 1301 1302 1303
      #   Checks whether an associated object with the given conditions exists.
      #   Uses the same rules as ActiveRecord::Base.exists?.
      # [collection.build(attributes = {})]
      #   Returns a new object of the collection type that has been instantiated
1304
      #   with +attributes+ and linked to this object through the join table, but has not yet been saved.
P
Pratik Naik 已提交
1305 1306
      # [collection.create(attributes = {})]
      #   Returns a new object of the collection type that has been instantiated
1307
      #   with +attributes+, linked to this object through the join table, and that has already been
1308
      #   saved (if it passed the validation).
D
Initial  
David Heinemeier Hansson 已提交
1309
      #
P
Pratik Naik 已提交
1310 1311 1312 1313 1314 1315
      # (+collection+ is replaced with the symbol passed as the first argument, so
      # <tt>has_and_belongs_to_many :categories</tt> would add among others <tt>categories.empty?</tt>.)
      #
      # === Example
      #
      # A Developer class declares <tt>has_and_belongs_to_many :projects</tt>, which will add:
D
Initial  
David Heinemeier Hansson 已提交
1316 1317 1318
      # * <tt>Developer#projects</tt>
      # * <tt>Developer#projects<<</tt>
      # * <tt>Developer#projects.delete</tt>
1319
      # * <tt>Developer#projects=</tt>
1320
      # * <tt>Developer#project_ids</tt>
1321
      # * <tt>Developer#project_ids=</tt>
D
Initial  
David Heinemeier Hansson 已提交
1322 1323 1324 1325
      # * <tt>Developer#projects.clear</tt>
      # * <tt>Developer#projects.empty?</tt>
      # * <tt>Developer#projects.size</tt>
      # * <tt>Developer#projects.find(id)</tt>
1326
      # * <tt>Developer#projects.exists?(...)</tt>
1327 1328
      # * <tt>Developer#projects.build</tt> (similar to <tt>Project.new("project_id" => id)</tt>)
      # * <tt>Developer#projects.create</tt> (similar to <tt>c = Project.new("project_id" => id); c.save; c</tt>)
D
Initial  
David Heinemeier Hansson 已提交
1329
      # The declaration may include an options hash to specialize the behavior of the association.
1330
      #
P
Pratik Naik 已提交
1331 1332 1333 1334
      # === Options
      #
      # [:class_name]
      #   Specify the class name of the association. Use it only if that name can't be inferred
1335
      #   from the association name. So <tt>has_and_belongs_to_many :projects</tt> will by default be linked to the
P
Pratik Naik 已提交
1336
      #   Project class, but if the real class name is SuperProject, you'll have to specify it with this option.
P
Pratik Naik 已提交
1337 1338 1339 1340 1341 1342
      # [:join_table]
      #   Specify the name of the join table if the default based on lexical order isn't what you want.
      #   <b>WARNING:</b> If you're overwriting the table name of either class, the +table_name+ method
      #   MUST be declared underneath any +has_and_belongs_to_many+ declaration in order to work.
      # [:foreign_key]
      #   Specify the foreign key used for the association. By default this is guessed to be the name
1343 1344
      #   of this class in lower-case and "_id" suffixed. So a Person class that makes
      #   a +has_and_belongs_to_many+ association to Project will use "person_id" as the
1345
      #   default <tt>:foreign_key</tt>.
P
Pratik Naik 已提交
1346
      # [:association_foreign_key]
P
Pratik Naik 已提交
1347 1348 1349 1350
      #   Specify the foreign key used for the association on the receiving side of the association.
      #   By default this is guessed to be the name of the associated class in lower-case and "_id" suffixed.
      #   So if a Person class makes a +has_and_belongs_to_many+ association to Project,
      #   the association will use "project_id" as the default <tt>:association_foreign_key</tt>.
P
Pratik Naik 已提交
1351 1352
      # [:conditions]
      #   Specify the conditions that the associated object must meet in order to be included as a +WHERE+
1353
      #   SQL fragment, such as <tt>authorized = 1</tt>.  Record creations from the association are
1354
      #   scoped if a hash is used.
1355
      #   <tt>has_many :posts, :conditions => {:published => true}</tt> will create published posts with <tt>@blog.posts.create</tt>
1356
      #   or <tt>@blog.posts.build</tt>.
P
Pratik Naik 已提交
1357 1358
      # [:order]
      #   Specify the order in which the associated objects are returned as an <tt>ORDER BY</tt> SQL fragment,
1359
      #   such as <tt>last_name, first_name DESC</tt>
P
Pratik Naik 已提交
1360 1361 1362 1363
      # [:uniq]
      #   If true, duplicate associated objects will be ignored by accessors and query methods.
      # [:finder_sql]
      #   Overwrite the default generated SQL statement used to fetch the association with a manual statement
1364 1365
      # [:counter_sql]
      #   Specify a complete SQL statement to fetch the size of the association. If <tt>:finder_sql</tt> is
1366
      #   specified but not <tt>:counter_sql</tt>, <tt>:counter_sql</tt> will be generated by
1367
      #   replacing <tt>SELECT ... FROM</tt> with <tt>SELECT COUNT(*) FROM</tt>.
P
Pratik Naik 已提交
1368 1369
      # [:delete_sql]
      #   Overwrite the default generated SQL statement used to remove links between the associated
P
Pratik Naik 已提交
1370
      #   classes with a manual statement.
P
Pratik Naik 已提交
1371 1372
      # [:insert_sql]
      #   Overwrite the default generated SQL statement used to add links between the associated classes
P
Pratik Naik 已提交
1373
      #   with a manual statement.
P
Pratik Naik 已提交
1374 1375 1376 1377 1378 1379
      # [:extend]
      #   Anonymous module for extending the proxy, see "Association extensions".
      # [:include]
      #   Specify second-order associations that should be eager loaded when the collection is loaded.
      # [:group]
      #   An attribute name by which the result should be grouped. Uses the <tt>GROUP BY</tt> SQL-clause.
1380
      # [:having]
1381
      #   Combined with +:group+ this can be used to filter the records that a <tt>GROUP BY</tt> returns.
1382
      #   Uses the <tt>HAVING</tt> SQL-clause.
P
Pratik Naik 已提交
1383 1384 1385
      # [:limit]
      #   An integer determining the limit on the number of rows that should be returned.
      # [:offset]
1386
      #   An integer determining the offset from where the rows should be fetched. So at 5,
1387
      #   it would skip the first 4 rows.
P
Pratik Naik 已提交
1388
      # [:select]
1389 1390
      #   By default, this is <tt>*</tt> as in <tt>SELECT * FROM</tt>, but can be changed if, for example,
      #   you want to do a join but not include the joined columns. Do not forget to include the primary
1391
      #   and foreign keys, otherwise it will raise an error.
P
Pratik Naik 已提交
1392 1393 1394
      # [:readonly]
      #   If true, all the associated objects are readonly through the association.
      # [:validate]
1395
      #   If +false+, don't validate the associated objects when saving the parent object. +true+ by default.
1396
      # [:autosave]
1397
      #   If true, always save the associated objects or destroy them if marked for destruction, when
1398
      #   saving the parent object.
1399 1400
      #   If false, never save or destroy the associated objects.
      #   By default, only save associated objects that are new records.
D
Initial  
David Heinemeier Hansson 已提交
1401 1402 1403
      #
      # Option examples:
      #   has_and_belongs_to_many :projects
1404
      #   has_and_belongs_to_many :projects, :include => [ :milestones, :manager ]
D
Initial  
David Heinemeier Hansson 已提交
1405 1406
      #   has_and_belongs_to_many :nations, :class_name => "Country"
      #   has_and_belongs_to_many :categories, :join_table => "prods_cats"
1407
      #   has_and_belongs_to_many :categories, :readonly => true
1408
      #   has_and_belongs_to_many :active_projects, :join_table => 'developers_projects', :delete_sql =>
1409
      #   'DELETE FROM developers_projects WHERE active=1 AND developer_id = #{id} AND project_id = #{record.id}'
1410
      def has_and_belongs_to_many(association_id, options = {}, &extension)
1411 1412
        reflection = create_has_and_belongs_to_many_reflection(association_id, options, &extension)
        collection_accessor_methods(reflection, HasAndBelongsToManyAssociation)
D
Initial  
David Heinemeier Hansson 已提交
1413

1414 1415
        # Don't use a before_destroy callback since users' before_destroy
        # callbacks will be executed after the association is wiped out.
1416 1417 1418 1419 1420 1421 1422 1423
        include Module.new {
          class_eval <<-RUBY, __FILE__, __LINE__ + 1
            def destroy                     # def destroy
              super                         #   super
              #{reflection.name}.clear      #   posts.clear
            end                             # end
          RUBY
        }
1424

1425
        add_association_callbacks(reflection.name, options)
D
Initial  
David Heinemeier Hansson 已提交
1426 1427 1428
      end

      private
P
Pratik Naik 已提交
1429
        # Generates a join table name from two provided table names.
P
Pratik Naik 已提交
1430
        # The names in the join table names end up in lexicographic order.
P
Pratik Naik 已提交
1431 1432 1433
        #
        #   join_table_name("members", "clubs")         # => "clubs_members"
        #   join_table_name("members", "special_clubs") # => "members_special_clubs"
D
Initial  
David Heinemeier Hansson 已提交
1434 1435 1436 1437 1438 1439 1440 1441 1442
        def join_table_name(first_table_name, second_table_name)
          if first_table_name < second_table_name
            join_table = "#{first_table_name}_#{second_table_name}"
          else
            join_table = "#{second_table_name}_#{first_table_name}"
          end

          table_name_prefix + join_table + table_name_suffix
        end
1443

1444
        def association_accessor_methods(reflection, association_proxy_class)
1445
          redefine_method(reflection.name) do |*params|
1446
            force_reload = params.first unless params.empty?
1447
            association = association_instance_get(reflection.name)
1448

1449
            if association.nil? || force_reload
1450
              association = association_proxy_class.new(self, reflection)
1451
              retval = force_reload ? reflection.klass.uncached { association.reload } : association.reload
1452
              if retval.nil? and association_proxy_class == BelongsToAssociation
1453
                association_instance_set(reflection.name, nil)
1454 1455
                return nil
              end
1456
              association_instance_set(reflection.name, association)
1457
            end
1458 1459

            association.target.nil? ? nil : association
1460 1461
          end

1462
          redefine_method("loaded_#{reflection.name}?") do
1463
            association = association_instance_get(reflection.name)
1464 1465
            association && association.loaded?
          end
1466

1467
          redefine_method("#{reflection.name}=") do |new_value|
1468
            association = association_instance_get(reflection.name)
1469

1470
            if association.nil? || association.target != new_value
1471
              association = association_proxy_class.new(self, reflection)
1472
            end
1473

1474 1475
            association.replace(new_value)
            association_instance_set(reflection.name, new_value.nil? ? nil : association)
1476
          end
1477

1478
          redefine_method("set_#{reflection.name}_target") do |target|
1479
            return if target.nil? and association_proxy_class == BelongsToAssociation
1480
            association = association_proxy_class.new(self, reflection)
1481
            association.target = target
1482
            association_instance_set(reflection.name, association)
1483
          end
1484 1485
        end

1486
        def collection_reader_method(reflection, association_proxy_class)
1487
          redefine_method(reflection.name) do |*params|
1488
            force_reload = params.first unless params.empty?
1489
            association = association_instance_get(reflection.name)
1490

1491
            unless association
1492
              association = association_proxy_class.new(self, reflection)
1493
              association_instance_set(reflection.name, association)
D
Initial  
David Heinemeier Hansson 已提交
1494
            end
1495

1496
            reflection.klass.uncached { association.reload } if force_reload
1497

1498 1499
            association
          end
1500

1501
          redefine_method("#{reflection.name.to_s.singularize}_ids") do
1502
            if send(reflection.name).loaded? || reflection.options[:finder_sql]
1503
              send(reflection.name).map { |r| r.id }
1504
            else
1505 1506 1507
              if reflection.through_reflection && reflection.source_reflection.belongs_to?
                through = reflection.through_reflection
                primary_key = reflection.source_reflection.primary_key_name
1508
                send(through.name).select("DISTINCT #{through.quoted_table_name}.#{primary_key}").map! { |r| r.send(primary_key) }
1509
              else
1510
                send(reflection.name).select("#{reflection.quoted_table_name}.#{reflection.klass.primary_key}").except(:includes).map! { |r| r.id }
1511
              end
1512
            end
1513
          end
1514

1515
        end
D
Initial  
David Heinemeier Hansson 已提交
1516

1517
        def collection_accessor_methods(reflection, association_proxy_class, writer = true)
1518 1519
          collection_reader_method(reflection, association_proxy_class)

1520
          if writer
1521
            redefine_method("#{reflection.name}=") do |new_value|
1522 1523 1524 1525 1526
              # Loads proxy class instance (defined in collection_reader_method) if not already loaded
              association = send(reflection.name)
              association.replace(new_value)
              association
            end
J
José Valim 已提交
1527

1528
            redefine_method("#{reflection.name.to_s.singularize}_ids=") do |new_value|
J
José Valim 已提交
1529
              pk_column = reflection.primary_key_column
1530
              ids = (new_value || []).reject { |nid| nid.blank? }
J
José Valim 已提交
1531
              ids.map!{ |i| pk_column.type_cast(i) }
1532
              send("#{reflection.name}=", reflection.klass.find(ids).index_by{ |r| r.id }.values_at(*ids))
1533
            end
1534
          end
D
Initial  
David Heinemeier Hansson 已提交
1535
        end
1536

1537
        def association_constructor_method(constructor, reflection, association_proxy_class)
1538
          redefine_method("#{constructor}_#{reflection.name}") do |*params|
1539 1540
            attributees      = params.first unless params.empty?
            replace_existing = params[1].nil? ? true : params[1]
1541
            association      = association_instance_get(reflection.name)
1542

1543
            unless association
1544
              association = association_proxy_class.new(self, reflection)
1545
              association_instance_set(reflection.name, association)
1546 1547
            end

1548 1549 1550 1551 1552
            if association_proxy_class == HasOneAssociation
              association.send(constructor, attributees, replace_existing)
            else
              association.send(constructor, attributees)
            end
1553 1554
          end
        end
1555

1556 1557 1558 1559 1560 1561
        def add_counter_cache_callbacks(reflection)
          cache_column = reflection.counter_cache_column

          method_name = "belongs_to_counter_cache_after_create_for_#{reflection.name}".to_sym
          define_method(method_name) do
            association = send(reflection.name)
1562
            association.class.increment_counter(cache_column, association.id) unless association.nil?
1563 1564 1565 1566 1567 1568
          end
          after_create(method_name)

          method_name = "belongs_to_counter_cache_before_destroy_for_#{reflection.name}".to_sym
          define_method(method_name) do
            association = send(reflection.name)
1569
            association.class.decrement_counter(cache_column, association.id) unless association.nil?
1570 1571 1572 1573
          end
          before_destroy(method_name)

          module_eval(
1574
            "#{reflection.class_name}.send(:attr_readonly,\"#{cache_column}\".intern) if defined?(#{reflection.class_name}) && #{reflection.class_name}.respond_to?(:attr_readonly)", __FILE__, __LINE__
1575 1576
          )
        end
1577

1578
        def add_touch_callbacks(reflection, touch_attribute)
1579 1580
          method_name = :"belongs_to_touch_after_save_or_destroy_for_#{reflection.name}"
          redefine_method(method_name) do
1581
            association = send(reflection.name)
1582

1583 1584 1585 1586 1587 1588 1589
            if touch_attribute == true
              association.touch unless association.nil?
            else
              association.touch(touch_attribute) unless association.nil?
            end
          end
          after_save(method_name)
1590
          after_touch(method_name)
1591 1592 1593
          after_destroy(method_name)
        end

1594 1595 1596
        # Creates before_destroy callback methods that nullify, delete or destroy
        # has_many associated objects, according to the defined :dependent rule.
        #
1597 1598 1599
        # See HasManyAssociation#delete_records for more information. In general
        #  - delete children if the option is set to :destroy or :delete_all
        #  - set the foreign key to NULL if the option is set to :nullify
1600
        #  - do not delete the parent record if there is any child record if the
1601
        #    option is set to :restrict
1602 1603 1604 1605
        #
        # The +extra_conditions+ parameter, which is not used within the main
        # Active Record codebase, is meant to allow plugins to define extra
        # finder conditions.
1606
        def configure_dependency_for_has_many(reflection, extra_conditions = nil)
1607 1608 1609
          if reflection.options.include?(:dependent)
            case reflection.options[:dependent]
              when :destroy
1610 1611
                method_name = "has_many_dependent_destroy_for_#{reflection.name}".to_sym
                define_method(method_name) do
1612 1613 1614 1615 1616 1617 1618 1619 1620 1621
                  send(reflection.name).each do |o|
                    # No point in executing the counter update since we're going to destroy the parent anyway
                    counter_method = ('belongs_to_counter_cache_before_destroy_for_' + self.class.name.downcase).to_sym
                    if(o.respond_to? counter_method) then
                      class << o
                        self
                      end.send(:define_method, counter_method, Proc.new {})
                    end
                    o.destroy
                  end
1622 1623
                end
                before_destroy method_name
1624
              when :delete_all
1625 1626 1627 1628 1629
                before_destroy do |record|
                  self.class.send(:delete_all_has_many_dependencies,
                  record,
                  reflection.name,
                  reflection.klass,
1630
                  reflection.dependent_conditions(record, self.class, extra_conditions))
1631
                end
1632
              when :nullify
1633 1634 1635 1636 1637 1638
                before_destroy do |record|
                  self.class.send(:nullify_has_many_dependencies,
                  record,
                  reflection.name,
                  reflection.klass,
                  reflection.primary_key_name,
1639
                  reflection.dependent_conditions(record, self.class, extra_conditions))
1640 1641 1642 1643 1644 1645
                end
              when :restrict
                method_name = "has_many_dependent_restrict_for_#{reflection.name}".to_sym
                define_method(method_name) do
                  unless send(reflection.name).empty?
                    raise DeleteRestrictionError.new(reflection)
1646
                  end
1647 1648
                end
                before_destroy method_name
1649
              else
1650
                raise ArgumentError, "The :dependent option expects either :destroy, :delete_all, :nullify or :restrict (#{reflection.options[:dependent].inspect})"
1651
            end
1652 1653
          end
        end
1654

P
Pratik Naik 已提交
1655 1656
        # Creates before_destroy callback methods that nullify, delete or destroy
        # has_one associated objects, according to the defined :dependent rule.
1657 1658
        # If the association is marked as :dependent => :restrict, create a callback
        # that prevents deleting entirely.
1659
        def configure_dependency_for_has_one(reflection)
1660
          if reflection.options.include?(:dependent)
A
Aaron Patterson 已提交
1661 1662 1663 1664 1665
            name = reflection.options[:dependent]
            method_name = :"has_one_dependent_#{name}_for_#{reflection.name}"

            case name
              when :destroy, :delete
1666 1667 1668 1669 1670 1671
                class_eval <<-eoruby, __FILE__, __LINE__ + 1
                  def #{method_name}
                    association = #{reflection.name}
                    association.#{name} if association
                  end
                eoruby
1672
              when :nullify
1673 1674 1675 1676 1677 1678
                class_eval <<-eoruby, __FILE__, __LINE__ + 1
                  def #{method_name}
                    association = #{reflection.name}
                    association.update_attribute(#{reflection.primary_key_name.inspect}, nil) if association
                  end
                eoruby
1679 1680 1681 1682 1683 1684 1685 1686
              when :restrict
                method_name = "has_one_dependent_restrict_for_#{reflection.name}".to_sym
                define_method(method_name) do
                  unless send(reflection.name).nil?
                    raise DeleteRestrictionError.new(reflection)
                  end
                end
                before_destroy method_name
1687
              else
1688
                raise ArgumentError, "The :dependent option expects either :destroy, :delete, :nullify or :restrict (#{reflection.options[:dependent].inspect})"
1689
            end
A
Aaron Patterson 已提交
1690 1691

            before_destroy method_name
1692 1693 1694
          end
        end

1695 1696
        def configure_dependency_for_belongs_to(reflection)
          if reflection.options.include?(:dependent)
1697 1698 1699 1700 1701 1702 1703
            name = reflection.options[:dependent]

            unless [:destroy, :delete].include?(name)
              raise ArgumentError, "The :dependent option expects either :destroy or :delete (#{reflection.options[:dependent].inspect})"
            end

            method_name = :"belongs_to_dependent_#{name}_for_#{reflection.name}"
1704 1705 1706 1707 1708 1709
            class_eval <<-eoruby, __FILE__, __LINE__ + 1
              def #{method_name}
                association = #{reflection.name}
                association.#{name} if association
              end
            eoruby
1710
            after_destroy method_name
1711 1712 1713
          end
        end

1714 1715 1716 1717 1718 1719 1720 1721
        def delete_all_has_many_dependencies(record, reflection_name, association_class, dependent_conditions)
          association_class.delete_all(dependent_conditions)
        end

        def nullify_has_many_dependencies(record, reflection_name, association_class, primary_key_name, dependent_conditions)
          association_class.update_all("#{primary_key_name} = NULL", dependent_conditions)
        end

1722 1723 1724 1725
        mattr_accessor :valid_keys_for_has_many_association
        @@valid_keys_for_has_many_association = [
          :class_name, :table_name, :foreign_key, :primary_key,
          :dependent,
1726
          :select, :conditions, :include, :order, :group, :having, :limit, :offset,
1727 1728 1729 1730 1731
          :as, :through, :source, :source_type,
          :uniq,
          :finder_sql, :counter_sql,
          :before_add, :after_add, :before_remove, :after_remove,
          :extend, :readonly,
1732
          :validate, :inverse_of
1733 1734
        ]

1735
        def create_has_many_reflection(association_id, options, &extension)
1736
          options.assert_valid_keys(valid_keys_for_has_many_association)
1737
          options[:extend] = create_extension_modules(association_id, extension, options[:extend])
1738

1739
          create_reflection(:has_many, association_id, options, self)
1740 1741
        end

1742 1743 1744 1745
        mattr_accessor :valid_keys_for_has_one_association
        @@valid_keys_for_has_one_association = [
          :class_name, :foreign_key, :remote, :select, :conditions, :order,
          :include, :dependent, :counter_cache, :extend, :as, :readonly,
1746
          :validate, :primary_key, :inverse_of
1747 1748
        ]

1749
        def create_has_one_reflection(association_id, options)
1750
          options.assert_valid_keys(valid_keys_for_has_one_association)
1751
          create_reflection(:has_one, association_id, options, self)
1752
        end
1753

1754 1755
        def create_has_one_through_reflection(association_id, options)
          options.assert_valid_keys(
1756
            :class_name, :foreign_key, :remote, :select, :conditions, :order, :include, :dependent, :counter_cache, :extend, :as, :through, :source, :source_type, :validate
1757 1758 1759
          )
          create_reflection(:has_one, association_id, options, self)
        end
1760

1761 1762
        mattr_accessor :valid_keys_for_belongs_to_association
        @@valid_keys_for_belongs_to_association = [
1763
          :class_name, :primary_key, :foreign_key, :foreign_type, :remote, :select, :conditions,
1764
          :include, :dependent, :counter_cache, :extend, :polymorphic, :readonly,
1765
          :validate, :touch, :inverse_of
1766 1767
        ]

1768
        def create_belongs_to_reflection(association_id, options)
1769
          options.assert_valid_keys(valid_keys_for_belongs_to_association)
1770 1771 1772 1773 1774 1775 1776 1777
          reflection = create_reflection(:belongs_to, association_id, options, self)

          if options[:polymorphic]
            reflection.options[:foreign_type] ||= reflection.class_name.underscore + "_type"
          end

          reflection
        end
1778

1779 1780 1781
        mattr_accessor :valid_keys_for_has_and_belongs_to_many_association
        @@valid_keys_for_has_and_belongs_to_many_association = [
          :class_name, :table_name, :join_table, :foreign_key, :association_foreign_key,
1782
          :select, :conditions, :include, :order, :group, :having, :limit, :offset,
1783
          :uniq,
1784
          :finder_sql, :counter_sql, :delete_sql, :insert_sql,
1785 1786 1787 1788 1789
          :before_add, :after_add, :before_remove, :after_remove,
          :extend, :readonly,
          :validate
        ]

1790
        def create_has_and_belongs_to_many_reflection(association_id, options, &extension)
1791
          options.assert_valid_keys(valid_keys_for_has_and_belongs_to_many_association)
1792
          options[:extend] = create_extension_modules(association_id, extension, options[:extend])
1793 1794

          reflection = create_reflection(:has_and_belongs_to_many, association_id, options, self)
1795

1796 1797 1798
          if reflection.association_foreign_key == reflection.primary_key_name
            raise HasAndBelongsToManyAssociationForeignKeyNeeded.new(reflection)
          end
1799

1800
          reflection.options[:join_table] ||= join_table_name(undecorated_table_name(self.to_s), undecorated_table_name(reflection.class_name))
1801 1802 1803
          if connection.supports_primary_key? && (connection.primary_key(reflection.options[:join_table]) rescue false)
             raise HasAndBelongsToManyAssociationWithPrimaryKeyError.new(reflection)
          end
1804

1805 1806 1807
          reflection
        end

1808
        def add_association_callbacks(association_name, options)
1809 1810
          callbacks = %w(before_add after_add before_remove after_remove)
          callbacks.each do |callback_name|
1811
            full_callback_name = "#{callback_name}_for_#{association_name}"
1812 1813 1814
            defined_callbacks = options[callback_name.to_sym]
            if options.has_key?(callback_name.to_sym)
              class_inheritable_reader full_callback_name.to_sym
1815 1816 1817
              write_inheritable_attribute(full_callback_name.to_sym, [defined_callbacks].flatten)
            else
              write_inheritable_attribute(full_callback_name.to_sym, [])
1818 1819
            end
          end
1820
        end
1821

1822
        def create_extension_modules(association_id, block_extension, extensions)
1823
          if block_extension
1824
            extension_module_name = "#{self.to_s.demodulize}#{association_id.to_s.camelize}AssociationExtension"
1825

1826
            silence_warnings do
1827
              self.parent.const_set(extension_module_name, Module.new(&block_extension))
1828
            end
1829
            Array.wrap(extensions).push("#{self.parent}::#{extension_module_name}".constantize)
1830
          else
1831
            Array.wrap(extensions)
1832 1833
          end
        end
1834

D
David Heinemeier Hansson 已提交
1835
        class JoinDependency # :nodoc:
1836
          attr_reader :joins, :reflections, :table_aliases
1837

1838
          def initialize(base, associations, joins)
1839
            @joins                 = [JoinBase.new(base, joins)]
1840 1841 1842 1843
            @associations          = associations
            @reflections           = []
            @base_records_hash     = {}
            @base_records_in_order = []
1844 1845
            @table_aliases         = Hash.new { |aliases, table| aliases[table] = 0 }
            @table_aliases[base.table_name] = 1
1846 1847 1848
            build(associations)
          end

1849 1850 1851
          def graft(*associations)
            associations.each do |association|
              join_associations.detect {|a| association == a} ||
1852
              build(association.reflection.name, association.find_parent_in(self) || join_base, association.join_class)
1853 1854 1855 1856
            end
            self
          end

1857 1858 1859 1860 1861 1862 1863 1864
          def join_associations
            @joins[1..-1].to_a
          end

          def join_base
            @joins[0]
          end

1865
          def count_aliases_from_table_joins(name)
1866 1867
            # quoted_name should be downcased as some database adapters (Oracle) return quoted name in uppercase
            quoted_name = join_base.active_record.connection.quote_table_name(name.downcase).downcase
1868 1869 1870 1871 1872 1873 1874 1875
            join_sql = join_base.table_joins.to_s.downcase
            join_sql.blank? ? 0 :
              # Table names
              join_sql.scan(/join(?:\s+\w+)?\s+#{quoted_name}\son/).size +
              # Table aliases
              join_sql.scan(/join(?:\s+\w+)?\s+\S+\s+#{quoted_name}\son/).size
          end

1876 1877 1878 1879 1880 1881
          def instantiate(rows)
            rows.each_with_index do |row, i|
              primary_id = join_base.record_id(row)
              unless @base_records_hash[primary_id]
                @base_records_in_order << (@base_records_hash[primary_id] = join_base.instantiate(row))
              end
1882
              construct(@base_records_hash[primary_id], @associations, join_associations.dup, row)
1883
            end
1884
            remove_duplicate_results!(join_base.active_record, @base_records_in_order, @associations)
1885 1886 1887
            return @base_records_in_order
          end

1888 1889 1890 1891
          def remove_duplicate_results!(base, records, associations)
            case associations
              when Symbol, String
                reflection = base.reflections[associations]
J
José Valim 已提交
1892
                remove_uniq_by_reflection(reflection, records)
1893 1894 1895 1896 1897 1898 1899
              when Array
                associations.each do |association|
                  remove_duplicate_results!(base, records, association)
                end
              when Hash
                associations.keys.each do |name|
                  reflection = base.reflections[name]
J
José Valim 已提交
1900
                  remove_uniq_by_reflection(reflection, records)
1901

J
Jeremy Kemper 已提交
1902 1903 1904
                  parent_records = []
                  records.each do |record|
                    if descendant = record.send(reflection.name)
1905
                      if reflection.collection?
J
Jeremy Kemper 已提交
1906 1907 1908 1909 1910 1911
                        parent_records.concat descendant.target.uniq
                      else
                        parent_records << descendant
                      end
                    end
                  end
1912

1913
                  remove_duplicate_results!(reflection.klass, parent_records, associations[name]) unless parent_records.empty?
1914 1915 1916 1917
                end
            end
          end

1918
          protected
J
José Valim 已提交
1919

1920
            def build(associations, parent = nil, join_class = Arel::InnerJoin)
1921 1922 1923 1924 1925 1926
              parent ||= @joins.last
              case associations
                when Symbol, String
                  reflection = parent.reflections[associations.to_s.intern] or
                  raise ConfigurationError, "Association named '#{ associations }' was not found; perhaps you misspelled it?"
                  @reflections << reflection
1927
                  @joins << build_join_association(reflection, parent).with_join_class(join_class)
1928 1929
                when Array
                  associations.each do |association|
1930
                    build(association, parent, join_class)
1931 1932 1933
                  end
                when Hash
                  associations.keys.sort{|a,b|a.to_s<=>b.to_s}.each do |name|
1934 1935
                    build(name, parent, join_class)
                    build(associations[name], nil, join_class)
1936 1937 1938 1939 1940 1941
                  end
                else
                  raise ConfigurationError, associations.inspect
              end
            end

J
José Valim 已提交
1942 1943 1944 1945 1946 1947
            def remove_uniq_by_reflection(reflection, records)
              if reflection && reflection.collection?
                records.each { |record| record.send(reflection.name).target.uniq! }
              end
            end

1948 1949 1950 1951
            def build_join_association(reflection, parent)
              JoinAssociation.new(reflection, self, parent)
            end

1952 1953 1954
            def construct(parent, associations, joins, row)
              case associations
                when Symbol, String
1955 1956 1957 1958
                  join = joins.detect{|j| j.reflection.name.to_s == associations.to_s && j.parent_table_name == parent.class.table_name }
                  raise(ConfigurationError, "No such association") if join.nil?

                  joins.delete(join)
1959 1960 1961 1962 1963 1964 1965
                  construct_association(parent, join, row)
                when Array
                  associations.each do |association|
                    construct(parent, association, joins, row)
                  end
                when Hash
                  associations.keys.sort{|a,b|a.to_s<=>b.to_s}.each do |name|
1966 1967 1968 1969 1970
                    join = joins.detect{|j| j.reflection.name.to_s == name.to_s && j.parent_table_name == parent.class.table_name }
                    raise(ConfigurationError, "No such association") if join.nil?

                    association = construct_association(parent, join, row)
                    joins.delete(join)
1971 1972 1973 1974 1975 1976 1977 1978 1979 1980 1981 1982
                    construct(association, associations[name], joins, row) if association
                  end
                else
                  raise ConfigurationError, associations.inspect
              end
            end

            def construct_association(record, join, row)
              case join.reflection.macro
                when :has_many, :has_and_belongs_to_many
                  collection = record.send(join.reflection.name)
                  collection.loaded
1983

1984 1985
                  return nil if record.id.to_s != join.parent.record_id(row).to_s or row[join.aliased_primary_key].nil?
                  association = join.instantiate(row)
1986
                  collection.target.push(association)
1987
                  collection.__send__(:set_inverse_instance, association, record)
1988 1989
                when :has_one
                  return if record.id.to_s != join.parent.record_id(row).to_s
1990
                  return if record.instance_variable_defined?("@#{join.reflection.name}")
1991
                  association = join.instantiate(row) unless row[join.aliased_primary_key].nil?
1992
                  set_target_and_inverse(join, association, record)
1993
                when :belongs_to
1994 1995
                  return if record.id.to_s != join.parent.record_id(row).to_s or row[join.aliased_primary_key].nil?
                  association = join.instantiate(row)
1996
                  set_target_and_inverse(join, association, record)
1997 1998 1999 2000 2001 2002
                else
                  raise ConfigurationError, "unknown macro: #{join.reflection.macro}"
              end
              return association
            end

2003 2004 2005 2006 2007
            def set_target_and_inverse(join, association, record)
              association_proxy = record.send("set_#{join.reflection.name}_target", association)
              association_proxy.__send__(:set_inverse_instance, association, record)
            end

D
David Heinemeier Hansson 已提交
2008
          class JoinBase # :nodoc:
2009
            attr_reader :active_record, :table_joins
2010
            delegate    :table_name, :column_names, :primary_key, :reflections, :sanitize_sql, :arel_engine, :to => :active_record
2011

2012
            def initialize(active_record, joins = nil)
2013 2014
              @active_record = active_record
              @cached_record = {}
2015
              @table_joins   = joins
2016 2017
            end

2018
            def ==(other)
2019
              other.class == self.class &&
2020 2021 2022 2023
              other.active_record == active_record &&
              other.table_joins == table_joins
            end

2024 2025 2026 2027 2028
            def aliased_prefix
              "t0"
            end

            def aliased_primary_key
2029
              "#{aliased_prefix}_r0"
2030 2031 2032 2033 2034 2035 2036
            end

            def aliased_table_name
              active_record.table_name
            end

            def column_names_with_alias
2037
              unless defined?(@column_names_with_alias)
2038
                @column_names_with_alias = []
2039

2040
                ([primary_key] + (column_names - [primary_key])).each_with_index do |column_name, i|
2041
                  @column_names_with_alias << [column_name, "#{aliased_prefix}_r#{i}"]
2042 2043
                end
              end
2044 2045

              @column_names_with_alias
2046 2047 2048 2049 2050 2051 2052 2053 2054 2055 2056
            end

            def extract_record(row)
              column_names_with_alias.inject({}){|record, (cn, an)| record[cn] = row[an]; record}
            end

            def record_id(row)
              row[aliased_primary_key]
            end

            def instantiate(row)
2057
              @cached_record[record_id(row)] ||= active_record.send(:instantiate, extract_record(row))
2058 2059 2060
            end
          end

D
David Heinemeier Hansson 已提交
2061
          class JoinAssociation < JoinBase # :nodoc:
2062
            attr_reader :reflection, :parent, :aliased_table_name, :aliased_prefix, :aliased_join_table_name, :parent_table_name, :join_class
2063
            delegate    :options, :klass, :through_reflection, :source_reflection, :to => :reflection
2064 2065

            def initialize(reflection, join_dependency, parent = nil)
2066 2067 2068 2069 2070
              reflection.check_validity!
              if reflection.options[:polymorphic]
                raise EagerLoadPolymorphicError.new(reflection)
              end

2071
              super(reflection.klass)
2072
              @join_dependency    = join_dependency
2073 2074 2075
              @parent             = parent
              @reflection         = reflection
              @aliased_prefix     = "t#{ join_dependency.joins.size }"
2076
              @parent_table_name  = parent.active_record.table_name
2077
              @aliased_table_name = aliased_table_name_for(table_name)
2078
              @join               = nil
2079
              @join_class         = Arel::InnerJoin
2080

2081 2082
              if reflection.macro == :has_and_belongs_to_many
                @aliased_join_table_name = aliased_table_name_for(reflection.options[:join_table], "_join")
2083
              end
2084

2085
              if [:has_many, :has_one].include?(reflection.macro) && reflection.options[:through]
2086
                @aliased_join_table_name = aliased_table_name_for(reflection.through_reflection.klass.table_name, "_join")
2087
              end
2088 2089
            end

2090
            def ==(other)
2091
              other.class == self.class &&
2092 2093 2094 2095 2096 2097 2098 2099 2100 2101 2102 2103 2104 2105 2106
              other.reflection == reflection &&
              other.parent == parent
            end

            def find_parent_in(other_join_dependency)
              other_join_dependency.joins.detect do |join|
                self.parent == join
              end
            end

            def with_join_class(join_class)
              @join_class = join_class
              self
            end

2107
            def association_join
2108
              return @join if @join
2109

2110 2111 2112 2113 2114 2115 2116
              aliased_table = Arel::Table.new(table_name, :as      => @aliased_table_name,
                                                          :engine  => arel_engine,
                                                          :columns => klass.columns)

              parent_table = Arel::Table.new(parent.table_name, :as      => parent.aliased_table_name,
                                                                :engine  => arel_engine,
                                                                :columns => parent.active_record.columns)
2117

2118
              @join = case reflection.macro
2119
              when :has_and_belongs_to_many
2120
                join_table = Arel::Table.new(options[:join_table], :as => aliased_join_table_name, :engine => arel_engine)
2121 2122 2123 2124 2125 2126 2127 2128 2129
                fk = options[:foreign_key] || reflection.active_record.to_s.foreign_key
                klass_fk = options[:association_foreign_key] || klass.to_s.foreign_key

                [
                  join_table[fk].eq(parent_table[reflection.active_record.primary_key]),
                  aliased_table[klass.primary_key].eq(join_table[klass_fk])
                ]
              when :has_many, :has_one
                if reflection.options[:through]
2130
                  join_table = Arel::Table.new(through_reflection.klass.table_name, :as => aliased_join_table_name, :engine => arel_engine)
2131 2132 2133 2134 2135 2136 2137 2138 2139 2140 2141 2142 2143 2144 2145 2146
                  jt_foreign_key = jt_as_extra = jt_source_extra = jt_sti_extra = nil
                  first_key = second_key = as_extra = nil

                  if through_reflection.options[:as] # has_many :through against a polymorphic join
                    jt_foreign_key = through_reflection.options[:as].to_s + '_id'
                    jt_as_extra = join_table[through_reflection.options[:as].to_s + '_type'].eq(parent.active_record.base_class.name)
                  else
                    jt_foreign_key = through_reflection.primary_key_name
                  end

                  case source_reflection.macro
                  when :has_many
                    if source_reflection.options[:as]
                      first_key   = "#{source_reflection.options[:as]}_id"
                      second_key  = options[:foreign_key] || primary_key
                      as_extra    = aliased_table["#{source_reflection.options[:as]}_type"].eq(source_reflection.active_record.base_class.name)
E
Emilio Tagua 已提交
2147
                    else
2148 2149
                      first_key   = through_reflection.klass.base_class.to_s.foreign_key
                      second_key  = options[:foreign_key] || primary_key
E
Emilio Tagua 已提交
2150 2151
                    end

2152 2153 2154 2155 2156 2157 2158 2159 2160 2161
                    unless through_reflection.klass.descends_from_active_record?
                      jt_sti_extra = join_table[through_reflection.active_record.inheritance_column].eq(through_reflection.klass.sti_name)
                    end
                  when :belongs_to
                    first_key = primary_key
                    if reflection.options[:source_type]
                      second_key = source_reflection.association_foreign_key
                      jt_source_extra = join_table[reflection.source_reflection.options[:foreign_type]].eq(reflection.options[:source_type])
                    else
                      second_key = source_reflection.primary_key_name
E
Emilio Tagua 已提交
2162
                    end
2163
                  end
2164 2165 2166 2167

                  [
                    [parent_table[parent.primary_key].eq(join_table[jt_foreign_key]), jt_as_extra, jt_source_extra, jt_sti_extra].reject{|x| x.blank? },
                    aliased_table[first_key].eq(join_table[second_key])
2168
                  ]
2169 2170 2171 2172 2173 2174 2175 2176 2177
                elsif reflection.options[:as]
                  id_rel = aliased_table["#{reflection.options[:as]}_id"].eq(parent_table[parent.primary_key])
                  type_rel = aliased_table["#{reflection.options[:as]}_type"].eq(parent.active_record.base_class.name)
                  [id_rel, type_rel]
                else
                  foreign_key = options[:foreign_key] || reflection.active_record.name.foreign_key
                  [aliased_table[foreign_key].eq(parent_table[reflection.options[:primary_key] || parent.primary_key])]
                end
              when :belongs_to
2178
                [aliased_table[options[:primary_key] || reflection.klass.primary_key].eq(parent_table[options[:foreign_key] || reflection.primary_key_name])]
2179 2180 2181 2182 2183
              end

              unless klass.descends_from_active_record?
                sti_column = aliased_table[klass.inheritance_column]
                sti_condition = sti_column.eq(klass.sti_name)
2184
                klass.descendants.each {|subclass| sti_condition = sti_condition.or(sti_column.eq(subclass.sti_name)) }
2185 2186

                @join << sti_condition
E
Emilio Tagua 已提交
2187
              end
2188 2189

              [through_reflection, reflection].each do |ref|
2190 2191 2192
                if ref && ref.options[:conditions]
                  @join << interpolate_sql(sanitize_sql(ref.options[:conditions], aliased_table_name))
                end
2193 2194
              end

2195
              @join
2196
            end
2197

2198
            def relation
2199 2200 2201
              aliased = Arel::Table.new(table_name, :as => @aliased_table_name,
                                                    :engine => arel_engine,
                                                    :columns => klass.columns)
2202

2203
              if reflection.macro == :has_and_belongs_to_many
2204
                [Arel::Table.new(options[:join_table], :as => aliased_join_table_name, :engine => arel_engine), aliased]
2205
              elsif reflection.options[:through]
2206
                [Arel::Table.new(through_reflection.klass.table_name, :as => aliased_join_table_name, :engine => arel_engine), aliased]
2207
              else
2208
                aliased
2209 2210 2211
              end
            end

2212
            def join_relation(joining_relation, join = nil)
2213
              joining_relation.joins(self.with_join_class(Arel::OuterJoin))
2214 2215
            end

2216
            protected
2217

2218
              def aliased_table_name_for(name, suffix = nil)
2219 2220
                if @join_dependency.table_aliases[name].zero?
                  @join_dependency.table_aliases[name] = @join_dependency.count_aliases_from_table_joins(name)
2221 2222
                end

2223
                if !@join_dependency.table_aliases[name].zero? # We need an alias
2224 2225
                  name = active_record.connection.table_alias_for "#{pluralize(reflection.name)}_#{parent_table_name}#{suffix}"
                  @join_dependency.table_aliases[name] += 1
2226 2227 2228 2229 2230 2231
                  if @join_dependency.table_aliases[name] == 1 # First time we've seen this name
                    # Also need to count the aliases from the table_aliases to avoid incorrect count
                    @join_dependency.table_aliases[name] += @join_dependency.count_aliases_from_table_joins(name)
                  end
                  table_index = @join_dependency.table_aliases[name]
                  name = name[0..active_record.connection.table_alias_length-3] + "_#{table_index}" if table_index > 1
2232 2233 2234 2235 2236 2237
                else
                  @join_dependency.table_aliases[name] += 1
                end

                name
              end
2238

2239 2240 2241
              def pluralize(table_name)
                ActiveRecord::Base.pluralize_table_names ? table_name.to_s.pluralize : table_name
              end
2242

2243
              def table_alias_for(table_name, table_alias)
2244
                 "#{table_name} #{table_alias if table_name != table_alias}".strip
2245 2246 2247 2248 2249
              end

              def table_name_and_alias
                table_alias_for table_name, @aliased_table_name
              end
2250 2251

              def interpolate_sql(sql)
2252
                instance_eval("%@#{sql.gsub('@', '\@')}@", __FILE__, __LINE__)
2253
              end
2254
          end
2255
        end
2256
    end
D
Initial  
David Heinemeier Hansson 已提交
2257
  end
2258
end