associations.rb 107.8 KB
Newer Older
1
require 'active_record/associations/association_proxy'
D
Initial  
David Heinemeier Hansson 已提交
2
require 'active_record/associations/association_collection'
3
require 'active_record/associations/belongs_to_association'
4
require 'active_record/associations/belongs_to_polymorphic_association'
5
require 'active_record/associations/has_one_association'
D
Initial  
David Heinemeier Hansson 已提交
6
require 'active_record/associations/has_many_association'
7
require 'active_record/associations/has_many_through_association'
D
Initial  
David Heinemeier Hansson 已提交
8
require 'active_record/associations/has_and_belongs_to_many_association'
9
require 'active_record/associations/has_one_through_association'
D
Initial  
David Heinemeier Hansson 已提交
10 11

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

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

24 25 26 27 28
  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
29

D
David Heinemeier Hansson 已提交
30
  class HasManyThroughSourceAssociationNotFoundError < ActiveRecordError #:nodoc:
31
    def initialize(reflection)
32 33 34 35
      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'}?")
36 37 38
    end
  end

39
  class HasManyThroughSourceAssociationMacroError < ActiveRecordError #:nodoc:
40
    def initialize(reflection)
41 42 43
      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.")
44 45 46
    end
  end

47 48 49 50 51
  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
52 53 54 55 56 57
  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

58 59 60 61 62 63
  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 已提交
64
  class EagerLoadPolymorphicError < ActiveRecordError #:nodoc:
65
    def initialize(reflection)
66
      super("Can not eagerly load the polymorphic association #{reflection.name.inspect}")
67 68 69
    end
  end

70 71
  class ReadOnlyAssociation < ActiveRecordError #:nodoc:
    def initialize(reflection)
72
      super("Can not add to a has_many :through association.  Try adding to #{reflection.through_reflection.name.inspect}.")
73 74 75
    end
  end

D
Initial  
David Heinemeier Hansson 已提交
76
  module Associations # :nodoc:
77
    def self.included(base)
78 79 80
      base.extend(ClassMethods)
    end

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

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

715
        configure_dependency_for_has_many(reflection)
716

717 718
        add_multiple_associated_validation_callbacks(reflection.name) unless options[:validate] == false
        add_multiple_associated_save_callbacks(reflection.name)
719 720
        add_association_callbacks(reflection.name, reflection.options)

721
        if options[:through]
722
          collection_accessor_methods(reflection, HasManyThroughAssociation)
723 724
        else
          collection_accessor_methods(reflection, HasManyAssociation)
D
Initial  
David Heinemeier Hansson 已提交
725 726 727
        end
      end

728
      # Adds the following methods for retrieval and query of a single associated object:
729
      # +association+ is replaced with the symbol passed as the first argument, so
730
      # <tt>has_one :manager</tt> would add among others <tt>manager.nil?</tt>.
P
Pratik Naik 已提交
731 732
      # * <tt>association(force_reload = false)</tt> - Returns the associated object. +nil+ is returned if none is found.
      # * <tt>association=(associate)</tt> - Assigns the associate object, extracts the primary key, sets it as the foreign key,
D
Initial  
David Heinemeier Hansson 已提交
733
      #   and saves the associate object.
P
Pratik Naik 已提交
734 735
      # * <tt>association.nil?</tt> - Returns +true+ if there is no associated object.
      # * <tt>build_association(attributes = {})</tt> - Returns a new object of the associated type that has been instantiated
736 737
      #   with +attributes+ and linked to this object through a foreign key, but has not yet been saved. Note: This ONLY works if
      #   an association already exists. It will NOT work if the association is +nil+.
P
Pratik Naik 已提交
738
      # * <tt>create_association(attributes = {})</tt> - Returns a new object of the associated type that has been instantiated
739
      #   with +attributes+, linked to this object through a foreign key, and that has already been saved (if it passed the validation).
D
Initial  
David Heinemeier Hansson 已提交
740 741
      #
      # Example: An Account class declares <tt>has_one :beneficiary</tt>, which will add:
742
      # * <tt>Account#beneficiary</tt> (similar to <tt>Beneficiary.find(:first, :conditions => "account_id = #{id}")</tt>)
D
Initial  
David Heinemeier Hansson 已提交
743 744
      # * <tt>Account#beneficiary=(beneficiary)</tt> (similar to <tt>beneficiary.account_id = account.id; beneficiary.save</tt>)
      # * <tt>Account#beneficiary.nil?</tt>
745 746 747
      # * <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>)
      #
D
Initial  
David Heinemeier Hansson 已提交
748
      # The declaration can also include an options hash to specialize the behavior of the association.
749
      #
D
Initial  
David Heinemeier Hansson 已提交
750
      # Options are:
P
Pratik Naik 已提交
751 752 753 754
      # * <tt>:class_name</tt> - Specify the class name of the association. Use it only if that name can't be inferred
      #   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.
      # * <tt>:conditions</tt> - Specify the conditions that the associated object must meet in order to be included as a +WHERE+
755
      #   SQL fragment, such as <tt>rank = 5</tt>.
P
Pratik Naik 已提交
756 757 758
      # * <tt>:order</tt> - Specify the order in which the associated objects are returned as an <tt>ORDER BY</tt> SQL fragment,
      #   such as <tt>last_name, first_name DESC</tt>.
      # * <tt>:dependent</tt> - If set to <tt>:destroy</tt>, the associated object is destroyed when this object is. If set to
759 760
      #   <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 已提交
761 762 763
      # * <tt>:foreign_key</tt> - Specify the foreign key used for the association. By default this is guessed to be the name
      #   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>.
764
      # * <tt>:primary_key</tt> - Specify the method that returns the primary key used for the association. By default this is +id+.
P
Pratik Naik 已提交
765 766
      # * <tt>:include</tt> - Specify second-order associations that should be eager loaded when this object is loaded.
      # * <tt>:as</tt> - Specifies a polymorphic interface (See <tt>belongs_to</tt>).
767 768
      # * <tt>:select</tt> - By default, this is <tt>*</tt> as in <tt>SELECT * FROM</tt>, but can be changed if, for example, you want to do a join
      #   but not include the joined columns. Do not forget to include the primary and foreign keys, otherwise it will raise an error.
769 770 771
      # * <tt>:through</tt>: Specifies a Join Model through which to perform the query.  Options for <tt>:class_name</tt> and <tt>:foreign_key</tt>
      #   are ignored, as the association uses the source reflection. You can only use a <tt>:through</tt> query through a 
      #   <tt>has_one</tt> or <tt>belongs_to</tt> association on the join model.
P
Pratik Naik 已提交
772
      # * <tt>:source</tt> - Specifies the source association name used by <tt>has_one :through</tt> queries.  Only use it if the name cannot be
773
      #   inferred from the association.  <tt>has_one :favorite, :through => :favorites</tt> will look for a
P
Pratik Naik 已提交
774
      #   <tt>:favorite</tt> on Favorite, unless a <tt>:source</tt> is given.      
775 776
      # * <tt>:source_type</tt> - Specifies type of the source association used by <tt>has_one :through</tt> queries where the source
      #   association is a polymorphic +belongs_to+.      
P
Pratik Naik 已提交
777
      # * <tt>:readonly</tt> - If true, the associated object is readonly through the association.
778
      # * <tt>:validate</tt> - If false, don't validate the associated object when saving the parent object. +false+ by default.
779
      # * <tt>:accessible</tt> - Mass assignment is allowed for this assocation (similar to <tt>ActiveRecord::Base#attr_accessible</tt>).
780
      #
D
Initial  
David Heinemeier Hansson 已提交
781
      # Option examples:
782
      #   has_one :credit_card, :dependent => :destroy  # destroys the associated credit card
783
      #   has_one :credit_card, :dependent => :nullify  # updates the associated records foreign key value to NULL rather than destroying it
D
Initial  
David Heinemeier Hansson 已提交
784 785
      #   has_one :last_comment, :class_name => "Comment", :order => "posted_on"
      #   has_one :project_manager, :class_name => "Person", :conditions => "role = 'project_manager'"
786
      #   has_one :attachment, :as => :attachable
787
      #   has_one :boss, :readonly => :true
788 789
      #   has_one :club, :through => :membership
      #   has_one :primary_address, :through => :addressables, :conditions => ["addressable.primary = ?", true], :source => :addressable
D
Initial  
David Heinemeier Hansson 已提交
790
      def has_one(association_id, options = {})
791 792 793 794 795
        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 已提交
796

797
          ivar = "@#{reflection.name}"
798

799 800 801
          method_name = "has_one_after_save_for_#{reflection.name}".to_sym
          define_method(method_name) do
            association = instance_variable_get("#{ivar}") if instance_variable_defined?("#{ivar}")
802

803 804 805 806
            if !association.nil? && (new_record? || association.new_record? || association["#{reflection.primary_key_name}"] != id)
              association["#{reflection.primary_key_name}"] = id
              association.save(true)
            end
807
          end
808
          after_save method_name
809

810
          add_single_associated_validation_callbacks(reflection.name) if options[:validate] == true
811 812 813
          association_accessor_methods(reflection, HasOneAssociation)
          association_constructor_method(:build,  reflection, HasOneAssociation)
          association_constructor_method(:create, reflection, HasOneAssociation)
814

815 816
          configure_dependency_for_has_one(reflection)
        end
D
Initial  
David Heinemeier Hansson 已提交
817 818
      end

819
      # Adds the following methods for retrieval and query for a single associated object for which this object holds an id:
820
      # +association+ is replaced with the symbol passed as the first argument, so
821
      # <tt>belongs_to :author</tt> would add among others <tt>author.nil?</tt>.
P
Pratik Naik 已提交
822 823 824 825
      # * <tt>association(force_reload = false)</tt> - Returns the associated object. +nil+ is returned if none is found.
      # * <tt>association=(associate)</tt> - Assigns the associate object, extracts the primary key, and sets it as the foreign key.
      # * <tt>association.nil?</tt> - Returns +true+ if there is no associated object.
      # * <tt>build_association(attributes = {})</tt> - Returns a new object of the associated type that has been instantiated
826
      #   with +attributes+ and linked to this object through a foreign key, but has not yet been saved.
P
Pratik Naik 已提交
827
      # * <tt>create_association(attributes = {})</tt> - Returns a new object of the associated type that has been instantiated
828
      #   with +attributes+, linked to this object through a foreign key, and that has already been saved (if it passed the validation).
D
Initial  
David Heinemeier Hansson 已提交
829
      #
830
      # Example: A Post class declares <tt>belongs_to :author</tt>, which will add:
D
Initial  
David Heinemeier Hansson 已提交
831 832 833 834
      # * <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 已提交
835 836
      # * <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 已提交
837
      # The declaration can also include an options hash to specialize the behavior of the association.
838
      #
D
Initial  
David Heinemeier Hansson 已提交
839
      # Options are:
P
Pratik Naik 已提交
840 841 842 843
      # * <tt>:class_name</tt> - Specify the class name of the association. Use it only if that name can't be inferred
      #   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.
      # * <tt>:conditions</tt> - Specify the conditions that the associated object must meet in order to be included as a +WHERE+
844
      #   SQL fragment, such as <tt>authorized = 1</tt>.
845 846
      # * <tt>:select</tt> - By default, this is <tt>*</tt> as in <tt>SELECT * FROM</tt>, but can be changed if, for example, you want to do a join
      #   but not include the joined columns. Do not forget to include the primary and foreign keys, otherwise it will raise an error.
P
Pratik Naik 已提交
847 848 849 850 851
      # * <tt>:foreign_key</tt> - Specify the foreign key used for the association. By default this is guessed to be the name
      #   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".
      # * <tt>:dependent</tt> - If set to <tt>:destroy</tt>, the associated object is destroyed when this object is. If set to
852 853 854
      #   <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 已提交
855
      # * <tt>:counter_cache</tt> - Caches the number of belonging objects on the associate class through the use of +increment_counter+
856
      #   and +decrement_counter+. The counter cache is incremented when an object of this class is created and decremented when it's
P
Pratik Naik 已提交
857 858
      #   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
859
      #   a column name instead of a +true+/+false+ value to this option (e.g., <tt>:counter_cache => :my_custom_counter</tt>.)
860
      #   When creating a counter cache column, the database statement or migration must specify a default value of <tt>0</tt>, failing to do 
P
Pratik Naik 已提交
861 862 863 864
      #   this results in a counter with +NULL+ value, which will never increment.
      #   Note: Specifying a counter cache will add it to that model's list of readonly attributes using +attr_readonly+.
      # * <tt>:include</tt> - Specify second-order associations that should be eager loaded when this object is loaded.
      # * <tt>:polymorphic</tt> - Specify this association is a polymorphic association by passing +true+.
865
      #   Note: If you've enabled the counter cache, then you may want to add the counter cache attribute
P
Pratik Naik 已提交
866 867
      #   to the +attr_readonly+ list in the associated classes (e.g. <tt>class Post; attr_readonly :comments_count; end</tt>).
      # * <tt>:readonly</tt> - If true, the associated object is readonly through the association.
P
Pratik Naik 已提交
868
      # * <tt>:validate</tt> - If false, don't validate the associated objects when saving the parent object. +false+ by default.
869
      # * <tt>:accessible</tt> - Mass assignment is allowed for this assocation (similar to <tt>ActiveRecord::Base#attr_accessible</tt>).
D
Initial  
David Heinemeier Hansson 已提交
870 871 872 873
      #
      # Option examples:
      #   belongs_to :firm, :foreign_key => "client_of"
      #   belongs_to :author, :class_name => "Person", :foreign_key => "author_id"
874
      #   belongs_to :valid_coupon, :class_name => "Coupon", :foreign_key => "coupon_id",
D
Initial  
David Heinemeier Hansson 已提交
875
      #              :conditions => 'discounts > #{payments_count}'
R
Rick Olson 已提交
876
      #   belongs_to :attachable, :polymorphic => true
877
      #   belongs_to :project, :readonly => true
878
      #   belongs_to :post, :counter_cache => true
879
      def belongs_to(association_id, options = {})
880
        reflection = create_belongs_to_reflection(association_id, options)
881 882 883

        ivar = "@#{reflection.name}"

884 885
        if reflection.options[:polymorphic]
          association_accessor_methods(reflection, BelongsToPolymorphicAssociation)
886

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

891 892 893 894
            if association && association.target
              if association.new_record?
                association.save(true)
              end
895

896 897 898
              if association.updated?
                self["#{reflection.primary_key_name}"] = association.id
                self["#{reflection.options[:foreign_type]}"] = association.class.base_class.name.to_s
899
              end
900
            end
901
          end
902
          before_save method_name
903
        else
904 905 906
          association_accessor_methods(reflection, BelongsToAssociation)
          association_constructor_method(:build,  reflection, BelongsToAssociation)
          association_constructor_method(:create, reflection, BelongsToAssociation)
907

908 909 910
          method_name = "belongs_to_before_save_for_#{reflection.name}".to_sym
          define_method(method_name) do
            association = instance_variable_get("#{ivar}") if instance_variable_defined?("#{ivar}")
911

912 913 914 915
            if !association.nil?
              if association.new_record?
                association.save(true)
              end
916

917 918
              if association.updated?
                self["#{reflection.primary_key_name}"] = association.id
919
              end
920
            end
921
          end
922
          before_save method_name
923
        end
924

925
        # Create the callbacks to update counter cache
926
        if options[:counter_cache]
927 928 929 930
          cache_column = options[:counter_cache] == true ?
            "#{self.to_s.underscore.pluralize}_count" :
            options[:counter_cache]

931 932 933 934 935 936
          method_name = "belongs_to_counter_cache_after_create_for_#{reflection.name}".to_sym
          define_method(method_name) do
            association = send("#{reflection.name}")
            association.class.increment_counter("#{cache_column}", send("#{reflection.primary_key_name}")) unless association.nil?
          end
          after_create method_name
937

938 939 940 941 942 943
          method_name = "belongs_to_counter_cache_before_destroy_for_#{reflection.name}".to_sym
          define_method(method_name) do
            association = send("#{reflection.name}")
            association.class.decrement_counter("#{cache_column}", send("#{reflection.primary_key_name}")) unless association.nil?
          end
          before_destroy method_name
944

945
          module_eval(
946
            "#{reflection.class_name}.send(:attr_readonly,\"#{cache_column}\".intern) if defined?(#{reflection.class_name}) && #{reflection.class_name}.respond_to?(:attr_readonly)"
947
          )
948
        end
949

950
        add_single_associated_validation_callbacks(reflection.name) if options[:validate] == true
951

952
        configure_dependency_for_belongs_to(reflection)
953 954
      end

D
Initial  
David Heinemeier Hansson 已提交
955
      # Associates two classes via an intermediate join table.  Unless the join table is explicitly specified as
P
Pratik Naik 已提交
956 957 958
      # an option, it is guessed using the lexical order of the class names. So a join between Developer and Project
      # 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,
959
      # and the strings are equal when compared up to the shortest length, then the longer string is considered of higher
P
Pratik Naik 已提交
960 961 962 963
      # 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 已提交
964
      #
965
      # Deprecated: Any additional fields added to the join table will be placed as attributes when pulling records out through
966
      # +has_and_belongs_to_many+ associations. Records returned from join tables with additional attributes will be marked as
P
Pratik Naik 已提交
967
      # readonly (because we can't save changes to the additional attributes). It's strongly recommended that you upgrade any
968
      # associations with attributes to a real join model (see introduction).
D
Initial  
David Heinemeier Hansson 已提交
969
      #
970
      # Adds the following methods for retrieval and query:
971
      # +collection+ is replaced with the symbol passed as the first argument, so
972
      # <tt>has_and_belongs_to_many :categories</tt> would add among others <tt>categories.empty?</tt>.
P
Pratik Naik 已提交
973
      # * <tt>collection(force_reload = false)</tt> - Returns an array of all the associated objects.
974
      #   An empty array is returned if none are found.
P
Pratik Naik 已提交
975
      # * <tt>collection<<(object, ...)</tt> - Adds one or more objects to the collection by creating associations in the join table
976
      #   (<tt>collection.push</tt> and <tt>collection.concat</tt> are aliases to this method).
P
Pratik Naik 已提交
977
      # * <tt>collection.delete(object, ...)</tt> - Removes one or more objects from the collection by removing their associations from the join table.
D
Initial  
David Heinemeier Hansson 已提交
978
      #   This does not destroy the objects.
P
Pratik Naik 已提交
979 980 981 982 983 984 985
      # * <tt>collection=objects</tt> - Replaces the collection's content by deleting and adding objects as appropriate.
      # * <tt>collection_singular_ids</tt> - Returns an array of the associated objects' ids.
      # * <tt>collection_singular_ids=ids</tt> - Replace the collection by the objects identified by the primary keys in +ids+.
      # * <tt>collection.clear</tt> - Removes every object from the collection. This does not destroy the objects.
      # * <tt>collection.empty?</tt> - Returns +true+ if there are no associated objects.
      # * <tt>collection.size</tt> - Returns the number of associated objects.
      # * <tt>collection.find(id)</tt> - Finds an associated object responding to the +id+ and that
986
      #   meets the condition that it has to be associated with this object.
P
Pratik Naik 已提交
987
      # * <tt>collection.build(attributes = {})</tt> - Returns a new object of the collection type that has been instantiated
988
      #   with +attributes+ and linked to this object through the join table, but has not yet been saved.
P
Pratik Naik 已提交
989
      # * <tt>collection.create(attributes = {})</tt> - Returns a new object of the collection type that has been instantiated
990
      #   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 已提交
991
      #
992
      # Example: A Developer class declares <tt>has_and_belongs_to_many :projects</tt>, which will add:
D
Initial  
David Heinemeier Hansson 已提交
993 994 995
      # * <tt>Developer#projects</tt>
      # * <tt>Developer#projects<<</tt>
      # * <tt>Developer#projects.delete</tt>
996
      # * <tt>Developer#projects=</tt>
997
      # * <tt>Developer#project_ids</tt>
998
      # * <tt>Developer#project_ids=</tt>
D
Initial  
David Heinemeier Hansson 已提交
999 1000 1001 1002
      # * <tt>Developer#projects.clear</tt>
      # * <tt>Developer#projects.empty?</tt>
      # * <tt>Developer#projects.size</tt>
      # * <tt>Developer#projects.find(id)</tt>
1003 1004
      # * <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 已提交
1005
      # The declaration may include an options hash to specialize the behavior of the association.
1006
      #
D
Initial  
David Heinemeier Hansson 已提交
1007
      # Options are:
P
Pratik Naik 已提交
1008
      # * <tt>:class_name</tt> - Specify the class name of the association. Use it only if that name can't be inferred
1009
      #   from the association name. So <tt>has_and_belongs_to_many :projects</tt> will by default be linked to the
P
Pratik Naik 已提交
1010 1011
      #   Project class, but if the real class name is SuperProject, you'll have to specify it with this option.
      # * <tt>:join_table</tt> - Specify the name of the join table if the default based on lexical order isn't what you want.
1012 1013
      #   WARNING: 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.
P
Pratik Naik 已提交
1014 1015 1016 1017 1018 1019 1020
      # * <tt>:foreign_key</tt> - Specify the foreign key used for the association. By default this is guessed to be the name
      #   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>.
      # * <tt>:association_foreign_key</tt> - Specify the association foreign key used for the association. By default this is
      #   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>.
      # * <tt>:conditions</tt> - Specify the conditions that the associated object must meet in order to be included as a +WHERE+
1021 1022 1023
      #   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 已提交
1024
      # * <tt>:order</tt> - Specify the order in which the associated objects are returned as an <tt>ORDER BY</tt> SQL fragment,
1025
      #   such as <tt>last_name, first_name DESC</tt>
P
Pratik Naik 已提交
1026 1027 1028 1029 1030 1031 1032 1033 1034 1035 1036 1037
      # * <tt>:uniq</tt> - If true, duplicate associated objects will be ignored by accessors and query methods.
      # * <tt>:finder_sql</tt> - Overwrite the default generated SQL statement used to fetch the association with a manual statement
      # * <tt>:delete_sql</tt> - Overwrite the default generated SQL statement used to remove links between the associated
      #   classes with a manual statement.
      # * <tt>:insert_sql</tt> - Overwrite the default generated SQL statement used to add links between the associated classes
      #   with a manual statement.
      # * <tt>:extend</tt> - Anonymous module for extending the proxy, see "Association extensions".
      # * <tt>:include</tt> - Specify second-order associations that should be eager loaded when the collection is loaded.
      # * <tt>:group</tt> - An attribute name by which the result should be grouped. Uses the <tt>GROUP BY</tt> SQL-clause.
      # * <tt>:limit</tt> - An integer determining the limit on the number of rows that should be returned.
      # * <tt>:offset</tt> - An integer determining the offset from where the rows should be fetched. So at 5, it would skip the first 4 rows.
      # * <tt>:select</tt> - 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
1038
      #   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 已提交
1039
      # * <tt>:readonly</tt> - If true, all the associated objects are readonly through the association.
1040
      # * <tt>:validate</tt> - If false, don't validate the associated objects when saving the parent object. +true+ by default.
1041
      # * <tt>:accessible</tt> - Mass assignment is allowed for this assocation (similar to <tt>ActiveRecord::Base#attr_accessible</tt>).
D
Initial  
David Heinemeier Hansson 已提交
1042 1043 1044
      #
      # Option examples:
      #   has_and_belongs_to_many :projects
1045
      #   has_and_belongs_to_many :projects, :include => [ :milestones, :manager ]
D
Initial  
David Heinemeier Hansson 已提交
1046 1047
      #   has_and_belongs_to_many :nations, :class_name => "Country"
      #   has_and_belongs_to_many :categories, :join_table => "prods_cats"
1048
      #   has_and_belongs_to_many :categories, :readonly => true
1049
      #   has_and_belongs_to_many :active_projects, :join_table => 'developers_projects', :delete_sql =>
1050
      #   'DELETE FROM developers_projects WHERE active=1 AND developer_id = #{id} AND project_id = #{record.id}'
1051
      def has_and_belongs_to_many(association_id, options = {}, &extension)
1052
        reflection = create_has_and_belongs_to_many_reflection(association_id, options, &extension)
1053

1054 1055
        add_multiple_associated_validation_callbacks(reflection.name) unless options[:validate] == false
        add_multiple_associated_save_callbacks(reflection.name)
1056
        collection_accessor_methods(reflection, HasAndBelongsToManyAssociation)
D
Initial  
David Heinemeier Hansson 已提交
1057

1058 1059
        # Don't use a before_destroy callback since users' before_destroy
        # callbacks will be executed after the association is wiped out.
1060
        old_method = "destroy_without_habtm_shim_for_#{reflection.name}"
1061
        class_eval <<-end_eval unless method_defined?(old_method)
1062 1063
          alias_method :#{old_method}, :destroy_without_callbacks
          def destroy_without_callbacks
1064
            #{reflection.name}.clear
1065 1066 1067 1068
            #{old_method}
          end
        end_eval

1069
        add_association_callbacks(reflection.name, options)
D
Initial  
David Heinemeier Hansson 已提交
1070 1071 1072
      end

      private
1073 1074 1075 1076 1077 1078
        # Generate a join table name from two provided tables names.
        # The order of names in join name is determined by lexical precedence.
        #   join_table_name("members", "clubs")
        #   => "clubs_members"
        #   join_table_name("members", "special_clubs")
        #   => "members_special_clubs"
D
Initial  
David Heinemeier Hansson 已提交
1079 1080 1081 1082 1083 1084 1085 1086 1087
        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
1088

1089
        def association_accessor_methods(reflection, association_proxy_class)
1090 1091
          ivar = "@#{reflection.name}"

1092
          define_method(reflection.name) do |*params|
1093
            force_reload = params.first unless params.empty?
1094 1095

            association = instance_variable_get(ivar) if instance_variable_defined?(ivar)
1096 1097 1098

            if association.nil? || force_reload
              association = association_proxy_class.new(self, reflection)
1099
              retval = association.reload
1100
              if retval.nil? and association_proxy_class == BelongsToAssociation
1101
                instance_variable_set(ivar, nil)
1102 1103
                return nil
              end
1104
              instance_variable_set(ivar, association)
1105
            end
1106 1107

            association.target.nil? ? nil : association
1108 1109
          end

1110
          define_method("#{reflection.name}=") do |new_value|
1111 1112
            association = instance_variable_get(ivar) if instance_variable_defined?(ivar)

1113
            if association.nil? || association.target != new_value
1114
              association = association_proxy_class.new(self, reflection)
1115
            end
1116

1117 1118
            new_value = reflection.klass.new(new_value) if reflection.options[:accessible] && new_value.is_a?(Hash)

1119 1120 1121 1122 1123 1124
            if association_proxy_class == HasOneThroughAssociation
              association.create_through_record(new_value)
              self.send(reflection.name, new_value)
            else
              association.replace(new_value)              
            end
1125

1126
            instance_variable_set(ivar, new_value.nil? ? nil : association)
1127
          end
1128

1129
          define_method("set_#{reflection.name}_target") do |target|
1130
            return if target.nil? and association_proxy_class == BelongsToAssociation
1131
            association = association_proxy_class.new(self, reflection)
1132
            association.target = target
1133
            instance_variable_set(ivar, association)
1134
          end
1135 1136
        end

1137 1138
        def collection_reader_method(reflection, association_proxy_class)
          define_method(reflection.name) do |*params|
1139 1140
            ivar = "@#{reflection.name}"

1141
            force_reload = params.first unless params.empty?
1142
            association = instance_variable_get(ivar) if instance_variable_defined?(ivar)
1143

1144
            unless association.respond_to?(:loaded?)
1145
              association = association_proxy_class.new(self, reflection)
1146
              instance_variable_set(ivar, association)
D
Initial  
David Heinemeier Hansson 已提交
1147
            end
1148

1149
            association.reload if force_reload
1150

1151 1152
            association
          end
1153 1154

          define_method("#{reflection.name.to_s.singularize}_ids") do
1155
            send(reflection.name).map { |record| record.id }
1156
          end
1157
        end
D
Initial  
David Heinemeier Hansson 已提交
1158

1159
        def collection_accessor_methods(reflection, association_proxy_class, writer = true)
1160 1161
          collection_reader_method(reflection, association_proxy_class)

1162 1163 1164 1165 1166 1167 1168
          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
1169

1170 1171 1172 1173
            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
1174
          end
D
Initial  
David Heinemeier Hansson 已提交
1175
        end
1176
        
1177
        def add_single_associated_validation_callbacks(association_name)
1178 1179 1180 1181 1182 1183 1184 1185 1186 1187 1188
          method_name = "validate_associated_records_for_#{association_name}".to_sym
          define_method(method_name) do
            association = instance_variable_get("@#{association_name}")
            if !association.nil?
              errors.add "#{association_name}" unless association.target.nil? || association.valid?
            end
          end
        
          validate method_name
        end
        
1189
        def add_multiple_associated_validation_callbacks(association_name)
1190
          method_name = "validate_associated_records_for_#{association_name}".to_sym
1191 1192
          ivar = "@#{association_name}"

1193
          define_method(method_name) do
1194 1195
            association = instance_variable_get(ivar) if instance_variable_defined?(ivar)

1196
            if association.respond_to?(:loaded?)
1197
              if new_record?
1198
                association
1199
              elsif association.loaded?
1200
                association.select { |record| record.new_record? }
1201 1202
              else
                association.target.select { |record| record.new_record? }
1203 1204
              end.each do |record|
                errors.add "#{association_name}" unless record.valid?
D
Initial  
David Heinemeier Hansson 已提交
1205
              end
1206
            end
1207
          end
D
Initial  
David Heinemeier Hansson 已提交
1208

1209
          validate method_name
1210 1211 1212 1213
        end

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

1215 1216 1217 1218 1219 1220 1221 1222 1223
          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
1224
            association = instance_variable_get("#{ivar}") if instance_variable_defined?("#{ivar}")
1225

1226 1227 1228 1229
            records_to_save = if @new_record_before_save
              association
            elsif association.respond_to?(:loaded?) && association.loaded?
              association.select { |record| record.new_record? }
1230 1231
            elsif association.respond_to?(:loaded?) && !association.loaded?
              association.target.select { |record| record.new_record? }
1232 1233 1234
            else
              []
            end
1235
            records_to_save.each { |record| association.send(:insert_record, record) } unless records_to_save.blank?
1236

1237 1238
            # reconstruct the SQL queries now that we know the owner's id
            association.send(:construct_sql) if association.respond_to?(:construct_sql)
1239
          end
1240

1241
          # Doesn't use after_save as that would save associations added in after_create/after_update twice
1242 1243
          after_create method_name
          after_update method_name
D
Initial  
David Heinemeier Hansson 已提交
1244
        end
1245

1246 1247
        def association_constructor_method(constructor, reflection, association_proxy_class)
          define_method("#{constructor}_#{reflection.name}") do |*params|
1248 1249
            ivar = "@#{reflection.name}"

1250 1251
            attributees      = params.first unless params.empty?
            replace_existing = params[1].nil? ? true : params[1]
1252
            association      = instance_variable_get(ivar) if instance_variable_defined?(ivar)
1253 1254

            if association.nil?
1255
              association = association_proxy_class.new(self, reflection)
1256
              instance_variable_set(ivar, association)
1257 1258
            end

1259 1260 1261 1262 1263
            if association_proxy_class == HasOneAssociation
              association.send(constructor, attributees, replace_existing)
            else
              association.send(constructor, attributees)
            end
1264 1265
          end
        end
1266

1267
        def find_with_associations(options = {})
1268
          catch :invalid_query do
1269
            join_dependency = JoinDependency.new(self, merge_includes(scope(:find, :include), options[:include]), options[:joins])
1270 1271 1272 1273
            rows = select_all_rows(options, join_dependency)
            return join_dependency.instantiate(rows)
          end
          []
1274
        end
1275

1276 1277
        # See HasManyAssociation#delete_records.  Dependent associations
        # delete children, otherwise foreign key is set to NULL.
1278
        def configure_dependency_for_has_many(reflection)
1279 1280 1281 1282 1283 1284 1285
          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]
            dependent_conditions = dependent_conditions.collect {|where| "(#{where})" }.join(" AND ")
1286

1287 1288
            case reflection.options[:dependent]
              when :destroy
1289 1290 1291 1292 1293
                method_name = "has_many_dependent_destroy_for_#{reflection.name}".to_sym
                define_method(method_name) do
                  send("#{reflection.name}").each { |o| o.destroy }
                end
                before_destroy method_name
1294 1295 1296 1297 1298 1299 1300
              when :delete_all
                module_eval "before_destroy { |record| #{reflection.class_name}.delete_all(%(#{dependent_conditions})) }"
              when :nullify
                module_eval "before_destroy { |record| #{reflection.class_name}.update_all(%(#{reflection.primary_key_name} = NULL),  %(#{dependent_conditions})) }"
              else
                raise ArgumentError, "The :dependent option expects either :destroy, :delete_all, or :nullify (#{reflection.options[:dependent].inspect})"
            end
1301 1302
          end
        end
1303

1304
        def configure_dependency_for_has_one(reflection)
1305 1306 1307
          if reflection.options.include?(:dependent)
            case reflection.options[:dependent]
              when :destroy
1308 1309 1310 1311 1312 1313
                method_name = "has_one_dependent_destroy_for_#{reflection.name}".to_sym
                define_method(method_name) do
                  association = send("#{reflection.name}")
                  association.destroy unless association.nil?
                end
                before_destroy method_name
1314
              when :delete
1315 1316 1317 1318 1319 1320
                method_name = "has_one_dependent_delete_for_#{reflection.name}".to_sym
                define_method(method_name) do
                  association = send("#{reflection.name}")
                  association.class.delete(association.id) unless association.nil?
                end
                before_destroy method_name
1321
              when :nullify
1322 1323 1324 1325 1326 1327
                method_name = "has_one_dependent_nullify_for_#{reflection.name}".to_sym
                define_method(method_name) do
                  association = send("#{reflection.name}")
                  association.update_attribute("#{reflection.primary_key_name}", nil) unless association.nil?
                end
                before_destroy method_name
1328 1329 1330
              else
                raise ArgumentError, "The :dependent option expects either :destroy, :delete or :nullify (#{reflection.options[:dependent].inspect})"
            end
1331 1332 1333
          end
        end

1334 1335 1336 1337
        def configure_dependency_for_belongs_to(reflection)
          if reflection.options.include?(:dependent)
            case reflection.options[:dependent]
              when :destroy
1338 1339 1340 1341 1342 1343
                method_name = "belongs_to_dependent_destroy_for_#{reflection.name}".to_sym
                define_method(method_name) do
                  association = send("#{reflection.name}")
                  association.destroy unless association.nil?
                end
                before_destroy method_name
1344
              when :delete
1345 1346 1347 1348 1349 1350
                method_name = "belongs_to_dependent_delete_for_#{reflection.name}".to_sym
                define_method(method_name) do
                  association = send("#{reflection.name}")
                  association.class.delete(association.id) unless association.nil?
                end
                before_destroy method_name
1351 1352 1353 1354 1355 1356
              else
                raise ArgumentError, "The :dependent option expects either :destroy or :delete (#{reflection.options[:dependent].inspect})"
            end
          end
        end

1357 1358
        def create_has_many_reflection(association_id, options, &extension)
          options.assert_valid_keys(
1359
            :class_name, :table_name, :foreign_key, :primary_key,
1360
            :dependent,
1361
            :select, :conditions, :include, :order, :group, :limit, :offset,
1362
            :as, :through, :source, :source_type,
1363
            :uniq,
1364 1365
            :finder_sql, :counter_sql,
            :before_add, :after_add, :before_remove, :after_remove,
1366
            :extend, :readonly,
1367
            :validate, :accessible
1368 1369
          )

1370
          options[:extend] = create_extension_modules(association_id, extension, options[:extend])
1371

1372
          create_reflection(:has_many, association_id, options, self)
1373 1374 1375 1376
        end

        def create_has_one_reflection(association_id, options)
          options.assert_valid_keys(
1377
            :class_name, :foreign_key, :remote, :select, :conditions, :order, :include, :dependent, :counter_cache, :extend, :as, :readonly, :validate, :primary_key, :accessible
1378 1379
          )

1380
          create_reflection(:has_one, association_id, options, self)
1381
        end
1382 1383 1384
        
        def create_has_one_through_reflection(association_id, options)
          options.assert_valid_keys(
1385
            :class_name, :foreign_key, :remote, :select, :conditions, :order, :include, :dependent, :counter_cache, :extend, :as, :through, :source, :source_type, :validate
1386 1387 1388
          )
          create_reflection(:has_one, association_id, options, self)
        end
1389 1390 1391

        def create_belongs_to_reflection(association_id, options)
          options.assert_valid_keys(
1392
            :class_name, :foreign_key, :foreign_type, :remote, :select, :conditions, :include, :dependent,
1393
            :counter_cache, :extend, :polymorphic, :readonly, :validate, :accessible
1394
          )
1395

1396 1397 1398 1399 1400 1401 1402 1403
          reflection = create_reflection(:belongs_to, association_id, options, self)

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

          reflection
        end
1404

1405 1406
        def create_has_and_belongs_to_many_reflection(association_id, options, &extension)
          options.assert_valid_keys(
1407
            :class_name, :table_name, :join_table, :foreign_key, :association_foreign_key,
1408
            :select, :conditions, :include, :order, :group, :limit, :offset,
1409
            :uniq,
1410
            :finder_sql, :delete_sql, :insert_sql,
1411
            :before_add, :after_add, :before_remove, :after_remove,
1412
            :extend, :readonly,
1413
            :validate, :accessible
1414 1415
          )

1416
          options[:extend] = create_extension_modules(association_id, extension, options[:extend])
1417 1418 1419 1420

          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))
1421

1422 1423 1424
          reflection
        end

1425
        def reflect_on_included_associations(associations)
1426
          [ associations ].flatten.collect { |association| reflect_on_association(association.to_s.intern) }
1427 1428
        end

1429 1430 1431
        def guard_against_unlimitable_reflections(reflections, options)
          if (options[:offset] || options[:limit]) && !using_limitable_reflections?(reflections)
            raise(
1432
              ConfigurationError,
1433 1434 1435 1436 1437
              "You can not use offset and limit together with has_many or has_and_belongs_to_many associations"
            )
          end
        end

1438
        def select_all_rows(options, join_dependency)
1439
          connection.select_all(
1440
            construct_finder_sql_with_included_associations(options, join_dependency),
1441 1442 1443
            "#{name} Load Including Associations"
          )
        end
1444

1445
        def construct_finder_sql_with_included_associations(options, join_dependency)
1446
          scope = scope(:find)
1447
          sql = "SELECT #{column_aliases(join_dependency)} FROM #{(scope && scope[:from]) || options[:from] || quoted_table_name} "
1448
          sql << join_dependency.join_associations.collect{|join| join.association_join }.join
1449

1450 1451
          add_joins!(sql, options, scope)
          add_conditions!(sql, options[:conditions], scope)
1452
          add_limited_ids_condition!(sql, options, join_dependency) if !using_limitable_reflections?(join_dependency.reflections) && ((scope && scope[:limit]) || options[:limit])
1453

1454
          add_group!(sql, options[:group], scope)
1455
          add_order!(sql, options[:order], scope)
1456
          add_limit!(sql, options, scope) if using_limitable_reflections?(join_dependency.reflections)
1457
          add_lock!(sql, options, scope)
1458

1459 1460
          return sanitize_sql(sql)
        end
1461

1462 1463
        def add_limited_ids_condition!(sql, options, join_dependency)
          unless (id_list = select_limited_ids_list(options, join_dependency)).empty?
1464
            sql << "#{condition_word(sql)} #{connection.quote_table_name table_name}.#{primary_key} IN (#{id_list}) "
1465 1466
          else
            throw :invalid_query
1467 1468
          end
        end
1469

1470
        def select_limited_ids_list(options, join_dependency)
1471 1472
          pk = columns_hash[primary_key]

1473
          connection.select_all(
1474
            construct_finder_sql_for_association_limiting(options, join_dependency),
1475
            "#{name} Load IDs For Limited Eager Loading"
1476
          ).collect { |row| connection.quote(row[primary_key], pk) }.join(", ")
1477
        end
1478

1479
        def construct_finder_sql_for_association_limiting(options, join_dependency)
1480
          scope       = scope(:find)
1481 1482 1483 1484 1485

          # 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
1486 1487 1488 1489
          distinct_join_associations = all_tables.uniq.map{|table|
            join_dependency.joins_for_table_name(table)
          }.flatten.compact.uniq

1490 1491 1492 1493 1494
          order = options[:order]
          if scoped_order = (scope && scope[:order])
            order = order ? "#{order}, #{scoped_order}" : scoped_order
          end

1495
          is_distinct = !options[:joins].blank? || include_eager_conditions?(options, tables_from_conditions) || include_eager_order?(options, tables_from_order)
1496
          sql = "SELECT "
1497
          if is_distinct
1498
            sql << connection.distinct("#{connection.quote_table_name table_name}.#{primary_key}", order)
1499 1500 1501
          else
            sql << primary_key
          end
1502
          sql << " FROM #{connection.quote_table_name table_name} "
1503

1504
          if is_distinct
1505
            sql << distinct_join_associations.collect { |assoc| assoc.association_join }.join
1506
            add_joins!(sql, options, scope)
1507
          end
1508

1509
          add_conditions!(sql, options[:conditions], scope)
1510
          add_group!(sql, options[:group], scope)
1511

1512 1513
          if order && is_distinct
            connection.add_order_by_for_association_limiting!(sql, :order => order)
1514 1515
          else
            add_order!(sql, options[:order], scope)
1516
          end
1517

1518
          add_limit!(sql, options, scope)
1519

1520 1521
          return sanitize_sql(sql)
        end
1522

1523
        def conditions_tables(options)
1524 1525 1526 1527 1528 1529 1530 1531
          # 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
              else            all << cond
            end
          end
1532 1533 1534 1535
          conditions.join(' ').scan(/([\.\w]+).?\./).flatten
        end

        def order_tables(options)
1536
          order = [options[:order], scope(:find, :order) ].join(", ")
1537 1538 1539 1540
          return [] unless order && order.is_a?(String)
          order.scan(/([\.\w]+).?\./).flatten
        end

1541 1542 1543 1544 1545 1546
        def selects_tables(options)
          select = options[:select]
          return [] unless select && select.is_a?(String)
          select.scan(/"?([\.\w]+)"?.?\./).flatten
        end

1547
        # Checks if the conditions reference a table other than the current model table
1548 1549
        def include_eager_conditions?(options, tables = nil)
          ((tables || conditions_tables(options)) - [table_name]).any?
1550
        end
1551

1552
        # Checks if the query order references a table other than the current model's table.
1553 1554
        def include_eager_order?(options, tables = nil)
          ((tables || order_tables(options)) - [table_name]).any?
1555
        end
1556

1557
        def include_eager_select?(options)
1558
          (selects_tables(options) - [table_name]).any?
1559 1560
        end

1561
        def references_eager_loaded_tables?(options)
1562
          include_eager_order?(options) || include_eager_conditions?(options) || include_eager_select?(options)
1563 1564
        end

1565 1566 1567 1568
        def using_limitable_reflections?(reflections)
          reflections.reject { |r| [ :belongs_to, :has_one ].include?(r.macro) }.length.zero?
        end

1569 1570
        def column_aliases(join_dependency)
          join_dependency.joins.collect{|join| join.column_names_with_alias.collect{|column_name, aliased_name|
1571
              "#{connection.quote_table_name join.aliased_table_name}.#{connection.quote_column_name column_name} AS #{aliased_name}"}}.flatten.join(", ")
1572 1573
        end

1574
        def add_association_callbacks(association_name, options)
1575 1576
          callbacks = %w(before_add after_add before_remove after_remove)
          callbacks.each do |callback_name|
1577
            full_callback_name = "#{callback_name}_for_#{association_name}"
1578 1579 1580
            defined_callbacks = options[callback_name.to_sym]
            if options.has_key?(callback_name.to_sym)
              class_inheritable_reader full_callback_name.to_sym
1581 1582 1583
              write_inheritable_attribute(full_callback_name.to_sym, [defined_callbacks].flatten)
            else
              write_inheritable_attribute(full_callback_name.to_sym, [])
1584 1585
            end
          end
1586
        end
1587

1588 1589 1590
        def condition_word(sql)
          sql =~ /where/i ? " AND " : "WHERE "
        end
1591

1592
        def create_extension_modules(association_id, block_extension, extensions)
1593 1594
          if block_extension
            extension_module_name = "#{self.to_s}#{association_id.to_s.camelize}AssociationExtension"
1595

1596 1597 1598 1599 1600 1601
            silence_warnings do
              Object.const_set(extension_module_name, Module.new(&block_extension))
            end
            Array(extensions).push(extension_module_name.constantize)
          else
            Array(extensions)
1602 1603
          end
        end
1604

D
David Heinemeier Hansson 已提交
1605
        class JoinDependency # :nodoc:
1606
          attr_reader :joins, :reflections, :table_aliases
1607

1608
          def initialize(base, associations, joins)
1609
            @joins                 = [JoinBase.new(base, joins)]
1610 1611 1612 1613
            @associations          = associations
            @reflections           = []
            @base_records_hash     = {}
            @base_records_in_order = []
1614 1615
            @table_aliases         = Hash.new { |aliases, table| aliases[table] = 0 }
            @table_aliases[base.table_name] = 1
1616 1617 1618 1619 1620 1621 1622 1623 1624 1625 1626 1627 1628 1629 1630 1631 1632
            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
1633
              construct(@base_records_hash[primary_id], @associations, join_associations.dup, row)
1634
            end
1635
            remove_duplicate_results!(join_base.active_record, @base_records_in_order, @associations)
1636 1637 1638
            return @base_records_in_order
          end

1639 1640 1641 1642 1643 1644 1645 1646 1647 1648 1649 1650 1651 1652 1653 1654 1655
          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|
1656 1657 1658 1659
                    descendant = record.send(reflection.name)
                    next unless descendant
                    descendant.target.uniq! if is_collection
                    descendant
1660 1661 1662 1663 1664 1665 1666
                  end.flatten.compact

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

1667
          def join_for_table_name(table_name)
1668 1669 1670
            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
1671 1672 1673 1674 1675 1676 1677 1678 1679 1680 1681 1682 1683 1684 1685
          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

1686 1687 1688 1689 1690 1691 1692 1693
          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
1694
                  @joins << build_join_association(reflection, parent)
1695 1696 1697 1698 1699 1700 1701 1702 1703 1704 1705 1706 1707 1708
                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

1709 1710 1711 1712 1713
            # overridden in InnerJoinDependency subclass
            def build_join_association(reflection, parent)
              JoinAssociation.new(reflection, self, parent)
            end

1714 1715 1716 1717 1718 1719 1720 1721 1722 1723 1724 1725 1726 1727 1728 1729 1730 1731 1732 1733 1734 1735 1736 1737 1738 1739
            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
1740

1741 1742
                  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)
1743
                  collection.target.push(association)
1744 1745 1746 1747 1748
                when :has_one
                  return if record.id.to_s != join.parent.record_id(row).to_s
                  association = join.instantiate(row) unless row[join.aliased_primary_key].nil?
                  record.send("set_#{join.reflection.name}_target", association)
                when :belongs_to
1749 1750 1751 1752 1753 1754 1755 1756 1757
                  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 已提交
1758
          class JoinBase # :nodoc:
1759
            attr_reader :active_record, :table_joins
1760
            delegate    :table_name, :column_names, :primary_key, :reflections, :sanitize_sql, :to => :active_record
1761

1762
            def initialize(active_record, joins = nil)
1763 1764
              @active_record = active_record
              @cached_record = {}
1765
              @table_joins   = joins
1766 1767 1768 1769 1770 1771 1772 1773 1774 1775 1776 1777 1778 1779 1780
            end

            def aliased_prefix
              "t0"
            end

            def aliased_primary_key
              "#{ aliased_prefix }_r0"
            end

            def aliased_table_name
              active_record.table_name
            end

            def column_names_with_alias
1781
              unless defined?(@column_names_with_alias)
1782
                @column_names_with_alias = []
1783

1784 1785 1786 1787
                ([primary_key] + (column_names - [primary_key])).each_with_index do |column_name, i|
                  @column_names_with_alias << [column_name, "#{ aliased_prefix }_r#{ i }"]
                end
              end
1788 1789

              @column_names_with_alias
1790 1791 1792 1793 1794 1795 1796 1797 1798 1799 1800
            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)
1801
              @cached_record[record_id(row)] ||= active_record.send(:instantiate, extract_record(row))
1802 1803 1804
            end
          end

D
David Heinemeier Hansson 已提交
1805
          class JoinAssociation < JoinBase # :nodoc:
1806
            attr_reader :reflection, :parent, :aliased_table_name, :aliased_prefix, :aliased_join_table_name, :parent_table_name
1807
            delegate    :options, :klass, :through_reflection, :source_reflection, :to => :reflection
1808 1809

            def initialize(reflection, join_dependency, parent = nil)
1810 1811 1812 1813 1814
              reflection.check_validity!
              if reflection.options[:polymorphic]
                raise EagerLoadPolymorphicError.new(reflection)
              end

1815
              super(reflection.klass)
1816
              @join_dependency    = join_dependency
1817 1818 1819
              @parent             = parent
              @reflection         = reflection
              @aliased_prefix     = "t#{ join_dependency.joins.size }"
1820
              @parent_table_name  = parent.active_record.table_name
1821 1822 1823 1824
              @aliased_table_name = aliased_table_name_for(table_name)
              
              if reflection.macro == :has_and_belongs_to_many
                @aliased_join_table_name = aliased_table_name_for(reflection.options[:join_table], "_join")
1825
              end
1826 1827 1828
        
              if reflection.macro == :has_many && reflection.options[:through]
                @aliased_join_table_name = aliased_table_name_for(reflection.through_reflection.klass.table_name, "_join")
1829
              end
1830 1831 1832
            end

            def association_join
1833
              connection = reflection.active_record.connection
1834
              join = case reflection.macro
1835
                when :has_and_belongs_to_many
1836
                  " #{join_type} %s ON %s.%s = %s.%s " % [
1837
                     table_alias_for(options[:join_table], aliased_join_table_name),
1838
                     connection.quote_table_name(aliased_join_table_name),
1839
                     options[:foreign_key] || reflection.active_record.to_s.foreign_key,
1840
                     connection.quote_table_name(parent.aliased_table_name),
1841
                     reflection.active_record.primary_key] +
1842
                  " #{join_type} %s ON %s.%s = %s.%s " % [
1843
                     table_name_and_alias,
1844 1845
                     connection.quote_table_name(aliased_table_name),
                     klass.primary_key,
1846
                     connection.quote_table_name(aliased_join_table_name),
1847
                     options[:association_foreign_key] || klass.to_s.foreign_key
1848 1849
                     ]
                when :has_many, :has_one
1850 1851
                  case
                    when reflection.macro == :has_many && reflection.options[:through]
1852
                      through_conditions = through_reflection.options[:conditions] ? "AND #{interpolate_sql(sanitize_sql(through_reflection.options[:conditions]))}" : ''
1853 1854 1855 1856

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

1857
                      if through_reflection.options[:as] # has_many :through against a polymorphic join
1858 1859
                        jt_foreign_key = through_reflection.options[:as].to_s + '_id'
                        jt_as_extra = " AND %s.%s = %s" % [
1860 1861
                          connection.quote_table_name(aliased_join_table_name),
                          connection.quote_column_name(through_reflection.options[:as].to_s + '_type'),
1862
                          klass.quote_value(parent.active_record.base_class.name)
1863
                        ]
1864
                      else
1865
                        jt_foreign_key = through_reflection.primary_key_name
1866
                      end
1867

1868 1869
                      case source_reflection.macro
                      when :has_many
1870 1871 1872
                        if source_reflection.options[:as]
                          first_key   = "#{source_reflection.options[:as]}_id"
                          second_key  = options[:foreign_key] || primary_key
1873
                          as_extra    = " AND %s.%s = %s" % [
1874 1875
                            connection.quote_table_name(aliased_table_name),
                            connection.quote_column_name("#{source_reflection.options[:as]}_type"),
1876
                            klass.quote_value(source_reflection.active_record.base_class.name)
1877 1878
                          ]
                        else
1879 1880 1881
                          first_key   = through_reflection.klass.base_class.to_s.foreign_key
                          second_key  = options[:foreign_key] || primary_key
                        end
1882

1883 1884
                        unless through_reflection.klass.descends_from_active_record?
                          jt_sti_extra = " AND %s.%s = %s" % [
1885 1886
                            connection.quote_table_name(aliased_join_table_name),
                            connection.quote_column_name(through_reflection.active_record.inheritance_column),
1887
                            through_reflection.klass.quote_value(through_reflection.klass.sti_name)]
1888 1889 1890 1891 1892 1893
                        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" % [
1894 1895
                            connection.quote_table_name(aliased_join_table_name),
                            connection.quote_column_name(reflection.source_reflection.options[:foreign_type]),
1896
                            klass.quote_value(reflection.options[:source_type])
1897
                          ]
1898
                        else
1899
                          second_key = source_reflection.primary_key_name
1900
                        end
1901
                      end
1902

1903
                      " #{join_type} %s ON (%s.%s = %s.%s%s%s%s) " % [
1904
                        table_alias_for(through_reflection.klass.table_name, aliased_join_table_name),
1905
                        connection.quote_table_name(parent.aliased_table_name),
1906
                        connection.quote_column_name(parent.primary_key),
1907
                        connection.quote_table_name(aliased_join_table_name),
1908
                        connection.quote_column_name(jt_foreign_key),
1909 1910
                        jt_as_extra, jt_source_extra, jt_sti_extra
                      ] +
1911
                      " #{join_type} %s ON (%s.%s = %s.%s%s) " % [
1912
                        table_name_and_alias,
1913
                        connection.quote_table_name(aliased_table_name),
1914
                        connection.quote_column_name(first_key),
1915
                        connection.quote_table_name(aliased_join_table_name),
1916
                        connection.quote_column_name(second_key),
1917 1918 1919
                        as_extra
                      ]

1920
                    when reflection.options[:as] && [:has_many, :has_one].include?(reflection.macro)
1921
                      " #{join_type} %s ON %s.%s = %s.%s AND %s.%s = %s" % [
1922
                        table_name_and_alias,
1923
                        connection.quote_table_name(aliased_table_name),
1924
                        "#{reflection.options[:as]}_id",
1925
                        connection.quote_table_name(parent.aliased_table_name),
1926
                        parent.primary_key,
1927
                        connection.quote_table_name(aliased_table_name),
1928
                        "#{reflection.options[:as]}_type",
1929
                        klass.quote_value(parent.active_record.base_class.name)
1930 1931
                      ]
                    else
1932
                      foreign_key = options[:foreign_key] || reflection.active_record.name.foreign_key
1933
                      " #{join_type} %s ON %s.%s = %s.%s " % [
1934
                        table_name_and_alias,
1935
                        aliased_table_name,
1936
                        foreign_key,
1937
                        parent.aliased_table_name,
1938
                        parent.primary_key
1939 1940
                      ]
                  end
1941
                when :belongs_to
1942
                  " #{join_type} %s ON %s.%s = %s.%s " % [
1943
                     table_name_and_alias,
1944 1945
                     connection.quote_table_name(aliased_table_name),
                     reflection.klass.primary_key,
1946
                     connection.quote_table_name(parent.aliased_table_name),
1947
                     options[:foreign_key] || reflection.primary_key_name
1948 1949 1950
                    ]
                else
                  ""
1951
              end || ''
1952
              join << %(AND %s.%s = %s ) % [
1953 1954
                connection.quote_table_name(aliased_table_name),
                connection.quote_column_name(klass.inheritance_column),
1955
                klass.quote_value(klass.sti_name)] unless klass.descends_from_active_record?
1956 1957 1958 1959 1960

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

1961
              join
1962
            end
1963

1964
            protected
1965 1966 1967 1968 1969 1970 1971 1972 1973 1974 1975 1976 1977 1978 1979 1980 1981 1982 1983
            
              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
              
1984 1985 1986
              def pluralize(table_name)
                ActiveRecord::Base.pluralize_table_names ? table_name.to_s.pluralize : table_name
              end
1987

1988
              def table_alias_for(table_name, table_alias)
1989
                 "#{reflection.active_record.connection.quote_table_name(table_name)} #{table_alias if table_name != table_alias}".strip
1990 1991 1992 1993 1994
              end

              def table_name_and_alias
                table_alias_for table_name, @aliased_table_name
              end
1995 1996

              def interpolate_sql(sql)
1997 1998
                instance_eval("%@#{sql.gsub('@', '\@')}@")
              end
1999 2000

            private
2001 2002 2003 2004
              def join_type
                "LEFT OUTER JOIN"
              end
          end
2005 2006 2007 2008 2009 2010 2011 2012 2013
        end

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

          class InnerJoinAssociation < JoinAssociation
2014 2015 2016 2017
            private
              def join_type
                "INNER JOIN"
              end
2018 2019
          end
        end
2020

2021
    end
D
Initial  
David Heinemeier Hansson 已提交
2022
  end
2023
end