associations.rb 113.9 KB
Newer Older
D
Initial  
David Heinemeier Hansson 已提交
1
module ActiveRecord
D
David Heinemeier Hansson 已提交
2
  class HasManyThroughAssociationNotFoundError < ActiveRecordError #:nodoc:
3 4
    def initialize(owner_class_name, reflection)
      super("Could not find the association #{reflection.options[:through].inspect} in model #{owner_class_name}")
5 6 7
    end
  end

D
David Heinemeier Hansson 已提交
8
  class HasManyThroughAssociationPolymorphicError < ActiveRecordError #:nodoc:
9
    def initialize(owner_class_name, reflection, source_reflection)
10
      super("Cannot have a has_many :through association '#{owner_class_name}##{reflection.name}' on the polymorphic object '#{source_reflection.class_name}##{source_reflection.name}'.")
11 12
    end
  end
13

14 15 16 17 18
  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
19

D
David Heinemeier Hansson 已提交
20
  class HasManyThroughSourceAssociationNotFoundError < ActiveRecordError #:nodoc:
21
    def initialize(reflection)
22 23 24 25
      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 }
      super("Could not find the source association(s) #{source_reflection_names.collect(&:inspect).to_sentence :connector => 'or'} 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 :connector => 'or'}?")
26 27 28
    end
  end

29
  class HasManyThroughSourceAssociationMacroError < ActiveRecordError #:nodoc:
30
    def initialize(reflection)
31 32 33
      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.")
34 35 36
    end
  end

37 38 39 40 41
  class HasManyThroughCantAssociateThroughHasManyReflection < ActiveRecordError #:nodoc:
    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
42 43 44 45 46 47
  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

48 49 50 51 52 53
  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

D
David Heinemeier Hansson 已提交
54
  class EagerLoadPolymorphicError < ActiveRecordError #:nodoc:
55
    def initialize(reflection)
56
      super("Can not eagerly load the polymorphic association #{reflection.name.inspect}")
57 58 59
    end
  end

60 61
  class ReadOnlyAssociation < ActiveRecordError #:nodoc:
    def initialize(reflection)
62
      super("Can not add to a has_many :through association.  Try adding to #{reflection.through_reflection.name.inspect}.")
63 64 65
    end
  end

P
Pratik Naik 已提交
66
  # See ActiveRecord::Associations::ClassMethods for documentation.
D
Initial  
David Heinemeier Hansson 已提交
67
  module Associations # :nodoc:
J
Joshua Peek 已提交
68 69 70 71 72 73 74 75 76 77 78 79
    # These classes will be loaded when associatoins are created.
    # 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'

80
    def self.included(base)
81 82 83
      base.extend(ClassMethods)
    end

84
    # Clears out the association cache
85 86 87
    def clear_association_cache #:nodoc:
      self.class.reflect_on_all_associations.to_a.each do |assoc|
        instance_variable_set "@#{assoc.name}", nil
88
      end unless self.new_record?
89
    end
90 91 92 93

    # 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 options hash. It works much the same way as Ruby's own <tt>attr*</tt>
D
Initial  
David Heinemeier Hansson 已提交
94 95 96 97
    # methods. Example:
    #
    #   class Project < ActiveRecord::Base
    #     belongs_to              :portfolio
98
    #     has_one                 :project_manager
D
Initial  
David Heinemeier Hansson 已提交
99 100 101 102 103
    #     has_many                :milestones
    #     has_and_belongs_to_many :categories
    #   end
    #
    # The project class now has the following methods (and more) to ease the traversal and manipulation of its relationships:
104
    # * <tt>Project#portfolio, Project#portfolio=(portfolio), Project#portfolio.nil?</tt>
D
Initial  
David Heinemeier Hansson 已提交
105 106
    # * <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>
107
    #   <tt>Project#milestones.delete(milestone), Project#milestones.find(milestone_id), Project#milestones.find(:all, options),</tt>
D
Initial  
David Heinemeier Hansson 已提交
108 109 110 111
    #   <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>
    #
112 113 114 115
    # === A word of warning
    #
    # Don't create associations that have the same name as instance methods of ActiveRecord::Base. Since the association
    # adds a method with that name to its model, it will override the inherited method and break things.
P
Pratik Naik 已提交
116
    # For instance, +attributes+ and +connection+ would be bad choices for association names.
117
    #
118
    # == Auto-generated methods
D
Initial  
David Heinemeier Hansson 已提交
119
    #
120
    # === Singular associations (one-to-one)
121 122 123 124 125 126 127 128
    #                                     |            |  belongs_to  |
    #   generated methods                 | belongs_to | :polymorphic | has_one
    #   ----------------------------------+------------+--------------+---------
    #   #other                            |     X      |      X       |    X
    #   #other=(other)                    |     X      |      X       |    X
    #   #build_other(attributes={})       |     X      |              |    X
    #   #create_other(attributes={})      |     X      |              |    X
    #   #other.create!(attributes={})     |            |              |    X
129
    #   #other.nil?                       |     X      |      X       |
130 131 132
    #
    # ===Collection associations (one-to-many / many-to-many)
    #                                     |       |          | has_many
133
    #   generated methods                 | habtm | has_many | :through
134 135
    #   ----------------------------------+-------+----------+----------
    #   #others                           |   X   |    X     |    X
136
    #   #others=(other,other,...)         |   X   |    X     |    X
137
    #   #other_ids                        |   X   |    X     |    X
138
    #   #other_ids=(id,id,...)            |   X   |    X     |    X
139 140 141
    #   #others<<                         |   X   |    X     |    X
    #   #others.push                      |   X   |    X     |    X
    #   #others.concat                    |   X   |    X     |    X
142 143
    #   #others.build(attributes={})      |   X   |    X     |    X
    #   #others.create(attributes={})     |   X   |    X     |    X
144
    #   #others.create!(attributes={})    |   X   |    X     |    X
145 146
    #   #others.size                      |   X   |    X     |    X
    #   #others.length                    |   X   |    X     |    X
P
Pratik Naik 已提交
147
    #   #others.count                     |   X   |    X     |    X
148
    #   #others.sum(args*,&block)         |   X   |    X     |    X
149
    #   #others.empty?                    |   X   |    X     |    X
150
    #   #others.clear                     |   X   |    X     |    X
151
    #   #others.delete(other,other,...)   |   X   |    X     |    X
152
    #   #others.delete_all                |   X   |    X     |
153
    #   #others.destroy_all               |   X   |    X     |    X
154
    #   #others.find(*args)               |   X   |    X     |    X
155
    #   #others.find_first                |   X   |          |
P
Pratik Naik 已提交
156
    #   #others.exist?                    |   X   |    X     |    X
157
    #   #others.uniq                      |   X   |    X     |    X
158
    #   #others.reset                     |   X   |    X     |    X
D
Initial  
David Heinemeier Hansson 已提交
159
    #
160
    # == Cardinality and associations
161
    #
P
Pratik Naik 已提交
162
    # Active Record associations can be used to describe one-to-one, one-to-many and many-to-many
P
Pratik Naik 已提交
163 164
    # 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
165
    # the foreign key.
166 167 168
    #
    # === One-to-one
    #
169
    # Use +has_one+ in the base, and +belongs_to+ in the associated model.
170 171 172 173 174 175 176 177 178 179
    #
    #   class Employee < ActiveRecord::Base
    #     has_one :office
    #   end
    #   class Office < ActiveRecord::Base
    #     belongs_to :employee    # foreign key - employee_id
    #   end
    #
    # === One-to-many
    #
180
    # Use +has_many+ in the base, and +belongs_to+ in the associated model.
181 182 183 184 185
    #
    #   class Manager < ActiveRecord::Base
    #     has_many :employees
    #   end
    #   class Employee < ActiveRecord::Base
186
    #     belongs_to :manager     # foreign key - manager_id
187 188 189 190 191 192
    #   end
    #
    # === Many-to-many
    #
    # There are two ways to build a many-to-many relationship.
    #
193
    # The first way uses a +has_many+ association with the <tt>:through</tt> option and a join model, so
194 195 196 197 198 199 200 201 202 203 204 205 206 207 208
    # 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
    #
209
    # For the second way, use +has_and_belongs_to_many+ in both models. This requires a join table
210 211 212 213 214 215 216 217 218
    # 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
    #
219
    # Choosing which way to build a many-to-many relationship is not always simple.
220
    # If you need to work with the relationship model as its own entity,
221
    # use <tt>has_many :through</tt>. Use +has_and_belongs_to_many+ when working with legacy schemas or when
222 223
    # you never work directly with the relationship itself.
    #
224
    # == Is it a +belongs_to+ or +has_one+ association?
D
Initial  
David Heinemeier Hansson 已提交
225
    #
226 227
    # Both express a 1-1 relationship. The difference is mostly where to place the foreign key, which goes on the table for the class
    # declaring the +belongs_to+ relationship. Example:
D
Initial  
David Heinemeier Hansson 已提交
228
    #
229 230 231
    #   class User < ActiveRecord::Base
    #     # I reference an account.
    #     belongs_to :account
D
Initial  
David Heinemeier Hansson 已提交
232 233
    #   end
    #
234 235 236
    #   class Account < ActiveRecord::Base
    #     # One user references me.
    #     has_one :user
D
Initial  
David Heinemeier Hansson 已提交
237 238 239 240
    #   end
    #
    # The tables for these classes could look something like:
    #
241
    #   CREATE TABLE users (
D
Initial  
David Heinemeier Hansson 已提交
242
    #     id int(11) NOT NULL auto_increment,
243 244
    #     account_id int(11) default NULL,
    #     name varchar default NULL,
D
Initial  
David Heinemeier Hansson 已提交
245 246 247
    #     PRIMARY KEY  (id)
    #   )
    #
248
    #   CREATE TABLE accounts (
D
Initial  
David Heinemeier Hansson 已提交
249 250 251 252 253
    #     id int(11) NOT NULL auto_increment,
    #     name varchar default NULL,
    #     PRIMARY KEY  (id)
    #   )
    #
254 255
    # == Unsaved objects and associations
    #
256
    # You can manipulate objects and associations before they are saved to the database, but there is some special behavior you should be
257 258 259 260
    # aware of, mostly involving the saving of associated objects.
    #
    # === One-to-one associations
    #
261 262 263
    # * 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 keys - except if the parent object is unsaved (<tt>new_record? == true</tt>).
    # * If either of these saves fail (due to one of the objects being invalid) the assignment statement returns +false+ and the assignment
264
    #   is cancelled.
P
Pratik Naik 已提交
265
    # * If you wish to assign an object to a +has_one+ association without saving it, use the <tt>association.build</tt> method (documented below).
266
    # * Assigning an object to a +belongs_to+ association does not save the object, since the foreign key field belongs on the parent. It
267
    #   does not save the parent either.
268 269 270
    #
    # === Collections
    #
271
    # * Adding an object to a collection (+has_many+ or +has_and_belongs_to_many+) automatically saves that object, except if the parent object
272
    #   (the owner of the collection) is not yet stored in the database.
P
Pratik Naik 已提交
273 274
    # * If saving any of the objects being added to a collection (via <tt>push</tt> or similar) fails, then <tt>push</tt> returns +false+.
    # * You can add an object to a collection without automatically saving it by using the <tt>collection.build</tt> method (documented below).
275
    # * All unsaved (<tt>new_record? == true</tt>) members of the collection are automatically saved when the parent is saved.
276
    #
277
    # === Association callbacks
278
    #
279
    # Similar to the normal callbacks that hook into the lifecycle of an Active Record object, you can also define callbacks that get
280
    # triggered when you add an object to or remove an object from an association collection. Example:
281 282 283 284 285 286 287
    #
    #   class Project
    #     has_and_belongs_to_many :developers, :after_add => :evaluate_velocity
    #
    #     def evaluate_velocity(developer)
    #       ...
    #     end
288
    #   end
289 290
    #
    # It's possible to stack callbacks by passing them as an array. Example:
291
    #
292
    #   class Project
293
    #     has_and_belongs_to_many :developers, :after_add => [:evaluate_velocity, Proc.new { |p, d| p.shipping_date = Time.now}]
294 295
    #   end
    #
296
    # Possible callbacks are: +before_add+, +after_add+, +before_remove+ and +after_remove+.
297
    #
298 299
    # 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 thrown the object doesn't get removed.
300
    #
301 302
    # === Association extensions
    #
303
    # The proxy objects that control the access to associations can be extended through anonymous modules. This is especially
304
    # beneficial for adding new finders, creators, and other factory-type methods that are only used as part of this association.
305 306 307
    # Example:
    #
    #   class Account < ActiveRecord::Base
308
    #     has_many :people do
309
    #       def find_or_create_by_name(name)
310
    #         first_name, last_name = name.split(" ", 2)
D
David Heinemeier Hansson 已提交
311
    #         find_or_create_by_first_name_and_last_name(first_name, last_name)
312
    #       end
313
    #     end
314 315 316 317 318 319
    #   end
    #
    #   person = Account.find(:first).people.find_or_create_by_name("David Heinemeier Hansson")
    #   person.first_name # => "David"
    #   person.last_name  # => "Heinemeier Hansson"
    #
320 321 322 323
    # If you need to share the same extensions between many associations, you can use a named extension module. Example:
    #
    #   module FindOrCreateByNameExtension
    #     def find_or_create_by_name(name)
324
    #       first_name, last_name = name.split(" ", 2)
D
David Heinemeier Hansson 已提交
325
    #       find_or_create_by_first_name_and_last_name(first_name, last_name)
326 327 328 329 330 331 332 333 334 335
    #     end
    #   end
    #
    #   class Account < ActiveRecord::Base
    #     has_many :people, :extend => FindOrCreateByNameExtension
    #   end
    #
    #   class Company < ActiveRecord::Base
    #     has_many :people, :extend => FindOrCreateByNameExtension
    #   end
336
    #
337
    # If you need to use multiple named extension modules, you can specify an array of modules with the <tt>:extend</tt> option.
338 339 340 341 342 343 344
    # In the case of name conflicts between methods in the modules, methods in modules later in the array supercede
    # those earlier in the array. Example:
    #
    #   class Account < ActiveRecord::Base
    #     has_many :people, :extend => [FindOrCreateByNameExtension, FindRecentExtension]
    #   end
    #
345 346
    # 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:
347
    #
348 349
    # * +proxy_owner+ - Returns the object the association is part of.
    # * +proxy_reflection+ - Returns the reflection object that describes the association.
350
    # * +proxy_target+ - Returns the associated object for +belongs_to+ and +has_one+, or the collection of associated objects for +has_many+ and +has_and_belongs_to_many+.
351
    #
R
Rick Olson 已提交
352
    # === Association Join Models
353
    #
354 355
    # 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 +has_and_belongs_to_many+ association.  The advantage is that you're able to add validations,
356
    # callbacks, and extra attributes on the join model.  Consider the following schema:
357
    #
358 359 360 361
    #   class Author < ActiveRecord::Base
    #     has_many :authorships
    #     has_many :books, :through => :authorships
    #   end
362
    #
363 364 365 366
    #   class Authorship < ActiveRecord::Base
    #     belongs_to :author
    #     belongs_to :book
    #   end
367
    #
368 369 370
    #   @author = Author.find :first
    #   @author.authorships.collect { |a| a.book } # selects all books that the author's authorships belong to.
    #   @author.books                              # selects all books by using the Authorship join model
371
    #
372
    # You can also go through a +has_many+ association on the join model:
373
    #
374 375 376 377
    #   class Firm < ActiveRecord::Base
    #     has_many   :clients
    #     has_many   :invoices, :through => :clients
    #   end
378
    #
379 380 381 382
    #   class Client < ActiveRecord::Base
    #     belongs_to :firm
    #     has_many   :invoices
    #   end
383
    #
384 385 386 387 388 389 390 391
    #   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
    #   @firm.invoices                                   # selects all invoices by going through the Client join model.
    #
R
Rick Olson 已提交
392
    # === Polymorphic Associations
393 394
    #
    # Polymorphic associations on models are not restricted on what types of models they can be associated with.  Rather, they
395
    # specify an interface that a +has_many+ association must adhere to.
396
    #
R
Rick Olson 已提交
397 398 399
    #   class Asset < ActiveRecord::Base
    #     belongs_to :attachable, :polymorphic => true
    #   end
400
    #
R
Rick Olson 已提交
401
    #   class Post < ActiveRecord::Base
402
    #     has_many :assets, :as => :attachable         # The :as option specifies the polymorphic interface to use.
R
Rick Olson 已提交
403 404 405
    #   end
    #
    #   @asset.attachable = @post
406
    #
R
Rick Olson 已提交
407
    # 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
408
    # an +attachable_id+ integer column and an +attachable_type+ string column.
R
Rick Olson 已提交
409
    #
D
David Heinemeier Hansson 已提交
410
    # Using polymorphic associations in combination with single table inheritance (STI) is a little tricky. In order
411
    # for the associations to work as expected, ensure that you store the base model for the STI models in the
D
David Heinemeier Hansson 已提交
412
    # type column of the polymorphic association. To continue with the asset example above, suppose there are guest posts
413
    # and member posts that use the posts table for STI. In this case, there must be a +type+ column in the posts table.
D
David Heinemeier Hansson 已提交
414 415 416
    #
    #   class Asset < ActiveRecord::Base
    #     belongs_to :attachable, :polymorphic => true
417
    #
D
David Heinemeier Hansson 已提交
418 419 420 421
    #     def attachable_type=(sType)
    #        super(sType.to_s.classify.constantize.base_class.to_s)
    #     end
    #   end
422
    #
D
David Heinemeier Hansson 已提交
423 424 425 426 427
    #   class Post < ActiveRecord::Base
    #     # because we store "Post" in attachable_type now :dependent => :destroy will work
    #     has_many :assets, :as => :attachable, :dependent => :destroy
    #   end
    #
428
    #   class GuestPost < Post
D
David Heinemeier Hansson 已提交
429 430
    #   end
    #
431
    #   class MemberPost < Post
D
David Heinemeier Hansson 已提交
432 433
    #   end
    #
D
Initial  
David Heinemeier Hansson 已提交
434 435 436
    # == Caching
    #
    # All of the methods are built on a simple caching principle that will keep the result of the last query around unless specifically
437
    # instructed not to. The cache is even shared across methods to make it even cheaper to use the macro-added methods without
D
Initial  
David Heinemeier Hansson 已提交
438 439 440 441 442 443 444 445
    # worrying too much about performance at the first go. Example:
    #
    #   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
    #
446 447
    # == Eager loading of associations
    #
P
Pratik Naik 已提交
448
    # Eager loading is a way to find objects of a certain class and a number of named associations. This is
449
    # 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
P
Pratik Naik 已提交
450
    # triggers 101 database queries. Through the use of eager loading, the 101 queries can be reduced to 2. Example:
451 452 453 454 455 456 457 458
    #
    #   class Post < ActiveRecord::Base
    #     belongs_to :author
    #     has_many   :comments
    #   end
    #
    # Consider the following loop using the class above:
    #
P
Pratik Naik 已提交
459
    #   for post in Post.all
460 461 462
    #     puts "Post:            " + post.title
    #     puts "Written by:      " + post.author.name
    #     puts "Last comment on: " + post.comments.first.created_on
463
    #   end
464 465 466
    #
    # To iterate over these one hundred posts, we'll generate 201 database queries. Let's first just optimize it for retrieving the author:
    #
467
    #   for post in Post.find(:all, :include => :author)
468
    #
P
Pratik Naik 已提交
469 470
    # 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 from 201 to 102.
471 472 473
    #
    # We can improve upon the situation further by referencing both associations in the finder with:
    #
474
    #   for post in Post.find(:all, :include => [ :author, :comments ])
475
    #
P
Pratik Naik 已提交
476 477
    # 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 named (except if some of the associations are polymorphic +belongs_to+ - see below).
478
    #
479
    # To include a deep hierarchy of associations, use a hash:
480 481 482 483 484 485 486
    #
    #   for post in Post.find(:all, :include => [ :author, { :comments => { :author => :gravatar } } ])
    #
    # 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 associations you want to load.
    #
    # 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
D
David Heinemeier Hansson 已提交
487 488
    # 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 cut down on the number of queries in a situation as the one described above.
489
    #
P
Pratik Naik 已提交
490 491 492 493
    # 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 used LEFT OUTER JOIN based strategy. For example
    #  
    #   Post.find(:all, :include => [ :author, :comments ], :conditions => ['comments.approved = ?', true])
494
    #
P
Pratik Naik 已提交
495 496 497 498 499 500 501 502 503 504 505 506 507 508 509 510
    # will result in a single SQL query with joins along the lines of: <tt>LEFT OUTER JOIN comments ON comments.post_id = posts.id</tt> and
    # <tt>LEFT OUTER JOIN authors ON authors.id = posts.author_id</tt>. Note that using conditions like this can have unintended consequences.
    # 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. You must disambiguate column references for this fallback to happen, for example
    # <tt>:order => "author.name DESC"</tt> will work but <tt>:order => "name DESC"</tt> will not. 
    #
    # If you do want eagerload only some members of an association it is usually more natural to <tt>:include</tt> an association
    # which has conditions defined on it:
    #
    #   class Post < ActiveRecord::Base
    #     has_many :approved_comments, :class_name => 'Comment', :conditions => ['approved = ?', true]
    #   end
    #
    #   Post.find(:all, :include => :approved_comments)
    #
    # will load posts and eager load the +approved_comments+ association, which contains only those comments that have been approved.
511
    #
512 513 514 515 516 517 518 519
    # If you eager load an association with a specified <tt>:limit</tt> option, it will be ignored, returning all the associated objects:
    #
    #   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.
    #
520 521
    # When eager loaded, conditions are interpolated in the context of the model class, not the model instance.  Conditions are lazily interpolated
    # before the actual model exists.
522
    #
P
Pratik Naik 已提交
523
    # Eager loading is supported with polymorphic associations.
524 525 526 527 528
    #
    #   class Address < ActiveRecord::Base
    #     belongs_to :addressable, :polymorphic => true
    #   end
    #
P
Pratik Naik 已提交
529
    # A call that tries to eager load the addressable model
530
    #
P
Pratik Naik 已提交
531
    #   Address.find(:all, :include => :addressable)
532
    #
P
Pratik Naik 已提交
533 534 535 536 537
    # will execute one query to load the addresses and load the addressables with one query per addressable type. 
    # 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 the back of the addresses loaded. This is not supported if Active Record has to fallback
    # 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 name cannot be put in the +FROM+/+JOIN+ clauses of that query.
538
    #
539 540
    # == Table Aliasing
    #
P
Pratik Naik 已提交
541
    # 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,
542
    # the standard table name is used.  The second time, the table is aliased as <tt>#{reflection_name}_#{parent_table_name}</tt>.  Indexes are appended
543
    # for any more successive uses of the table name.
544
    #
P
Pratik Naik 已提交
545 546 547 548 549 550
    #   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
551
    #
552
    # Acts as tree example:
553
    #
P
Pratik Naik 已提交
554 555 556 557 558 559 560 561 562
    #   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
563
    #
564
    # Has and Belongs to Many join tables use the same idea, but add a <tt>_join</tt> suffix:
565
    #
P
Pratik Naik 已提交
566 567 568 569 570 571 572 573 574
    #   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
575
    #
576
    # If you wish to specify your own custom joins using a <tt>:joins</tt> option, those table names will take precedence over the eager associations:
577
    #
P
Pratik Naik 已提交
578 579 580 581 582
    #   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 ...
583
    #                              INNER JOIN comments ...
584
    #
585
    # Table aliases are automatically truncated according to the maximum length of table identifiers according to the specific database.
586
    #
D
Initial  
David Heinemeier Hansson 已提交
587 588 589 590 591 592 593 594 595 596
    # == 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 已提交
597
    #       class Client < ActiveRecord::Base; end
D
Initial  
David Heinemeier Hansson 已提交
598 599 600
    #     end
    #   end
    #
P
Pratik Naik 已提交
601 602 603
    # When <tt>Firm#clients</tt> is called, it will in turn call <tt>MyApplication::Business::Client.find_all_by_firm_id(firm.id)</tt>.
    # If you want to associate with a class in another module scope, this can be done by specifying the complete class name.
    # Example:
D
Initial  
David Heinemeier Hansson 已提交
604 605 606 607 608 609 610 611 612 613 614 615 616
    #
    #   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
    #
617
    # == Type safety with <tt>ActiveRecord::AssociationTypeMismatch</tt>
D
Initial  
David Heinemeier Hansson 已提交
618 619
    #
    # If you attempt to assign an object to an association that doesn't match the inferred or specified <tt>:class_name</tt>, you'll
620
    # get an <tt>ActiveRecord::AssociationTypeMismatch</tt>.
D
Initial  
David Heinemeier Hansson 已提交
621 622 623
    #
    # == Options
    #
624
    # All of the association macros can be specialized through options. This makes cases more complex than the simple and guessable ones
D
Initial  
David Heinemeier Hansson 已提交
625 626
    # possible.
    module ClassMethods
P
Pratik Naik 已提交
627 628 629 630 631
      # 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 已提交
632
      #   An empty array is returned if none are found.
P
Pratik Naik 已提交
633 634 635 636
      # [collection<<(object, ...)]
      #   Adds one or more objects to the collection by setting their foreign keys to the collection's primary key.
      # [collection.delete(object, ...)]
      #   Removes one or more objects from the collection by setting their foreign keys to +NULL+.
637 638
      #   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 已提交
639 640 641 642 643 644 645 646 647 648 649 650 651 652 653 654 655 656 657 658 659 660 661 662 663 664 665 666 667 668 669 670 671 672
      # [collection=objects]
      #   Replaces the collections 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 with the objects identified by the primary keys in +ids+
      # [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+.
      # [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.
      # [collection.exist?(...)]
      #   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
      #   been saved. <b>Note:</b> This only works if an associated object already exists, not if
      #   it's +nil+!
      # [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
      #   been saved (if it passed the validation). <b>Note:</b> This only works if an associated
      #   object already exists, not if it's +nil+!
      #
      # (*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 已提交
673
      #
P
Pratik Naik 已提交
674 675
      # 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 已提交
676 677
      # * <tt>Firm#clients<<</tt>
      # * <tt>Firm#clients.delete</tt>
678
      # * <tt>Firm#clients=</tt>
679
      # * <tt>Firm#client_ids</tt>
680
      # * <tt>Firm#client_ids=</tt>
D
Initial  
David Heinemeier Hansson 已提交
681 682 683
      # * <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>)
684
      # * <tt>Firm#clients.find</tt> (similar to <tt>Client.find(id, :conditions => "firm_id = #{id}")</tt>)
P
Pratik Naik 已提交
685
      # * <tt>Firm#clients.exist?(:name => 'ACME')</tt> (similar to <tt>Client.exist?(:name => 'ACME', :firm_id => firm.id)</tt>)
D
Initial  
David Heinemeier Hansson 已提交
686
      # * <tt>Firm#clients.build</tt> (similar to <tt>Client.new("firm_id" => id)</tt>)
D
David Heinemeier Hansson 已提交
687
      # * <tt>Firm#clients.create</tt> (similar to <tt>c = Client.new("firm_id" => id); c.save; c</tt>)
D
Initial  
David Heinemeier Hansson 已提交
688
      # The declaration can also include an options hash to specialize the behavior of the association.
689
      #
P
Pratik Naik 已提交
690 691 692
      # === Supported options
      # [:class_name]
      #   Specify the class name of the association. Use it only if that name can't be inferred
P
Pratik Naik 已提交
693 694
      #   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 specify it with this option.
P
Pratik Naik 已提交
695 696
      # [:conditions]
      #   Specify the conditions that the associated objects must meet in order to be included as a +WHERE+
697 698 699
      #   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 posts with <tt>@blog.posts.create</tt>
      #   or <tt>@blog.posts.build</tt>.
P
Pratik Naik 已提交
700 701
      # [:order]
      #   Specify the order in which the associated objects are returned as an <tt>ORDER BY</tt> SQL fragment,
P
Pratik Naik 已提交
702
      #   such as <tt>last_name, first_name DESC</tt>.
P
Pratik Naik 已提交
703 704
      # [:foreign_key]
      #   Specify the foreign key used for the association. By default this is guessed to be the name
P
Pratik Naik 已提交
705 706
      #   of this class in lower-case and "_id" suffixed. So a Person class that makes a +has_many+ association will use "person_id"
      #   as the default <tt>:foreign_key</tt>.
P
Pratik Naik 已提交
707 708 709 710
      # [: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 已提交
711 712 713 714
      #   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
      #   objects' foreign keys are set to +NULL+ *without* calling their +save+ callbacks. *Warning:* This option is ignored when also using
      #   the <tt>:through</tt> option.
P
Pratik Naik 已提交
715 716
      # [:finder_sql]
      #   Specify a complete SQL statement to fetch the association. This is a good way to go for complex
D
David Heinemeier Hansson 已提交
717
      #   associations that depend on multiple tables. Note: When this option is used, +find_in_collection+ is _not_ added.
P
Pratik Naik 已提交
718 719
      # [:counter_sql]
      #   Specify a complete SQL statement to fetch the size of the association. If <tt>:finder_sql</tt> is
720
      #   specified but not <tt>:counter_sql</tt>, <tt>:counter_sql</tt> will be generated by replacing <tt>SELECT ... FROM</tt> with <tt>SELECT COUNT(*) FROM</tt>.
P
Pratik Naik 已提交
721 722 723 724 725 726
      # [: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.
727 728
      # [:having]
      #   Combined with +:group+ this can be used to filter the records that a <tt>GROUP BY</tt> returns. Uses the <tt>HAVING</tt> SQL-clause.
P
Pratik Naik 已提交
729 730 731 732 733 734
      # [:limit]
      #   An integer determining the limit on the number of rows that should be returned.
      # [:offset]
      #   An integer determining the offset from where the rows should be fetched. So at 5, it would skip the first 4 rows.
      # [:select]
      #   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
735
      #   but not include the joined columns. Do not forget to include the primary and foreign keys, otherwise it will raise an error.
P
Pratik Naik 已提交
736 737 738 739
      # [:as]
      #   Specifies a polymorphic interface (See <tt>belongs_to</tt>).
      # [:through]
      #   Specifies a Join Model through which to perform the query.  Options for <tt>:class_name</tt> and <tt>:foreign_key</tt>
740 741
      #   are ignored, as the association uses the source reflection. You can only use a <tt>:through</tt> query through a <tt>belongs_to</tt>
      #   or <tt>has_many</tt> association on the join model.
P
Pratik Naik 已提交
742 743
      # [:source]
      #   Specifies the source association name used by <tt>has_many :through</tt> queries.  Only use it if the name cannot be
744
      #   inferred from the association.  <tt>has_many :subscribers, :through => :subscriptions</tt> will look for either <tt>:subscribers</tt> or
P
Pratik Naik 已提交
745
      #   <tt>:subscriber</tt> on Subscription, unless a <tt>:source</tt> is given.
P
Pratik Naik 已提交
746 747
      # [:source_type]
      #   Specifies type of the source association used by <tt>has_many :through</tt> queries where the source
748
      #   association is a polymorphic +belongs_to+.
P
Pratik Naik 已提交
749 750 751 752 753 754
      # [: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]
      #   If false, don't validate the associated objects when saving the parent object. true by default.
D
Initial  
David Heinemeier Hansson 已提交
755 756
      # Option examples:
      #   has_many :comments, :order => "posted_on"
757
      #   has_many :comments, :include => :author
D
Initial  
David Heinemeier Hansson 已提交
758
      #   has_many :people, :class_name => "Person", :conditions => "deleted = 0", :order => "name"
759 760
      #   has_many :tracks, :order => "position", :dependent => :destroy
      #   has_many :comments, :dependent => :nullify
R
Rick Olson 已提交
761
      #   has_many :tags, :as => :taggable
762
      #   has_many :reports, :readonly => true
763
      #   has_many :subscribers, :through => :subscriptions, :source => :user
D
Initial  
David Heinemeier Hansson 已提交
764 765 766 767 768
      #   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'
769
      def has_many(association_id, options = {}, &extension)
770
        reflection = create_has_many_reflection(association_id, options, &extension)
771

772
        configure_dependency_for_has_many(reflection)
773

774 775
        add_multiple_associated_validation_callbacks(reflection.name) unless options[:validate] == false
        add_multiple_associated_save_callbacks(reflection.name)
776 777
        add_association_callbacks(reflection.name, reflection.options)

778
        if options[:through]
779
          collection_accessor_methods(reflection, HasManyThroughAssociation)
780 781
        else
          collection_accessor_methods(reflection, HasManyAssociation)
D
Initial  
David Heinemeier Hansson 已提交
782 783 784
        end
      end

P
Pratik Naik 已提交
785 786 787 788 789 790 791 792 793 794 795
      # 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 已提交
796
      #   and saves the associate object.
P
Pratik Naik 已提交
797 798 799 800 801 802 803 804 805 806 807 808 809 810
      # [association.nil?]
      #   Returns +true+ if there is no associated object.
      # [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 已提交
811
      #
P
Pratik Naik 已提交
812 813 814
      # === Example
      #
      # An Account class declares <tt>has_one :beneficiary</tt>, which will add:
815
      # * <tt>Account#beneficiary</tt> (similar to <tt>Beneficiary.find(:first, :conditions => "account_id = #{id}")</tt>)
D
Initial  
David Heinemeier Hansson 已提交
816 817
      # * <tt>Account#beneficiary=(beneficiary)</tt> (similar to <tt>beneficiary.account_id = account.id; beneficiary.save</tt>)
      # * <tt>Account#beneficiary.nil?</tt>
818 819 820
      # * <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 已提交
821 822
      # === Options
      #
D
Initial  
David Heinemeier Hansson 已提交
823
      # The declaration can also include an options hash to specialize the behavior of the association.
824
      #
D
Initial  
David Heinemeier Hansson 已提交
825
      # Options are:
P
Pratik Naik 已提交
826 827
      # [:class_name]
      #   Specify the class name of the association. Use it only if that name can't be inferred
P
Pratik Naik 已提交
828 829
      #   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 已提交
830 831
      # [:conditions]
      #   Specify the conditions that the associated object must meet in order to be included as a +WHERE+
832
      #   SQL fragment, such as <tt>rank = 5</tt>.
P
Pratik Naik 已提交
833 834
      # [:order]
      #   Specify the order in which the associated objects are returned as an <tt>ORDER BY</tt> SQL fragment,
P
Pratik Naik 已提交
835
      #   such as <tt>last_name, first_name DESC</tt>.
P
Pratik Naik 已提交
836 837
      # [:dependent]
      #   If set to <tt>:destroy</tt>, the associated object is destroyed when this object is. If set to
838 839
      #   <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+. Also, association is assigned.
P
Pratik Naik 已提交
840 841
      # [:foreign_key]
      #   Specify the foreign key used for the association. By default this is guessed to be the name
P
Pratik Naik 已提交
842 843
      #   of this class in lower-case and "_id" suffixed. So a Person class that makes a +has_one+ association will use "person_id"
      #   as the default <tt>:foreign_key</tt>.
P
Pratik Naik 已提交
844 845 846 847 848 849 850 851
      # [: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]
      #   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
852
      #   but not include the joined columns. Do not forget to include the primary and foreign keys, otherwise it will raise an error.
P
Pratik Naik 已提交
853 854
      # [:through]
      #   Specifies a Join Model through which to perform the query.  Options for <tt>:class_name</tt> and <tt>:foreign_key</tt>
855 856
      #   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> association on the join model.
P
Pratik Naik 已提交
857 858
      # [:source]
      #   Specifies the source association name used by <tt>has_one :through</tt> queries.  Only use it if the name cannot be
859
      #   inferred from the association.  <tt>has_one :favorite, :through => :favorites</tt> will look for a
P
Pratik Naik 已提交
860
      #   <tt>:favorite</tt> on Favorite, unless a <tt>:source</tt> is given.      
P
Pratik Naik 已提交
861 862
      # [:source_type]
      #   Specifies type of the source association used by <tt>has_one :through</tt> queries where the source
863
      #   association is a polymorphic +belongs_to+.      
P
Pratik Naik 已提交
864 865 866 867
      # [:readonly]
      #   If true, the associated object is readonly through the association.
      # [:validate]
      #   If false, don't validate the associated object when saving the parent object. +false+ by default.
868
      #
D
Initial  
David Heinemeier Hansson 已提交
869
      # Option examples:
870
      #   has_one :credit_card, :dependent => :destroy  # destroys the associated credit card
871
      #   has_one :credit_card, :dependent => :nullify  # updates the associated records foreign key value to NULL rather than destroying it
D
Initial  
David Heinemeier Hansson 已提交
872 873
      #   has_one :last_comment, :class_name => "Comment", :order => "posted_on"
      #   has_one :project_manager, :class_name => "Person", :conditions => "role = 'project_manager'"
874
      #   has_one :attachment, :as => :attachable
875
      #   has_one :boss, :readonly => :true
876 877
      #   has_one :club, :through => :membership
      #   has_one :primary_address, :through => :addressables, :conditions => ["addressable.primary = ?", true], :source => :addressable
D
Initial  
David Heinemeier Hansson 已提交
878
      def has_one(association_id, options = {})
879 880 881 882 883
        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)
D
Initial  
David Heinemeier Hansson 已提交
884

885
          ivar = "@#{reflection.name}"
886

887 888
          method_name = "has_one_after_save_for_#{reflection.name}".to_sym
          define_method(method_name) do
889
            association = instance_variable_get(ivar) if instance_variable_defined?(ivar)
890

891 892
            if !association.nil? && (new_record? || association.new_record? || association[reflection.primary_key_name] != id)
              association[reflection.primary_key_name] = id
893 894
              association.save(true)
            end
895
          end
896
          after_save method_name
897

898
          add_single_associated_validation_callbacks(reflection.name) if options[:validate] == true
899 900 901
          association_accessor_methods(reflection, HasOneAssociation)
          association_constructor_method(:build,  reflection, HasOneAssociation)
          association_constructor_method(:create, reflection, HasOneAssociation)
902

903 904
          configure_dependency_for_has_one(reflection)
        end
D
Initial  
David Heinemeier Hansson 已提交
905 906
      end

P
Pratik Naik 已提交
907 908 909 910 911 912 913 914 915 916 917 918 919 920 921 922
      # 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.
      # [association.nil?]
      #   Returns +true+ if there is no associated object.
      # [build_association(attributes = {})]
      #   Returns a new object of the associated type that has been instantiated
923
      #   with +attributes+ and linked to this object through a foreign key, but has not yet been saved.
P
Pratik Naik 已提交
924 925 926 927 928 929 930
      # [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 已提交
931
      #
P
Pratik Naik 已提交
932 933 934
      # === Example
      #
      # A Post class declares <tt>belongs_to :author</tt>, which will add:
D
Initial  
David Heinemeier Hansson 已提交
935 936 937 938
      # * <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>)
      # * <tt>Post#author?</tt> (similar to <tt>post.author == some_author</tt>)
      # * <tt>Post#author.nil?</tt>
D
David Heinemeier Hansson 已提交
939 940
      # * <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 已提交
941
      # The declaration can also include an options hash to specialize the behavior of the association.
942
      #
P
Pratik Naik 已提交
943 944 945 946
      # === Options
      #
      # [:class_name]
      #   Specify the class name of the association. Use it only if that name can't be inferred
P
Pratik Naik 已提交
947 948
      #   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 已提交
949 950
      # [:conditions]
      #   Specify the conditions that the associated object must meet in order to be included as a +WHERE+
951
      #   SQL fragment, such as <tt>authorized = 1</tt>.
P
Pratik Naik 已提交
952 953
      # [:select]
      #   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
954
      #   but not include the joined columns. Do not forget to include the primary and foreign keys, otherwise it will raise an error.
P
Pratik Naik 已提交
955 956
      # [:foreign_key]
      #   Specify the foreign key used for the association. By default this is guessed to be the name
P
Pratik Naik 已提交
957 958 959
      #   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 of "favorite_person_id".
P
Pratik Naik 已提交
960 961
      # [:dependent]
      #   If set to <tt>:destroy</tt>, the associated object is destroyed when this object is. If set to
962 963 964
      #   <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 a <tt>has_many</tt> relationship on another class because of the potential to leave
      #   orphaned records behind.
P
Pratik Naik 已提交
965 966
      # [:counter_cache]
      #   Caches the number of belonging objects on the associate class through the use of +increment_counter+
967
      #   and +decrement_counter+. The counter cache is incremented when an object of this class is created and decremented when it's
P
Pratik Naik 已提交
968 969
      #   destroyed. This requires that a column named <tt>#{table_name}_count</tt> (such as +comments_count+ for a belonging Comment class)
      #   is used on the associate class (such as a Post class). You can also specify a custom counter cache column by providing
970
      #   a column name instead of a +true+/+false+ value to this option (e.g., <tt>:counter_cache => :my_custom_counter</tt>.)
P
Pratik Naik 已提交
971
      #   Note: Specifying a counter cache will add it to that model's list of readonly attributes using +attr_readonly+.
P
Pratik Naik 已提交
972 973 974 975
      # [: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+.
976
      #   Note: If you've enabled the counter cache, then you may want to add the counter cache attribute
P
Pratik Naik 已提交
977
      #   to the +attr_readonly+ list in the associated classes (e.g. <tt>class Post; attr_readonly :comments_count; end</tt>).
P
Pratik Naik 已提交
978 979 980 981
      # [:readonly]
      #   If true, the associated object is readonly through the association.
      # [:validate]
      #   If false, don't validate the associated objects when saving the parent object. +false+ by default.
D
Initial  
David Heinemeier Hansson 已提交
982 983 984 985
      #
      # Option examples:
      #   belongs_to :firm, :foreign_key => "client_of"
      #   belongs_to :author, :class_name => "Person", :foreign_key => "author_id"
986
      #   belongs_to :valid_coupon, :class_name => "Coupon", :foreign_key => "coupon_id",
D
Initial  
David Heinemeier Hansson 已提交
987
      #              :conditions => 'discounts > #{payments_count}'
R
Rick Olson 已提交
988
      #   belongs_to :attachable, :polymorphic => true
989
      #   belongs_to :project, :readonly => true
990
      #   belongs_to :post, :counter_cache => true
991
      def belongs_to(association_id, options = {})
992
        reflection = create_belongs_to_reflection(association_id, options)
993 994 995

        ivar = "@#{reflection.name}"

996 997
        if reflection.options[:polymorphic]
          association_accessor_methods(reflection, BelongsToPolymorphicAssociation)
998

999 1000
          method_name = "polymorphic_belongs_to_before_save_for_#{reflection.name}".to_sym
          define_method(method_name) do
1001
            association = instance_variable_get(ivar) if instance_variable_defined?(ivar)
1002

1003 1004 1005 1006
            if association && association.target
              if association.new_record?
                association.save(true)
              end
1007

1008
              if association.updated?
1009 1010
                self[reflection.primary_key_name] = association.id
                self[reflection.options[:foreign_type]] = association.class.base_class.name.to_s
1011
              end
1012
            end
1013
          end
1014
          before_save method_name
1015
        else
1016 1017 1018
          association_accessor_methods(reflection, BelongsToAssociation)
          association_constructor_method(:build,  reflection, BelongsToAssociation)
          association_constructor_method(:create, reflection, BelongsToAssociation)
1019

1020 1021
          method_name = "belongs_to_before_save_for_#{reflection.name}".to_sym
          define_method(method_name) do
1022
            association = instance_variable_get(ivar) if instance_variable_defined?(ivar)
1023

1024 1025 1026 1027
            if !association.nil?
              if association.new_record?
                association.save(true)
              end
1028

1029
              if association.updated?
1030
                self[reflection.primary_key_name] = association.id
1031
              end
1032
            end
1033
          end
1034
          before_save method_name
1035
        end
1036

1037
        # Create the callbacks to update counter cache
1038
        if options[:counter_cache]
1039
          cache_column = options[:counter_cache] == true ?
1040
            "#{self.to_s.demodulize.underscore.pluralize}_count" :
1041 1042
            options[:counter_cache]

1043 1044
          method_name = "belongs_to_counter_cache_after_create_for_#{reflection.name}".to_sym
          define_method(method_name) do
1045 1046
            association = send(reflection.name)
            association.class.increment_counter(cache_column, send(reflection.primary_key_name)) unless association.nil?
1047 1048
          end
          after_create method_name
1049

1050 1051
          method_name = "belongs_to_counter_cache_before_destroy_for_#{reflection.name}".to_sym
          define_method(method_name) do
1052 1053
            association = send(reflection.name)
            association.class.decrement_counter(cache_column, send(reflection.primary_key_name)) unless association.nil?
1054 1055
          end
          before_destroy method_name
1056

1057
          module_eval(
1058
            "#{reflection.class_name}.send(:attr_readonly,\"#{cache_column}\".intern) if defined?(#{reflection.class_name}) && #{reflection.class_name}.respond_to?(:attr_readonly)"
1059
          )
1060
        end
1061

1062
        add_single_associated_validation_callbacks(reflection.name) if options[:validate] == true
1063

1064
        configure_dependency_for_belongs_to(reflection)
1065 1066
      end

P
Pratik Naik 已提交
1067 1068 1069
      # 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
P
Pratik Naik 已提交
1070 1071
      # 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,
1072
      # and the strings are equal when compared up to the shortest length, then the longer string is considered of higher
P
Pratik Naik 已提交
1073 1074 1075 1076
      # 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 已提交
1077
      #
1078
      # Deprecated: Any additional fields added to the join table will be placed as attributes when pulling records out through
1079
      # +has_and_belongs_to_many+ associations. Records returned from join tables with additional attributes will be marked as
P
Pratik Naik 已提交
1080
      # readonly (because we can't save changes to the additional attributes). It's strongly recommended that you upgrade any
1081
      # associations with attributes to a real join model (see introduction).
D
Initial  
David Heinemeier Hansson 已提交
1082
      #
1083
      # Adds the following methods for retrieval and query:
P
Pratik Naik 已提交
1084 1085 1086
      #
      # [collection(force_reload = false)]
      #   Returns an array of all the associated objects.
1087
      #   An empty array is returned if none are found.
P
Pratik Naik 已提交
1088 1089
      # [collection<<(object, ...)]
      #   Adds one or more objects to the collection by creating associations in the join table
1090
      #   (<tt>collection.push</tt> and <tt>collection.concat</tt> are aliases to this method).
P
Pratik Naik 已提交
1091 1092
      # [collection.delete(object, ...)]
      #   Removes one or more objects from the collection by removing their associations from the join table.
D
Initial  
David Heinemeier Hansson 已提交
1093
      #   This does not destroy the objects.
P
Pratik Naik 已提交
1094 1095 1096 1097 1098 1099 1100 1101 1102 1103 1104 1105 1106 1107
      # [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
1108
      #   meets the condition that it has to be associated with this object.
P
Pratik Naik 已提交
1109 1110 1111 1112 1113 1114
      #   Uses the same rules as ActiveRecord::Base.find.
      # [collection.exist?(...)]
      #   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
1115
      #   with +attributes+ and linked to this object through the join table, but has not yet been saved.
P
Pratik Naik 已提交
1116 1117
      # [collection.create(attributes = {})]
      #   Returns a new object of the collection type that has been instantiated
1118
      #   with +attributes+, linked to this object through the join table, and that has already been saved (if it passed the validation).
D
Initial  
David Heinemeier Hansson 已提交
1119
      #
P
Pratik Naik 已提交
1120 1121 1122 1123 1124 1125
      # (+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 已提交
1126 1127 1128
      # * <tt>Developer#projects</tt>
      # * <tt>Developer#projects<<</tt>
      # * <tt>Developer#projects.delete</tt>
1129
      # * <tt>Developer#projects=</tt>
1130
      # * <tt>Developer#project_ids</tt>
1131
      # * <tt>Developer#project_ids=</tt>
D
Initial  
David Heinemeier Hansson 已提交
1132 1133 1134 1135
      # * <tt>Developer#projects.clear</tt>
      # * <tt>Developer#projects.empty?</tt>
      # * <tt>Developer#projects.size</tt>
      # * <tt>Developer#projects.find(id)</tt>
P
Pratik Naik 已提交
1136
      # * <tt>Developer#clients.exist?(...)</tt>
1137 1138
      # * <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 已提交
1139
      # The declaration may include an options hash to specialize the behavior of the association.
1140
      #
P
Pratik Naik 已提交
1141 1142 1143 1144
      # === Options
      #
      # [:class_name]
      #   Specify the class name of the association. Use it only if that name can't be inferred
1145
      #   from the association name. So <tt>has_and_belongs_to_many :projects</tt> will by default be linked to the
P
Pratik Naik 已提交
1146
      #   Project class, but if the real class name is SuperProject, you'll have to specify it with this option.
P
Pratik Naik 已提交
1147 1148 1149 1150 1151 1152
      # [: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
P
Pratik Naik 已提交
1153 1154
      #   of this class in lower-case and "_id" suffixed. So a Person class that makes a +has_and_belongs_to_many+ association
      #   will use "person_id" as the default <tt>:foreign_key</tt>.
P
Pratik Naik 已提交
1155 1156
      # [:association_foreign_key]
      #   Specify the association foreign key used for the association. By default this is
P
Pratik Naik 已提交
1157 1158
      #   guessed to be the name of the associated class in lower-case and "_id" suffixed. So if the associated class is Project,
      #   the +has_and_belongs_to_many+ association will use "project_id" as the default <tt>:association_foreign_key</tt>.
P
Pratik Naik 已提交
1159 1160
      # [:conditions]
      #   Specify the conditions that the associated object must meet in order to be included as a +WHERE+
1161 1162 1163
      #   SQL fragment, such as <tt>authorized = 1</tt>.  Record creations from the association are scoped if a hash is used.  
      #   <tt>has_many :posts, :conditions => {:published => true}</tt> will create published posts with <tt>@blog.posts.create</tt> 
      #   or <tt>@blog.posts.build</tt>.
P
Pratik Naik 已提交
1164 1165
      # [:order]
      #   Specify the order in which the associated objects are returned as an <tt>ORDER BY</tt> SQL fragment,
1166
      #   such as <tt>last_name, first_name DESC</tt>
P
Pratik Naik 已提交
1167 1168 1169 1170
      # [: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
1171 1172 1173
      # [:counter_sql]
      #   Specify a complete SQL statement to fetch the size of the association. If <tt>:finder_sql</tt> is
      #   specified but not <tt>:counter_sql</tt>, <tt>:counter_sql</tt> will be generated by replacing <tt>SELECT ... FROM</tt> with <tt>SELECT COUNT(*) FROM</tt>.
P
Pratik Naik 已提交
1174 1175
      # [:delete_sql]
      #   Overwrite the default generated SQL statement used to remove links between the associated
P
Pratik Naik 已提交
1176
      #   classes with a manual statement.
P
Pratik Naik 已提交
1177 1178
      # [:insert_sql]
      #   Overwrite the default generated SQL statement used to add links between the associated classes
P
Pratik Naik 已提交
1179
      #   with a manual statement.
P
Pratik Naik 已提交
1180 1181 1182 1183 1184 1185
      # [: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.
1186 1187
      # [:having]
      #   Combined with +:group+ this can be used to filter the records that a <tt>GROUP BY</tt> returns. Uses the <tt>HAVING</tt> SQL-clause.
P
Pratik Naik 已提交
1188 1189 1190 1191 1192 1193
      # [:limit]
      #   An integer determining the limit on the number of rows that should be returned.
      # [:offset]
      #   An integer determining the offset from where the rows should be fetched. So at 5, it would skip the first 4 rows.
      # [:select]
      #   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
1194
      #   but not include the joined columns. Do not forget to include the primary and foreign keys, otherwise it will raise an error.
P
Pratik Naik 已提交
1195 1196 1197 1198
      # [:readonly]
      #   If true, all the associated objects are readonly through the association.
      # [:validate]
      #   If false, don't validate the associated objects when saving the parent object. +true+ by default.
D
Initial  
David Heinemeier Hansson 已提交
1199 1200 1201
      #
      # Option examples:
      #   has_and_belongs_to_many :projects
1202
      #   has_and_belongs_to_many :projects, :include => [ :milestones, :manager ]
D
Initial  
David Heinemeier Hansson 已提交
1203 1204
      #   has_and_belongs_to_many :nations, :class_name => "Country"
      #   has_and_belongs_to_many :categories, :join_table => "prods_cats"
1205
      #   has_and_belongs_to_many :categories, :readonly => true
1206
      #   has_and_belongs_to_many :active_projects, :join_table => 'developers_projects', :delete_sql =>
1207
      #   'DELETE FROM developers_projects WHERE active=1 AND developer_id = #{id} AND project_id = #{record.id}'
1208
      def has_and_belongs_to_many(association_id, options = {}, &extension)
1209
        reflection = create_has_and_belongs_to_many_reflection(association_id, options, &extension)
1210

1211 1212
        add_multiple_associated_validation_callbacks(reflection.name) unless options[:validate] == false
        add_multiple_associated_save_callbacks(reflection.name)
1213
        collection_accessor_methods(reflection, HasAndBelongsToManyAssociation)
D
Initial  
David Heinemeier Hansson 已提交
1214

1215 1216
        # Don't use a before_destroy callback since users' before_destroy
        # callbacks will be executed after the association is wiped out.
1217
        old_method = "destroy_without_habtm_shim_for_#{reflection.name}"
1218
        class_eval <<-end_eval unless method_defined?(old_method)
1219 1220
          alias_method :#{old_method}, :destroy_without_callbacks
          def destroy_without_callbacks
1221
            #{reflection.name}.clear
1222 1223 1224 1225
            #{old_method}
          end
        end_eval

1226
        add_association_callbacks(reflection.name, options)
D
Initial  
David Heinemeier Hansson 已提交
1227 1228 1229
      end

      private
P
Pratik Naik 已提交
1230 1231 1232 1233 1234
        # Generates a join table name from two provided table names.
        # The names in the join table namesme end up in lexicographic order.
        #
        #   join_table_name("members", "clubs")         # => "clubs_members"
        #   join_table_name("members", "special_clubs") # => "members_special_clubs"
D
Initial  
David Heinemeier Hansson 已提交
1235 1236 1237 1238 1239 1240 1241 1242 1243
        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
1244

1245
        def association_accessor_methods(reflection, association_proxy_class)
1246 1247
          ivar = "@#{reflection.name}"

1248
          define_method(reflection.name) do |*params|
1249
            force_reload = params.first unless params.empty?
1250 1251

            association = instance_variable_get(ivar) if instance_variable_defined?(ivar)
1252

1253
            if association.nil? || force_reload
1254
              association = association_proxy_class.new(self, reflection)
1255
              retval = association.reload
1256
              if retval.nil? and association_proxy_class == BelongsToAssociation
1257
                instance_variable_set(ivar, nil)
1258 1259
                return nil
              end
1260
              instance_variable_set(ivar, association)
1261
            end
1262 1263

            association.target.nil? ? nil : association
1264 1265
          end

1266 1267 1268 1269 1270
          define_method("loaded_#{reflection.name}?") do
            association = instance_variable_get(ivar) if instance_variable_defined?(ivar)
            association && association.loaded?
          end

1271
          define_method("#{reflection.name}=") do |new_value|
1272 1273
            association = instance_variable_get(ivar) if instance_variable_defined?(ivar)

1274
            if association.nil? || association.target != new_value
1275
              association = association_proxy_class.new(self, reflection)
1276
            end
1277

1278 1279 1280 1281
            if association_proxy_class == HasOneThroughAssociation
              association.create_through_record(new_value)
              self.send(reflection.name, new_value)
            else
P
pivotal 已提交
1282 1283
              association.replace(new_value)
              instance_variable_set(ivar, new_value.nil? ? nil : association)
1284
            end
1285
          end
1286

1287
          define_method("set_#{reflection.name}_target") do |target|
1288
            return if target.nil? and association_proxy_class == BelongsToAssociation
1289
            association = association_proxy_class.new(self, reflection)
1290
            association.target = target
1291
            instance_variable_set(ivar, association)
1292
          end
1293 1294
        end

1295 1296
        def collection_reader_method(reflection, association_proxy_class)
          define_method(reflection.name) do |*params|
1297 1298
            ivar = "@#{reflection.name}"

1299
            force_reload = params.first unless params.empty?
1300
            association = instance_variable_get(ivar) if instance_variable_defined?(ivar)
1301

1302
            unless association.respond_to?(:loaded?)
1303
              association = association_proxy_class.new(self, reflection)
1304
              instance_variable_set(ivar, association)
D
Initial  
David Heinemeier Hansson 已提交
1305
            end
1306

1307
            association.reload if force_reload
1308

1309 1310
            association
          end
1311 1312

          define_method("#{reflection.name.to_s.singularize}_ids") do
1313
            if send(reflection.name).loaded? || reflection.options[:finder_sql]
1314 1315
              send(reflection.name).map(&:id)
            else
1316
              send(reflection.name).all(:select => "#{reflection.quoted_table_name}.#{reflection.klass.primary_key}").map(&:id)
1317
            end
1318
          end
1319
        end
D
Initial  
David Heinemeier Hansson 已提交
1320

1321
        def collection_accessor_methods(reflection, association_proxy_class, writer = true)
1322 1323
          collection_reader_method(reflection, association_proxy_class)

1324 1325 1326 1327 1328 1329 1330
          if writer
            define_method("#{reflection.name}=") do |new_value|
              # Loads proxy class instance (defined in collection_reader_method) if not already loaded
              association = send(reflection.name)
              association.replace(new_value)
              association
            end
1331

1332 1333 1334 1335
            define_method("#{reflection.name.to_s.singularize}_ids=") do |new_value|
              ids = (new_value || []).reject { |nid| nid.blank? }
              send("#{reflection.name}=", reflection.class_name.constantize.find(ids))
            end
1336
          end
D
Initial  
David Heinemeier Hansson 已提交
1337
        end
1338

1339
        def add_single_associated_validation_callbacks(association_name)
1340 1341 1342 1343
          method_name = "validate_associated_records_for_#{association_name}".to_sym
          define_method(method_name) do
            association = instance_variable_get("@#{association_name}")
            if !association.nil?
1344
              errors.add association_name unless association.target.nil? || association.valid?
1345 1346
            end
          end
1347

1348 1349
          validate method_name
        end
1350

1351
        def add_multiple_associated_validation_callbacks(association_name)
1352
          method_name = "validate_associated_records_for_#{association_name}".to_sym
1353 1354
          ivar = "@#{association_name}"

1355
          define_method(method_name) do
1356 1357
            association = instance_variable_get(ivar) if instance_variable_defined?(ivar)

1358
            if association.respond_to?(:loaded?)
1359
              if new_record?
1360
                association
1361
              elsif association.loaded?
1362
                association.select { |record| record.new_record? }
1363 1364
              else
                association.target.select { |record| record.new_record? }
1365
              end.each do |record|
1366
                errors.add association_name unless record.valid?
D
Initial  
David Heinemeier Hansson 已提交
1367
              end
1368
            end
1369
          end
D
Initial  
David Heinemeier Hansson 已提交
1370

1371
          validate method_name
1372 1373 1374 1375
        end

        def add_multiple_associated_save_callbacks(association_name)
          ivar = "@#{association_name}"
1376

1377 1378 1379 1380 1381 1382 1383 1384 1385
          method_name = "before_save_associated_records_for_#{association_name}".to_sym
          define_method(method_name) do
            @new_record_before_save = new_record?
            true
          end
          before_save method_name

          method_name = "after_create_or_update_associated_records_for_#{association_name}".to_sym
          define_method(method_name) do
1386
            association = instance_variable_get(ivar) if instance_variable_defined?(ivar)
1387

1388 1389 1390 1391
            records_to_save = if @new_record_before_save
              association
            elsif association.respond_to?(:loaded?) && association.loaded?
              association.select { |record| record.new_record? }
1392 1393
            elsif association.respond_to?(:loaded?) && !association.loaded?
              association.target.select { |record| record.new_record? }
1394 1395 1396
            else
              []
            end
1397
            records_to_save.each { |record| association.send(:insert_record, record) } unless records_to_save.blank?
1398

1399 1400
            # reconstruct the SQL queries now that we know the owner's id
            association.send(:construct_sql) if association.respond_to?(:construct_sql)
1401
          end
1402

1403
          # Doesn't use after_save as that would save associations added in after_create/after_update twice
1404 1405
          after_create method_name
          after_update method_name
D
Initial  
David Heinemeier Hansson 已提交
1406
        end
1407

1408 1409
        def association_constructor_method(constructor, reflection, association_proxy_class)
          define_method("#{constructor}_#{reflection.name}") do |*params|
1410 1411
            ivar = "@#{reflection.name}"

1412 1413
            attributees      = params.first unless params.empty?
            replace_existing = params[1].nil? ? true : params[1]
1414
            association      = instance_variable_get(ivar) if instance_variable_defined?(ivar)
1415 1416

            if association.nil?
1417
              association = association_proxy_class.new(self, reflection)
1418
              instance_variable_set(ivar, association)
1419 1420
            end

1421 1422 1423 1424 1425
            if association_proxy_class == HasOneAssociation
              association.send(constructor, attributees, replace_existing)
            else
              association.send(constructor, attributees)
            end
1426 1427
          end
        end
1428

1429
        def find_with_associations(options = {})
1430
          catch :invalid_query do
1431
            join_dependency = JoinDependency.new(self, merge_includes(scope(:find, :include), options[:include]), options[:joins])
1432 1433 1434 1435
            rows = select_all_rows(options, join_dependency)
            return join_dependency.instantiate(rows)
          end
          []
1436
        end
1437

1438 1439 1440
        # Creates before_destroy callback methods that nullify, delete or destroy
        # has_many associated objects, according to the defined :dependent rule.
        #
1441 1442
        # See HasManyAssociation#delete_records.  Dependent associations
        # delete children, otherwise foreign key is set to NULL.
1443 1444 1445 1446 1447
        #
        # The +extra_conditions+ parameter, which is not used within the main
        # Active Record codebase, is meant to allow plugins to define extra
        # finder conditions.
        def configure_dependency_for_has_many(reflection, extra_conditions = nil)
1448 1449 1450 1451 1452 1453
          if reflection.options.include?(:dependent)
            # Add polymorphic type if the :as option is present
            dependent_conditions = []
            dependent_conditions << "#{reflection.primary_key_name} = \#{record.quoted_id}"
            dependent_conditions << "#{reflection.options[:as]}_type = '#{base_class.name}'" if reflection.options[:as]
            dependent_conditions << sanitize_sql(reflection.options[:conditions]) if reflection.options[:conditions]
1454
            dependent_conditions << extra_conditions if extra_conditions
1455
            dependent_conditions = dependent_conditions.collect {|where| "(#{where})" }.join(" AND ")
1456

1457 1458
            case reflection.options[:dependent]
              when :destroy
1459 1460
                method_name = "has_many_dependent_destroy_for_#{reflection.name}".to_sym
                define_method(method_name) do
1461
                  send(reflection.name).each { |o| o.destroy }
1462 1463
                end
                before_destroy method_name
1464
              when :delete_all
1465 1466 1467 1468 1469 1470 1471 1472
                module_eval %Q{
                  before_destroy do |record|
                    delete_all_has_many_dependencies(record,
                      "#{reflection.name}",
                      #{reflection.class_name},
                      "#{dependent_conditions}")
                  end
                }
1473
              when :nullify
1474 1475 1476 1477 1478 1479 1480 1481 1482
                module_eval %Q{
                  before_destroy do |record|
                    nullify_has_many_dependencies(record,
                      "#{reflection.name}",
                      #{reflection.class_name},
                      "#{reflection.primary_key_name}",
                      "#{dependent_conditions}")
                  end
                }
1483 1484 1485
              else
                raise ArgumentError, "The :dependent option expects either :destroy, :delete_all, or :nullify (#{reflection.options[:dependent].inspect})"
            end
1486 1487
          end
        end
1488

P
Pratik Naik 已提交
1489 1490
        # Creates before_destroy callback methods that nullify, delete or destroy
        # has_one associated objects, according to the defined :dependent rule.
1491
        def configure_dependency_for_has_one(reflection)
1492 1493 1494
          if reflection.options.include?(:dependent)
            case reflection.options[:dependent]
              when :destroy
1495 1496
                method_name = "has_one_dependent_destroy_for_#{reflection.name}".to_sym
                define_method(method_name) do
1497
                  association = send(reflection.name)
1498 1499 1500
                  association.destroy unless association.nil?
                end
                before_destroy method_name
1501
              when :delete
1502 1503
                method_name = "has_one_dependent_delete_for_#{reflection.name}".to_sym
                define_method(method_name) do
P
Pratik Naik 已提交
1504 1505 1506 1507
                  # Retrieve the associated object and delete it. The retrieval
                  # is necessary because there may be multiple associated objects
                  # with foreign keys pointing to this object, and we only want
                  # to delete the correct one, not all of them.
1508
                  association = send(reflection.name)
1509
                  association.delete unless association.nil?
1510 1511
                end
                before_destroy method_name
1512
              when :nullify
1513 1514
                method_name = "has_one_dependent_nullify_for_#{reflection.name}".to_sym
                define_method(method_name) do
1515 1516
                  association = send(reflection.name)
                  association.update_attribute(reflection.primary_key_name, nil) unless association.nil?
1517 1518
                end
                before_destroy method_name
1519 1520 1521
              else
                raise ArgumentError, "The :dependent option expects either :destroy, :delete or :nullify (#{reflection.options[:dependent].inspect})"
            end
1522 1523 1524
          end
        end

1525 1526 1527 1528
        def configure_dependency_for_belongs_to(reflection)
          if reflection.options.include?(:dependent)
            case reflection.options[:dependent]
              when :destroy
1529 1530
                method_name = "belongs_to_dependent_destroy_for_#{reflection.name}".to_sym
                define_method(method_name) do
1531
                  association = send(reflection.name)
1532 1533 1534
                  association.destroy unless association.nil?
                end
                before_destroy method_name
1535
              when :delete
1536 1537
                method_name = "belongs_to_dependent_delete_for_#{reflection.name}".to_sym
                define_method(method_name) do
1538
                  association = send(reflection.name)
1539
                  association.delete unless association.nil?
1540 1541
                end
                before_destroy method_name
1542 1543 1544 1545 1546 1547
              else
                raise ArgumentError, "The :dependent option expects either :destroy or :delete (#{reflection.options[:dependent].inspect})"
            end
          end
        end

1548 1549 1550 1551 1552 1553 1554 1555
        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

1556 1557 1558 1559
        mattr_accessor :valid_keys_for_has_many_association
        @@valid_keys_for_has_many_association = [
          :class_name, :table_name, :foreign_key, :primary_key,
          :dependent,
1560
          :select, :conditions, :include, :order, :group, :having, :limit, :offset,
1561 1562 1563 1564 1565
          :as, :through, :source, :source_type,
          :uniq,
          :finder_sql, :counter_sql,
          :before_add, :after_add, :before_remove, :after_remove,
          :extend, :readonly,
1566
          :validate
1567 1568
        ]

1569
        def create_has_many_reflection(association_id, options, &extension)
1570
          options.assert_valid_keys(valid_keys_for_has_many_association)
1571
          options[:extend] = create_extension_modules(association_id, extension, options[:extend])
1572

1573
          create_reflection(:has_many, association_id, options, self)
1574 1575
        end

1576 1577 1578 1579
        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,
1580
          :validate, :primary_key
1581 1582
        ]

1583
        def create_has_one_reflection(association_id, options)
1584
          options.assert_valid_keys(valid_keys_for_has_one_association)
1585
          create_reflection(:has_one, association_id, options, self)
1586
        end
1587

1588 1589
        def create_has_one_through_reflection(association_id, options)
          options.assert_valid_keys(
1590
            :class_name, :foreign_key, :remote, :select, :conditions, :order, :include, :dependent, :counter_cache, :extend, :as, :through, :source, :source_type, :validate
1591 1592 1593
          )
          create_reflection(:has_one, association_id, options, self)
        end
1594

1595 1596 1597 1598
        mattr_accessor :valid_keys_for_belongs_to_association
        @@valid_keys_for_belongs_to_association = [
          :class_name, :foreign_key, :foreign_type, :remote, :select, :conditions,
          :include, :dependent, :counter_cache, :extend, :polymorphic, :readonly,
1599
          :validate
1600 1601
        ]

1602
        def create_belongs_to_reflection(association_id, options)
1603
          options.assert_valid_keys(valid_keys_for_belongs_to_association)
1604 1605 1606 1607 1608 1609 1610 1611
          reflection = create_reflection(:belongs_to, association_id, options, self)

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

          reflection
        end
1612

1613 1614 1615
        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,
1616
          :select, :conditions, :include, :order, :group, :having, :limit, :offset,
1617
          :uniq,
1618
          :finder_sql, :counter_sql, :delete_sql, :insert_sql,
1619 1620 1621 1622 1623
          :before_add, :after_add, :before_remove, :after_remove,
          :extend, :readonly,
          :validate
        ]

1624
        def create_has_and_belongs_to_many_reflection(association_id, options, &extension)
1625
          options.assert_valid_keys(valid_keys_for_has_and_belongs_to_many_association)
1626

1627
          options[:extend] = create_extension_modules(association_id, extension, options[:extend])
1628 1629 1630 1631

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

          reflection.options[:join_table] ||= join_table_name(undecorated_table_name(self.to_s), undecorated_table_name(reflection.class_name))
1632

1633 1634 1635
          reflection
        end

1636
        def reflect_on_included_associations(associations)
1637
          [ associations ].flatten.collect { |association| reflect_on_association(association.to_s.intern) }
1638 1639
        end

1640 1641 1642
        def guard_against_unlimitable_reflections(reflections, options)
          if (options[:offset] || options[:limit]) && !using_limitable_reflections?(reflections)
            raise(
1643
              ConfigurationError,
1644 1645 1646 1647 1648
              "You can not use offset and limit together with has_many or has_and_belongs_to_many associations"
            )
          end
        end

1649
        def select_all_rows(options, join_dependency)
1650
          connection.select_all(
1651
            construct_finder_sql_with_included_associations(options, join_dependency),
1652 1653 1654
            "#{name} Load Including Associations"
          )
        end
1655

1656
        def construct_finder_sql_with_included_associations(options, join_dependency)
1657
          scope = scope(:find)
1658
          sql = "SELECT #{column_aliases(join_dependency)} FROM #{(scope && scope[:from]) || options[:from] || quoted_table_name} "
1659
          sql << join_dependency.join_associations.collect{|join| join.association_join }.join
1660

1661
          add_joins!(sql, options[:joins], scope)
1662
          add_conditions!(sql, options[:conditions], scope)
1663
          add_limited_ids_condition!(sql, options, join_dependency) if !using_limitable_reflections?(join_dependency.reflections) && ((scope && scope[:limit]) || options[:limit])
1664

1665
          add_group!(sql, options[:group], options[:having], scope)
1666
          add_order!(sql, options[:order], scope)
1667
          add_limit!(sql, options, scope) if using_limitable_reflections?(join_dependency.reflections)
1668
          add_lock!(sql, options, scope)
1669

1670 1671
          return sanitize_sql(sql)
        end
1672

1673 1674
        def add_limited_ids_condition!(sql, options, join_dependency)
          unless (id_list = select_limited_ids_list(options, join_dependency)).empty?
1675
            sql << "#{condition_word(sql)} #{connection.quote_table_name table_name}.#{primary_key} IN (#{id_list}) "
1676 1677
          else
            throw :invalid_query
1678 1679
          end
        end
1680

1681
        def select_limited_ids_list(options, join_dependency)
1682 1683
          pk = columns_hash[primary_key]

1684
          connection.select_all(
1685
            construct_finder_sql_for_association_limiting(options, join_dependency),
1686
            "#{name} Load IDs For Limited Eager Loading"
1687
          ).collect { |row| connection.quote(row[primary_key], pk) }.join(", ")
1688
        end
1689

1690
        def construct_finder_sql_for_association_limiting(options, join_dependency)
1691
          scope       = scope(:find)
1692 1693 1694 1695 1696

          # Only join tables referenced in order or conditions since this is particularly slow on the pre-query.
          tables_from_conditions = conditions_tables(options)
          tables_from_order      = order_tables(options)
          all_tables             = tables_from_conditions + tables_from_order
1697 1698 1699 1700
          distinct_join_associations = all_tables.uniq.map{|table|
            join_dependency.joins_for_table_name(table)
          }.flatten.compact.uniq

1701 1702 1703 1704 1705
          order = options[:order]
          if scoped_order = (scope && scope[:order])
            order = order ? "#{order}, #{scoped_order}" : scoped_order
          end

1706
          is_distinct = !options[:joins].blank? || include_eager_conditions?(options, tables_from_conditions) || include_eager_order?(options, tables_from_order)
1707
          sql = "SELECT "
1708
          if is_distinct
1709
            sql << connection.distinct("#{connection.quote_table_name table_name}.#{primary_key}", order)
1710 1711 1712
          else
            sql << primary_key
          end
1713
          sql << " FROM #{connection.quote_table_name table_name} "
1714

1715
          if is_distinct
1716
            sql << distinct_join_associations.collect { |assoc| assoc.association_join }.join
1717
            add_joins!(sql, options[:joins], scope)
1718
          end
1719

1720
          add_conditions!(sql, options[:conditions], scope)
1721
          add_group!(sql, options[:group], options[:having], scope)
1722

1723 1724
          if order && is_distinct
            connection.add_order_by_for_association_limiting!(sql, :order => order)
1725 1726
          else
            add_order!(sql, options[:order], scope)
1727
          end
1728

1729
          add_limit!(sql, options, scope)
1730

1731 1732
          return sanitize_sql(sql)
        end
1733

1734 1735 1736 1737 1738
        def tables_in_string(string)
          return [] if string.blank?
          string.scan(/([\.a-zA-Z_]+).?\./).flatten
        end

1739
        def conditions_tables(options)
1740 1741 1742 1743 1744
          # look in both sets of conditions
          conditions = [scope(:find, :conditions), options[:conditions]].inject([]) do |all, cond|
            case cond
              when nil   then all
              when Array then all << cond.first
1745
              when Hash  then all << cond.keys
1746 1747 1748
              else            all << cond
            end
          end
1749
          tables_in_string(conditions.join(' '))
1750 1751 1752
        end

        def order_tables(options)
1753
          order = [options[:order], scope(:find, :order) ].join(", ")
1754
          return [] unless order && order.is_a?(String)
1755
          tables_in_string(order)
1756 1757
        end

1758 1759 1760
        def selects_tables(options)
          select = options[:select]
          return [] unless select && select.is_a?(String)
1761 1762 1763 1764 1765 1766 1767 1768 1769 1770 1771 1772 1773 1774 1775 1776 1777 1778
          tables_in_string(select)
        end

        def joined_tables(options)
          scope = scope(:find)
          joins = options[:joins]
          merged_joins = scope && scope[:joins] && joins ? merge_joins(scope[:joins], joins) : (joins || scope && scope[:joins])
          [table_name] + case merged_joins
          when Symbol, Hash, Array
            if array_of_strings?(merged_joins)
              tables_in_string(merged_joins.join(' '))
            else
              join_dependency = ActiveRecord::Associations::ClassMethods::InnerJoinDependency.new(self, merged_joins, nil)
              [table_name] + join_dependency.join_associations.collect {|join_association| [join_association.aliased_join_table_name, join_association.aliased_table_name]}.flatten.compact
            end
          else
            tables_in_string(merged_joins)
          end
1779 1780
        end

1781
        # Checks if the conditions reference a table other than the current model table
1782 1783
        def include_eager_conditions?(options, tables = nil, joined_tables = nil)
          ((tables || conditions_tables(options)) - (joined_tables || joined_tables(options))).any?
1784
        end
1785

1786
        # Checks if the query order references a table other than the current model's table.
1787 1788
        def include_eager_order?(options, tables = nil, joined_tables = nil)
          ((tables || order_tables(options)) - (joined_tables || joined_tables(options))).any?
1789
        end
1790

1791 1792
        def include_eager_select?(options, joined_tables = nil)
          (selects_tables(options) - (joined_tables || joined_tables(options))).any?
1793 1794
        end

1795
        def references_eager_loaded_tables?(options)
1796 1797
          joined_tables = joined_tables(options)
          include_eager_order?(options, nil, joined_tables) || include_eager_conditions?(options, nil, joined_tables) || include_eager_select?(options, joined_tables)
1798 1799
        end

1800 1801 1802 1803
        def using_limitable_reflections?(reflections)
          reflections.reject { |r| [ :belongs_to, :has_one ].include?(r.macro) }.length.zero?
        end

1804 1805
        def column_aliases(join_dependency)
          join_dependency.joins.collect{|join| join.column_names_with_alias.collect{|column_name, aliased_name|
1806
              "#{connection.quote_table_name join.aliased_table_name}.#{connection.quote_column_name column_name} AS #{aliased_name}"}}.flatten.join(", ")
1807 1808
        end

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

1823 1824 1825
        def condition_word(sql)
          sql =~ /where/i ? " AND " : "WHERE "
        end
1826

1827
        def create_extension_modules(association_id, block_extension, extensions)
1828
          if block_extension
1829
            extension_module_name = "#{self.to_s.demodulize}#{association_id.to_s.camelize}AssociationExtension"
1830

1831
            silence_warnings do
1832
              self.parent.const_set(extension_module_name, Module.new(&block_extension))
1833
            end
1834
            Array(extensions).push("#{self.parent}::#{extension_module_name}".constantize)
1835 1836
          else
            Array(extensions)
1837 1838
          end
        end
1839

D
David Heinemeier Hansson 已提交
1840
        class JoinDependency # :nodoc:
1841
          attr_reader :joins, :reflections, :table_aliases
1842

1843
          def initialize(base, associations, joins)
1844
            @joins                 = [JoinBase.new(base, joins)]
1845 1846 1847 1848
            @associations          = associations
            @reflections           = []
            @base_records_hash     = {}
            @base_records_in_order = []
1849 1850
            @table_aliases         = Hash.new { |aliases, table| aliases[table] = 0 }
            @table_aliases[base.table_name] = 1
1851 1852 1853 1854 1855 1856 1857 1858 1859 1860 1861 1862 1863 1864 1865 1866 1867
            build(associations)
          end

          def join_associations
            @joins[1..-1].to_a
          end

          def join_base
            @joins[0]
          end

          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
1868
              construct(@base_records_hash[primary_id], @associations, join_associations.dup, row)
1869
            end
1870
            remove_duplicate_results!(join_base.active_record, @base_records_in_order, @associations)
1871 1872 1873
            return @base_records_in_order
          end

1874 1875 1876 1877 1878 1879 1880 1881 1882 1883 1884 1885 1886 1887 1888 1889 1890
          def remove_duplicate_results!(base, records, associations)
            case associations
              when Symbol, String
                reflection = base.reflections[associations]
                if reflection && [:has_many, :has_and_belongs_to_many].include?(reflection.macro)
                  records.each { |record| record.send(reflection.name).target.uniq! }
                end
              when Array
                associations.each do |association|
                  remove_duplicate_results!(base, records, association)
                end
              when Hash
                associations.keys.each do |name|
                  reflection = base.reflections[name]
                  is_collection = [:has_many, :has_and_belongs_to_many].include?(reflection.macro)

                  parent_records = records.map do |record|
1891 1892 1893 1894
                    descendant = record.send(reflection.name)
                    next unless descendant
                    descendant.target.uniq! if is_collection
                    descendant
1895 1896 1897 1898 1899 1900 1901
                  end.flatten.compact

                  remove_duplicate_results!(reflection.class_name.constantize, parent_records, associations[name]) unless parent_records.empty?
                end
            end
          end

1902
          def join_for_table_name(table_name)
1903 1904 1905
            join = (@joins.select{|j|j.aliased_table_name == table_name.gsub(/^\"(.*)\"$/){$1} }.first) rescue nil
            return join unless join.nil?
            @joins.select{|j|j.is_a?(JoinAssociation) && j.aliased_join_table_name == table_name.gsub(/^\"(.*)\"$/){$1} }.first rescue nil
1906 1907 1908 1909 1910 1911 1912 1913 1914 1915 1916 1917 1918 1919 1920
          end

          def joins_for_table_name(table_name)
            join = join_for_table_name(table_name)
            result = nil
            if join && join.is_a?(JoinAssociation)
              result = [join]
              if join.parent && join.parent.is_a?(JoinAssociation)
                result = joins_for_table_name(join.parent.aliased_table_name) +
                         result
              end
            end
            result
          end

1921 1922 1923 1924 1925 1926 1927 1928
          protected
            def build(associations, parent = nil)
              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
1929
                  @joins << build_join_association(reflection, parent)
1930 1931 1932 1933 1934 1935 1936 1937 1938 1939 1940 1941 1942 1943
                when Array
                  associations.each do |association|
                    build(association, parent)
                  end
                when Hash
                  associations.keys.sort{|a,b|a.to_s<=>b.to_s}.each do |name|
                    build(name, parent)
                    build(associations[name])
                  end
                else
                  raise ConfigurationError, associations.inspect
              end
            end

1944 1945 1946 1947 1948
            # overridden in InnerJoinDependency subclass
            def build_join_association(reflection, parent)
              JoinAssociation.new(reflection, self, parent)
            end

1949 1950 1951 1952 1953 1954 1955 1956 1957 1958 1959 1960 1961 1962 1963 1964 1965 1966 1967 1968 1969 1970 1971 1972 1973 1974
            def construct(parent, associations, joins, row)
              case associations
                when Symbol, String
                  while (join = joins.shift).reflection.name.to_s != associations.to_s
                    raise ConfigurationError, "Not Enough Associations" if joins.empty?
                  end
                  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|
                    association = construct_association(parent, joins.shift, row)
                    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
1975

1976 1977
                  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)
1978
                  collection.target.push(association)
1979 1980
                when :has_one
                  return if record.id.to_s != join.parent.record_id(row).to_s
1981
                  return if record.instance_variable_defined?("@#{join.reflection.name}")
1982 1983 1984
                  association = join.instantiate(row) unless row[join.aliased_primary_key].nil?
                  record.send("set_#{join.reflection.name}_target", association)
                when :belongs_to
1985 1986 1987 1988 1989 1990 1991 1992 1993
                  return if record.id.to_s != join.parent.record_id(row).to_s or row[join.aliased_primary_key].nil?
                  association = join.instantiate(row)
                  record.send("set_#{join.reflection.name}_target", association)
                else
                  raise ConfigurationError, "unknown macro: #{join.reflection.macro}"
              end
              return association
            end

D
David Heinemeier Hansson 已提交
1994
          class JoinBase # :nodoc:
1995
            attr_reader :active_record, :table_joins
1996
            delegate    :table_name, :column_names, :primary_key, :reflections, :sanitize_sql, :to => :active_record
1997

1998
            def initialize(active_record, joins = nil)
1999 2000
              @active_record = active_record
              @cached_record = {}
2001
              @table_joins   = joins
2002 2003 2004 2005 2006 2007 2008
            end

            def aliased_prefix
              "t0"
            end

            def aliased_primary_key
2009
              "#{aliased_prefix}_r0"
2010 2011 2012 2013 2014 2015 2016
            end

            def aliased_table_name
              active_record.table_name
            end

            def column_names_with_alias
2017
              unless defined?(@column_names_with_alias)
2018
                @column_names_with_alias = []
2019

2020
                ([primary_key] + (column_names - [primary_key])).each_with_index do |column_name, i|
2021
                  @column_names_with_alias << [column_name, "#{aliased_prefix}_r#{i}"]
2022 2023
                end
              end
2024 2025

              @column_names_with_alias
2026 2027 2028 2029 2030 2031 2032 2033 2034 2035 2036
            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)
2037
              @cached_record[record_id(row)] ||= active_record.send(:instantiate, extract_record(row))
2038 2039 2040
            end
          end

D
David Heinemeier Hansson 已提交
2041
          class JoinAssociation < JoinBase # :nodoc:
2042
            attr_reader :reflection, :parent, :aliased_table_name, :aliased_prefix, :aliased_join_table_name, :parent_table_name
2043
            delegate    :options, :klass, :through_reflection, :source_reflection, :to => :reflection
2044 2045

            def initialize(reflection, join_dependency, parent = nil)
2046 2047 2048 2049 2050
              reflection.check_validity!
              if reflection.options[:polymorphic]
                raise EagerLoadPolymorphicError.new(reflection)
              end

2051
              super(reflection.klass)
2052
              @join_dependency    = join_dependency
2053 2054 2055
              @parent             = parent
              @reflection         = reflection
              @aliased_prefix     = "t#{ join_dependency.joins.size }"
2056
              @parent_table_name  = parent.active_record.table_name
2057
              @aliased_table_name = aliased_table_name_for(table_name)
2058

2059 2060
              if reflection.macro == :has_and_belongs_to_many
                @aliased_join_table_name = aliased_table_name_for(reflection.options[:join_table], "_join")
2061
              end
2062

2063
              if [:has_many, :has_one].include?(reflection.macro) && reflection.options[:through]
2064
                @aliased_join_table_name = aliased_table_name_for(reflection.through_reflection.klass.table_name, "_join")
2065
              end
2066 2067 2068
            end

            def association_join
2069
              connection = reflection.active_record.connection
2070
              join = case reflection.macro
2071
                when :has_and_belongs_to_many
2072
                  " #{join_type} %s ON %s.%s = %s.%s " % [
2073
                     table_alias_for(options[:join_table], aliased_join_table_name),
2074
                     connection.quote_table_name(aliased_join_table_name),
2075
                     options[:foreign_key] || reflection.active_record.to_s.foreign_key,
2076
                     connection.quote_table_name(parent.aliased_table_name),
2077
                     reflection.active_record.primary_key] +
2078
                  " #{join_type} %s ON %s.%s = %s.%s " % [
2079
                     table_name_and_alias,
2080 2081
                     connection.quote_table_name(aliased_table_name),
                     klass.primary_key,
2082
                     connection.quote_table_name(aliased_join_table_name),
2083
                     options[:association_foreign_key] || klass.to_s.foreign_key
2084 2085
                     ]
                when :has_many, :has_one
2086
                  case
2087
                    when reflection.options[:through]
2088
                      through_conditions = through_reflection.options[:conditions] ? "AND #{interpolate_sql(sanitize_sql(through_reflection.options[:conditions]))}" : ''
2089 2090 2091 2092

                      jt_foreign_key = jt_as_extra = jt_source_extra = jt_sti_extra = nil
                      first_key = second_key = as_extra = nil

2093
                      if through_reflection.options[:as] # has_many :through against a polymorphic join
2094 2095
                        jt_foreign_key = through_reflection.options[:as].to_s + '_id'
                        jt_as_extra = " AND %s.%s = %s" % [
2096 2097
                          connection.quote_table_name(aliased_join_table_name),
                          connection.quote_column_name(through_reflection.options[:as].to_s + '_type'),
2098
                          klass.quote_value(parent.active_record.base_class.name)
2099
                        ]
2100
                      else
2101
                        jt_foreign_key = through_reflection.primary_key_name
2102
                      end
2103

2104 2105
                      case source_reflection.macro
                      when :has_many
2106 2107 2108
                        if source_reflection.options[:as]
                          first_key   = "#{source_reflection.options[:as]}_id"
                          second_key  = options[:foreign_key] || primary_key
2109
                          as_extra    = " AND %s.%s = %s" % [
2110 2111
                            connection.quote_table_name(aliased_table_name),
                            connection.quote_column_name("#{source_reflection.options[:as]}_type"),
2112
                            klass.quote_value(source_reflection.active_record.base_class.name)
2113 2114
                          ]
                        else
2115 2116 2117
                          first_key   = through_reflection.klass.base_class.to_s.foreign_key
                          second_key  = options[:foreign_key] || primary_key
                        end
2118

2119 2120
                        unless through_reflection.klass.descends_from_active_record?
                          jt_sti_extra = " AND %s.%s = %s" % [
2121 2122
                            connection.quote_table_name(aliased_join_table_name),
                            connection.quote_column_name(through_reflection.active_record.inheritance_column),
2123
                            through_reflection.klass.quote_value(through_reflection.klass.sti_name)]
2124 2125 2126 2127 2128 2129
                        end
                      when :belongs_to
                        first_key = primary_key
                        if reflection.options[:source_type]
                          second_key = source_reflection.association_foreign_key
                          jt_source_extra = " AND %s.%s = %s" % [
2130 2131
                            connection.quote_table_name(aliased_join_table_name),
                            connection.quote_column_name(reflection.source_reflection.options[:foreign_type]),
2132
                            klass.quote_value(reflection.options[:source_type])
2133
                          ]
2134
                        else
2135
                          second_key = source_reflection.primary_key_name
2136
                        end
2137
                      end
2138

2139
                      " #{join_type} %s ON (%s.%s = %s.%s%s%s%s) " % [
2140
                        table_alias_for(through_reflection.klass.table_name, aliased_join_table_name),
2141
                        connection.quote_table_name(parent.aliased_table_name),
2142
                        connection.quote_column_name(parent.primary_key),
2143
                        connection.quote_table_name(aliased_join_table_name),
2144
                        connection.quote_column_name(jt_foreign_key),
2145 2146
                        jt_as_extra, jt_source_extra, jt_sti_extra
                      ] +
2147
                      " #{join_type} %s ON (%s.%s = %s.%s%s) " % [
2148
                        table_name_and_alias,
2149
                        connection.quote_table_name(aliased_table_name),
2150
                        connection.quote_column_name(first_key),
2151
                        connection.quote_table_name(aliased_join_table_name),
2152
                        connection.quote_column_name(second_key),
2153 2154 2155
                        as_extra
                      ]

2156
                    when reflection.options[:as] && [:has_many, :has_one].include?(reflection.macro)
2157
                      " #{join_type} %s ON %s.%s = %s.%s AND %s.%s = %s" % [
2158
                        table_name_and_alias,
2159
                        connection.quote_table_name(aliased_table_name),
2160
                        "#{reflection.options[:as]}_id",
2161
                        connection.quote_table_name(parent.aliased_table_name),
2162
                        parent.primary_key,
2163
                        connection.quote_table_name(aliased_table_name),
2164
                        "#{reflection.options[:as]}_type",
2165
                        klass.quote_value(parent.active_record.base_class.name)
2166 2167
                      ]
                    else
2168
                      foreign_key = options[:foreign_key] || reflection.active_record.name.foreign_key
2169
                      " #{join_type} %s ON %s.%s = %s.%s " % [
2170
                        table_name_and_alias,
2171
                        aliased_table_name,
2172
                        foreign_key,
2173
                        parent.aliased_table_name,
2174
                        parent.primary_key
2175 2176
                      ]
                  end
2177
                when :belongs_to
2178
                  " #{join_type} %s ON %s.%s = %s.%s " % [
2179
                     table_name_and_alias,
2180 2181
                     connection.quote_table_name(aliased_table_name),
                     reflection.klass.primary_key,
2182
                     connection.quote_table_name(parent.aliased_table_name),
2183
                     options[:foreign_key] || reflection.primary_key_name
2184 2185 2186
                    ]
                else
                  ""
2187
              end || ''
2188 2189
              join << %(AND %s) % [
                klass.send(:type_condition, aliased_table_name)] unless klass.descends_from_active_record?
2190 2191 2192 2193 2194

              [through_reflection, reflection].each do |ref|
                join << "AND #{interpolate_sql(sanitize_sql(ref.options[:conditions]))} " if ref && ref.options[:conditions]
              end

2195
              join
2196
            end
2197

2198
            protected
2199

2200 2201 2202 2203 2204 2205 2206 2207 2208 2209 2210 2211 2212 2213 2214 2215 2216
              def aliased_table_name_for(name, suffix = nil)
                if !parent.table_joins.blank? && parent.table_joins.to_s.downcase =~ %r{join(\s+\w+)?\s+#{name.downcase}\son}
                  @join_dependency.table_aliases[name] += 1
                end

                unless @join_dependency.table_aliases[name].zero?
                  # if the table name has been used, then use an alias
                  name = active_record.connection.table_alias_for "#{pluralize(reflection.name)}_#{parent_table_name}#{suffix}"
                  table_index = @join_dependency.table_aliases[name]
                  @join_dependency.table_aliases[name] += 1
                  name = name[0..active_record.connection.table_alias_length-3] + "_#{table_index+1}" if table_index > 0
                else
                  @join_dependency.table_aliases[name] += 1
                end

                name
              end
2217

2218 2219 2220
              def pluralize(table_name)
                ActiveRecord::Base.pluralize_table_names ? table_name.to_s.pluralize : table_name
              end
2221

2222
              def table_alias_for(table_name, table_alias)
2223
                 "#{reflection.active_record.connection.quote_table_name(table_name)} #{table_alias if table_name != table_alias}".strip
2224 2225 2226 2227 2228
              end

              def table_name_and_alias
                table_alias_for table_name, @aliased_table_name
              end
2229 2230

              def interpolate_sql(sql)
2231 2232
                instance_eval("%@#{sql.gsub('@', '\@')}@")
              end
2233 2234

            private
2235 2236 2237 2238
              def join_type
                "LEFT OUTER JOIN"
              end
          end
2239 2240 2241 2242 2243 2244 2245 2246 2247
        end

        class InnerJoinDependency < JoinDependency # :nodoc:
          protected
            def build_join_association(reflection, parent)
              InnerJoinAssociation.new(reflection, self, parent)
            end

          class InnerJoinAssociation < JoinAssociation
2248 2249 2250 2251
            private
              def join_type
                "INNER JOIN"
              end
2252 2253
          end
        end
2254

2255
    end
D
Initial  
David Heinemeier Hansson 已提交
2256
  end
2257
end