base.rb 78.1 KB
Newer Older
D
Initial  
David Heinemeier Hansson 已提交
1
require 'yaml'
2
require 'set'
J
José Valim 已提交
3
require 'active_support/benchmarkable'
J
Jeremy Kemper 已提交
4
require 'active_support/dependencies'
5
require 'active_support/descendants_tracker'
6
require 'active_support/time'
7
require 'active_support/core_ext/class/attribute'
J
Jeremy Kemper 已提交
8 9 10 11 12 13 14 15
require 'active_support/core_ext/class/attribute_accessors'
require 'active_support/core_ext/class/delegating_attributes'
require 'active_support/core_ext/class/inheritable_attributes'
require 'active_support/core_ext/array/extract_options'
require 'active_support/core_ext/hash/deep_merge'
require 'active_support/core_ext/hash/indifferent_access'
require 'active_support/core_ext/hash/slice'
require 'active_support/core_ext/string/behavior'
16
require 'active_support/core_ext/kernel/singleton_class'
17
require 'active_support/core_ext/module/delegation'
18
require 'active_support/core_ext/module/introspection'
19
require 'active_support/core_ext/object/duplicable'
20
require 'active_support/core_ext/object/blank'
21 22
require 'arel'
require 'active_record/errors'
23
require 'active_record/log_subscriber'
D
Initial  
David Heinemeier Hansson 已提交
24 25

module ActiveRecord #:nodoc:
26
  # = Active Record
27
  #
28 29 30
  # Active Record objects don't specify their attributes directly, but rather infer them from
  # the table definition with which they're linked. Adding, removing, and changing attributes
  # and their type is done directly in the database. Any change is instantly reflected in the
31
  # Active Record objects. The mapping that binds a given Active Record class to a certain
32 33
  # database table will happen automatically in most common cases, but can be overwritten for the uncommon ones.
  #
34
  # See the mapping rules in table_name and the full example in link:files/activerecord/README_rdoc.html for more insight.
35
  #
D
Initial  
David Heinemeier Hansson 已提交
36
  # == Creation
37
  #
38 39
  # Active Records accept constructor parameters either in a hash or as a block. The hash
  # method is especially useful when you're receiving the data from somewhere else, like an
40
  # HTTP request. It works like this:
41
  #
42
  #   user = User.new(:name => "David", :occupation => "Code Artist")
D
Initial  
David Heinemeier Hansson 已提交
43
  #   user.name # => "David"
44
  #
D
Initial  
David Heinemeier Hansson 已提交
45
  # You can also use block initialization:
46
  #
D
Initial  
David Heinemeier Hansson 已提交
47 48 49 50
  #   user = User.new do |u|
  #     u.name = "David"
  #     u.occupation = "Code Artist"
  #   end
51
  #
D
Initial  
David Heinemeier Hansson 已提交
52
  # And of course you can just create a bare object and specify the attributes after the fact:
53
  #
D
Initial  
David Heinemeier Hansson 已提交
54 55 56
  #   user = User.new
  #   user.name = "David"
  #   user.occupation = "Code Artist"
57
  #
D
Initial  
David Heinemeier Hansson 已提交
58
  # == Conditions
59
  #
60
  # Conditions can either be specified as a string, array, or hash representing the WHERE-part of an SQL statement.
D
Initial  
David Heinemeier Hansson 已提交
61
  # The array form is to be used when the condition input is tainted and requires sanitization. The string form can
62
  # be used for statements that don't involve tainted data. The hash form works much like the array form, except
63
  # only equality and range is possible. Examples:
64
  #
65
  #   class User < ActiveRecord::Base
D
Initial  
David Heinemeier Hansson 已提交
66
  #     def self.authenticate_unsafely(user_name, password)
67
  #       where("user_name = '#{user_name}' AND password = '#{password}'").first
D
Initial  
David Heinemeier Hansson 已提交
68
  #     end
69
  #
D
Initial  
David Heinemeier Hansson 已提交
70
  #     def self.authenticate_safely(user_name, password)
71
  #       where("user_name = ? AND password = ?", user_name, password).first
D
Initial  
David Heinemeier Hansson 已提交
72
  #     end
73 74
  #
  #     def self.authenticate_safely_simply(user_name, password)
75
  #       where(:user_name => user_name, :password => password).first
76
  #     end
D
Initial  
David Heinemeier Hansson 已提交
77
  #   end
78
  #
79 80
  # The <tt>authenticate_unsafely</tt> method inserts the parameters directly into the query
  # and is thus susceptible to SQL-injection attacks if the <tt>user_name</tt> and +password+
81
  # parameters come directly from an HTTP request. The <tt>authenticate_safely</tt>  and
82 83
  # <tt>authenticate_safely_simply</tt> both will sanitize the <tt>user_name</tt> and +password+
  # before inserting them in the query, which will ensure that an attacker can't escape the
84
  # query and fake the login (or worse).
85
  #
86 87 88
  # When using multiple parameters in the conditions, it can easily become hard to read exactly
  # what the fourth or fifth question mark is supposed to represent. In those cases, you can
  # resort to named bind variables instead. That's done by replacing the question marks with
89
  # symbols and supplying a hash with values for the matching symbol keys:
90
  #
91
  #   Company.where(
92
  #     "id = :id AND name = :name AND division = :division AND created_at > :accounting_date",
93
  #     { :id => 3, :name => "37signals", :division => "First", :accounting_date => '2005-01-01' }
94
  #   ).first
95
  #
96 97 98
  # Similarly, a simple hash without a statement will generate conditions based on equality with the SQL AND
  # operator. For instance:
  #
99 100
  #   Student.where(:first_name => "Harvey", :status => 1)
  #   Student.where(params[:student])
101
  #
102 103
  # A range may be used in the hash to use the SQL BETWEEN operator:
  #
104
  #   Student.where(:grade => 9..12)
105
  #
P
Pratik Naik 已提交
106 107
  # An array may be used in the hash to use the SQL IN operator:
  #
108
  #   Student.where(:grade => [9,11,12])
P
Pratik Naik 已提交
109
  #
110
  # When joining tables, nested hashes or keys written in the form 'table_name.column_name'
111
  # can be used to qualify the table name of a particular condition. For instance:
P
Pratik Naik 已提交
112
  #
113 114
  #   Student.joins(:schools).where(:schools => { :type => 'public' })
  #   Student.joins(:schools).where('schools.type' => 'public' )
P
Pratik Naik 已提交
115
  #
D
Initial  
David Heinemeier Hansson 已提交
116
  # == Overwriting default accessors
117
  #
118 119 120 121
  # All column values are automatically available through basic accessors on the Active Record
  # object, but sometimes you want to specialize this behavior. This can be done by overwriting
  # the default accessors (using the same name as the attribute) and calling
  # <tt>read_attribute(attr_name)</tt> and <tt>write_attribute(attr_name, value)</tt> to actually
122
  # change things.
123
  #
D
Initial  
David Heinemeier Hansson 已提交
124 125
  #   class Song < ActiveRecord::Base
  #     # Uses an integer of seconds to hold the length of the song
126
  #
D
Initial  
David Heinemeier Hansson 已提交
127
  #     def length=(minutes)
128
  #       write_attribute(:length, minutes.to_i * 60)
D
Initial  
David Heinemeier Hansson 已提交
129
  #     end
130
  #
D
Initial  
David Heinemeier Hansson 已提交
131
  #     def length
132
  #       read_attribute(:length) / 60
D
Initial  
David Heinemeier Hansson 已提交
133 134
  #     end
  #   end
135
  #
136
  # You can alternatively use <tt>self[:attribute]=(value)</tt> and <tt>self[:attribute]</tt>
137
  # instead of <tt>write_attribute(:attribute, value)</tt> and <tt>read_attribute(:attribute)</tt>.
138
  #
139 140 141 142
  # == Attribute query methods
  #
  # In addition to the basic accessors, query methods are also automatically available on the Active Record object.
  # Query methods allow you to test whether an attribute value is present.
143
  #
144 145 146 147 148 149 150 151 152
  # For example, an Active Record User with the <tt>name</tt> attribute has a <tt>name?</tt> method that you can call
  # to determine whether the user has a name:
  #
  #   user = User.new(:name => "David")
  #   user.name? # => true
  #
  #   anonymous = User.new(:name => "")
  #   anonymous.name? # => false
  #
153
  # == Accessing attributes before they have been typecasted
154
  #
155 156 157
  # Sometimes you want to be able to read the raw attribute data without having the column-determined
  # typecast run its course first. That can be done by using the <tt><attribute>_before_type_cast</tt>
  # accessors that all attributes have. For example, if your Account model has a <tt>balance</tt> attribute,
158
  # you can call <tt>account.balance_before_type_cast</tt> or <tt>account.id_before_type_cast</tt>.
159
  #
160 161
  # This is especially useful in validation situations where the user might supply a string for an
  # integer field and you want to display the original string back in an error message. Accessing the
162
  # attribute normally would typecast the string to 0, which isn't what you want.
163
  #
164 165
  # == Dynamic attribute-based finders
  #
166 167 168 169 170
  # Dynamic attribute-based finders are a cleaner way of getting (and/or creating) objects
  # by simple queries without turning to SQL. They work by appending the name of an attribute
  # to <tt>find_by_</tt>, <tt>find_last_by_</tt>, or <tt>find_all_by_</tt> and thus produces finders
  # like <tt>Person.find_by_user_name</tt>, <tt>Person.find_all_by_last_name</tt>, and
  # <tt>Payment.find_by_transaction_id</tt>. Instead of writing
171
  # <tt>Person.where(:user_name => user_name).first</tt>, you just do <tt>Person.find_by_user_name(user_name)</tt>.
172
  # And instead of writing <tt>Person.where(:last_name => last_name).all</tt>, you just do
173
  # <tt>Person.find_all_by_last_name(last_name)</tt>.
174
  #
175
  # It's also possible to use multiple attributes in the same find by separating them with "_and_".
176
  #
177 178
  #  Person.where(:user_name => user_name, :password => password).first
  #  Person.find_by_user_name_and_password #with dynamic finder
179
  #
180 181
  #  Person.where(:user_name => user_name, :password => password, :gender => 'male').first
  #  Payment.find_by_user_name_and_password_and_gender
182
  #
183
  # It's even possible to call these dynamic finder methods on relations and named scopes.
184 185 186
  #
  #   Payment.order("created_on").find_all_by_amount(50)
  #   Payment.pending.find_last_by_amount(100)
187
  #
188 189 190 191
  # The same dynamic finder style can be used to create the object if it doesn't already exist.
  # This dynamic finder is called with <tt>find_or_create_by_</tt> and will return the object if
  # it already exists and otherwise creates it, then returns it. Protected attributes won't be set
  # unless they are given in a block.
192 193 194
  #
  #   # No 'Summer' tag exists
  #   Tag.find_or_create_by_name("Summer") # equal to Tag.create(:name => "Summer")
195
  #
196 197 198
  #   # Now the 'Summer' tag does exist
  #   Tag.find_or_create_by_name("Summer") # equal to Tag.find_by_name("Summer")
  #
199 200 201
  #   # Now 'Bob' exist and is an 'admin'
  #   User.find_or_create_by_name('Bob', :age => 40) { |u| u.admin = true }
  #
202
  # Use the <tt>find_or_initialize_by_</tt> finder if you want to return a new record without
203
  # saving it first. Protected attributes won't be set unless they are given in a block.
204 205 206
  #
  #   # No 'Winter' tag exists
  #   winter = Tag.find_or_initialize_by_name("Winter")
207
  #   winter.persisted? # false
208
  #
209
  # To find by a subset of the attributes to be used for instantiating a new object, pass a hash instead of
210
  # a list of parameters.
211 212 213
  #
  #   Tag.find_or_create_by_name(:name => "rails", :creator => current_user)
  #
214
  # That will either find an existing tag named "rails", or create a new one while setting the
215
  # user that created it.
216
  #
217 218 219 220
  # Just like <tt>find_by_*</tt>, you can also use <tt>scoped_by_*</tt> to retrieve data. The good thing about
  # using this feature is that the very first time result is returned using <tt>method_missing</tt> technique
  # but after that the method is declared on the class. Henceforth <tt>method_missing</tt> will not be hit.
  #
221
  #  User.scoped_by_user_name('David')
222
  #
223
  # == Saving arrays, hashes, and other non-mappable objects in text columns
224
  #
225
  # Active Record can serialize any object in text columns using YAML. To do so, you must
226
  # specify this with a call to the class method +serialize+.
227 228
  # This makes it possible to store arrays, hashes, and other non-mappable objects without doing
  # any additional work.
229
  #
D
Initial  
David Heinemeier Hansson 已提交
230 231 232
  #   class User < ActiveRecord::Base
  #     serialize :preferences
  #   end
233
  #
234
  #   user = User.create(:preferences => { "background" => "black", "display" => large })
D
Initial  
David Heinemeier Hansson 已提交
235
  #   User.find(user.id).preferences # => { "background" => "black", "display" => large }
236
  #
237
  # You can also specify a class option as the second parameter that'll raise an exception
238
  # if a serialized object is retrieved as a descendant of a class not in the hierarchy.
239
  #
D
Initial  
David Heinemeier Hansson 已提交
240
  #   class User < ActiveRecord::Base
241
  #     serialize :preferences, Hash
D
Initial  
David Heinemeier Hansson 已提交
242
  #   end
243
  #
244
  #   user = User.create(:preferences => %w( one two three ))
D
Initial  
David Heinemeier Hansson 已提交
245
  #   User.find(user.id).preferences    # raises SerializationTypeMismatch
246
  #
D
Initial  
David Heinemeier Hansson 已提交
247 248
  # == Single table inheritance
  #
249 250
  # Active Record allows inheritance by storing the name of the class in a column that by
  # default is named "type" (can be changed by overwriting <tt>Base.inheritance_column</tt>).
251
  # This means that an inheritance looking like this:
D
Initial  
David Heinemeier Hansson 已提交
252 253 254 255 256 257
  #
  #   class Company < ActiveRecord::Base; end
  #   class Firm < Company; end
  #   class Client < Company; end
  #   class PriorityClient < Client; end
  #
258 259
  # When you do <tt>Firm.create(:name => "37signals")</tt>, this record will be saved in
  # the companies table with type = "Firm". You can then fetch this row again using
260
  # <tt>Company.where(:name => '37signals').first</tt> and it will return a Firm object.
D
Initial  
David Heinemeier Hansson 已提交
261
  #
262 263
  # If you don't have a type column defined in your table, single-table inheritance won't
  # be triggered. In that case, it'll work just like normal subclasses with no special magic
264
  # for differentiating between them or reloading the right type with find.
265
  #
D
Initial  
David Heinemeier Hansson 已提交
266 267
  # Note, all the attributes for all the cases are kept in the same table. Read more:
  # http://www.martinfowler.com/eaaCatalog/singleTableInheritance.html
268
  #
D
Initial  
David Heinemeier Hansson 已提交
269 270
  # == Connection to multiple databases in different models
  #
271 272 273
  # Connections are usually created through ActiveRecord::Base.establish_connection and retrieved
  # by ActiveRecord::Base.connection. All classes inheriting from ActiveRecord::Base will use this
  # connection. But you can also set a class-specific connection. For example, if Course is an
274
  # ActiveRecord::Base, but resides in a different database, you can just say <tt>Course.establish_connection</tt>
P
Pratik Naik 已提交
275
  # and Course and all of its subclasses will use this connection instead.
D
Initial  
David Heinemeier Hansson 已提交
276
  #
277 278
  # This feature is implemented by keeping a connection pool in ActiveRecord::Base that is
  # a Hash indexed by the class. If a connection is requested, the retrieve_connection method
279
  # will go up the class-hierarchy until a connection is found in the connection pool.
D
Initial  
David Heinemeier Hansson 已提交
280 281
  #
  # == Exceptions
282
  #
P
Pratik Naik 已提交
283 284
  # * ActiveRecordError - Generic error class and superclass of all other errors raised by Active Record.
  # * AdapterNotSpecified - The configuration hash used in <tt>establish_connection</tt> didn't include an
D
Initial  
David Heinemeier Hansson 已提交
285
  #   <tt>:adapter</tt> key.
286
  # * AdapterNotFound - The <tt>:adapter</tt> key used in <tt>establish_connection</tt> specified a
287
  #   non-existent adapter
288
  #   (or a bad spelling of an existing one).
289
  # * AssociationTypeMismatch - The object assigned to the association wasn't of the type
290
  #   specified in the association definition.
P
Pratik Naik 已提交
291
  # * SerializationTypeMismatch - The serialized object wasn't of the class specified as the second parameter.
292
  # * ConnectionNotEstablished+ - No connection has been established. Use <tt>establish_connection</tt>
293
  #   before querying.
P
Pratik Naik 已提交
294 295 296 297 298
  # * RecordNotFound - No record responded to the +find+ method. Either the row with the given ID doesn't exist
  #   or the row didn't meet the additional restrictions. Some +find+ calls do not raise this exception to signal
  #   nothing was found, please check its documentation for further details.
  # * StatementInvalid - The database server rejected the SQL statement. The precise error is added in the message.
  # * MultiparameterAssignmentErrors - Collection of errors that occurred during a mass assignment using the
299
  #   <tt>attributes=</tt> method. The +errors+ property of this exception contains an array of
300
  #   AttributeAssignmentError
301
  #   objects that should be inspected to determine which attributes triggered the errors.
302
  # * AttributeAssignmentError - An error occurred while doing a mass assignment through the
303
  #   <tt>attributes=</tt> method.
304
  #   You can inspect the +attribute+ property of the exception object to determine which attribute
305
  #   triggered the error.
306
  #
307
  # *Note*: The attributes listed are class-level attributes (accessible from both the class and instance level).
P
Pratik Naik 已提交
308
  # So it's possible to assign a logger to the class through <tt>Base.logger=</tt> which will then be used by all
D
Initial  
David Heinemeier Hansson 已提交
309 310
  # instances in the current object space.
  class Base
E
Emilio Tagua 已提交
311
    ##
P
Pratik Naik 已提交
312
    # :singleton-method:
313 314
    # Accepts a logger conforming to the interface of Log4r or the default Ruby 1.8+ Logger class,
    # which is then passed on to any new database connections made and which can be retrieved on both
315
    # a class and instance level by calling +logger+.
316
    cattr_accessor :logger, :instance_writer => false
J
Jeremy Kemper 已提交
317

P
Pratik Naik 已提交
318 319
    ##
    # :singleton-method:
P
Pratik Naik 已提交
320 321 322 323
    # Contains the database configuration - as is typically stored in config/database.yml -
    # as a Hash.
    #
    # For example, the following database.yml...
E
Emilio Tagua 已提交
324
    #
P
Pratik Naik 已提交
325 326 327
    #   development:
    #     adapter: sqlite3
    #     database: db/development.sqlite3
E
Emilio Tagua 已提交
328
    #
P
Pratik Naik 已提交
329 330 331 332 333 334 335 336 337 338 339 340 341 342 343 344
    #   production:
    #     adapter: sqlite3
    #     database: db/production.sqlite3
    #
    # ...would result in ActiveRecord::Base.configurations to look like this:
    #
    #   {
    #      'development' => {
    #         'adapter'  => 'sqlite3',
    #         'database' => 'db/development.sqlite3'
    #      },
    #      'production' => {
    #         'adapter'  => 'sqlite3',
    #         'database' => 'db/production.sqlite3'
    #      }
    #   }
345
    cattr_accessor :configurations, :instance_writer => false
346 347
    @@configurations = {}

P
Pratik Naik 已提交
348 349
    ##
    # :singleton-method:
350 351 352
    # Accessor for the prefix type that will be prepended to every primary key column name.
    # The options are :table_name and :table_name_with_underscore. If the first is specified,
    # the Product class will look for "productid" instead of "id" as the primary column. If the
353
    # latter is specified, the Product class will look for "product_id" instead of "id". Remember
354
    # that this is a global setting for all Active Records.
355
    cattr_accessor :primary_key_prefix_type, :instance_writer => false
D
Initial  
David Heinemeier Hansson 已提交
356 357
    @@primary_key_prefix_type = nil

P
Pratik Naik 已提交
358 359
    ##
    # :singleton-method:
360 361 362
    # Accessor for the name of the prefix string to prepend to every table name. So if set
    # to "basecamp_", all table names will be named like "basecamp_projects", "basecamp_people",
    # etc. This is a convenient way of creating a namespace for tables in a shared database.
363
    # By default, the prefix is the empty string.
364
    #
365 366
    # If you are organising your models within modules you can add a prefix to the models within
    # a namespace by defining a singleton method in the parent module called table_name_prefix which
367
    # returns your chosen prefix.
368 369
    class_attribute :table_name_prefix, :instance_writer => false
    self.table_name_prefix = ""
D
Initial  
David Heinemeier Hansson 已提交
370

P
Pratik Naik 已提交
371 372
    ##
    # :singleton-method:
D
Initial  
David Heinemeier Hansson 已提交
373 374
    # Works like +table_name_prefix+, but appends instead of prepends (set to "_basecamp" gives "projects_basecamp",
    # "people_basecamp"). By default, the suffix is the empty string.
375 376
    class_attribute :table_name_suffix, :instance_writer => false
    self.table_name_suffix = ""
D
Initial  
David Heinemeier Hansson 已提交
377

P
Pratik Naik 已提交
378 379
    ##
    # :singleton-method:
380
    # Indicates whether table names should be the pluralized versions of the corresponding class names.
P
Pratik Naik 已提交
381
    # If true, the default table name for a Product class will be +products+. If false, it would just be +product+.
D
Initial  
David Heinemeier Hansson 已提交
382
    # See table_name for the full rules on table/class naming. This is true, by default.
383
    cattr_accessor :pluralize_table_names, :instance_writer => false
D
Initial  
David Heinemeier Hansson 已提交
384 385
    @@pluralize_table_names = true

P
Pratik Naik 已提交
386 387
    ##
    # :singleton-method:
388
    # Determines whether to use Time.local (using :local) or Time.utc (using :utc) when pulling
389
    # dates and times from the database. This is set to :local by default.
390
    cattr_accessor :default_timezone, :instance_writer => false
391
    @@default_timezone = :local
392

P
Pratik Naik 已提交
393 394
    ##
    # :singleton-method:
395 396
    # Specifies the format to use when dumping the database schema with Rails'
    # Rakefile.  If :sql, the schema is dumped as (potentially database-
397
    # specific) SQL statements.  If :ruby, the schema is dumped as an
398 399 400
    # ActiveRecord::Schema file which can be loaded into any database that
    # supports migrations.  Use :ruby if you want to have different database
    # adapters for, e.g., your development and test environments.
401
    cattr_accessor :schema_format , :instance_writer => false
402
    @@schema_format = :ruby
403

P
Pratik Naik 已提交
404 405
    ##
    # :singleton-method:
406
    # Specify whether or not to use timestamps for migration versions
407 408 409
    cattr_accessor :timestamped_migrations , :instance_writer => false
    @@timestamped_migrations = true

410
    # Determine whether to store the full constant name including namespace when using STI
411
    class_attribute :store_full_sti_class
412
    self.store_full_sti_class = true
413

414 415 416 417
    # Stores the default scope for the class
    class_inheritable_accessor :default_scoping, :instance_writer => false
    self.default_scoping = []

J
Jeremy Kemper 已提交
418 419 420 421 422
    # Returns a hash of all the attributes that have been specified for serialization as
    # keys and their class restriction as values.
    class_attribute :serialized_attributes
    self.serialized_attributes = {}

D
Initial  
David Heinemeier Hansson 已提交
423
    class << self # Class methods
424
      delegate :find, :first, :last, :all, :destroy, :destroy_all, :exists?, :delete, :delete_all, :update, :update_all, :to => :scoped
P
Pratik Naik 已提交
425
      delegate :find_each, :find_in_batches, :to => :scoped
426
      delegate :select, :group, :order, :except, :limit, :offset, :joins, :where, :preload, :eager_load, :includes, :from, :lock, :readonly, :having, :create_with, :to => :scoped
427
      delegate :count, :average, :minimum, :maximum, :sum, :calculate, :to => :scoped
428

P
Pratik Naik 已提交
429
      # Executes a custom SQL query against your database and returns all the results.  The results will
430
      # be returned as an array with columns requested encapsulated as attributes of the model you call
P
Pratik Naik 已提交
431 432
      # this method from.  If you call <tt>Product.find_by_sql</tt> then the results will be returned in
      # a Product object with the attributes you specified in the SQL query.
433
      #
434 435
      # If you call a complicated SQL query which spans multiple tables the columns specified by the
      # SELECT will be attributes of the model, whether or not they are columns of the corresponding
436 437
      # table.
      #
P
Pratik Naik 已提交
438
      # The +sql+ parameter is a full SQL query as a string.  It will be called as is, there will be
439 440
      # no database agnostic conversions performed.  This should be a last resort because using, for example,
      # MySQL specific terms will lock you to using that particular database engine or require you to
P
Pratik Naik 已提交
441
      # change your call if you switch engines.
442 443
      #
      # ==== Examples
P
Pratik Naik 已提交
444
      #   # A simple SQL query spanning multiple tables
445 446 447 448 449 450
      #   Post.find_by_sql "SELECT p.title, c.author FROM posts p, comments c WHERE p.id = c.post_id"
      #   > [#<Post:0x36bff9c @attributes={"title"=>"Ruby Meetup", "first_name"=>"Quentin"}>, ...]
      #
      #   # You can use the same string replacement techniques as you can with ActiveRecord#find
      #   Post.find_by_sql ["SELECT title FROM posts WHERE author = ? AND created > ?", author_id, start_date]
      #   > [#<Post:0x36bff9c @attributes={"first_name"=>"The Cheap Man Buys Twice"}>, ...]
451 452
      def find_by_sql(sql, binds = [])
        connection.select_all(sanitize_sql(sql), "#{name} Load", binds).collect! { |record| instantiate(record) }
D
Initial  
David Heinemeier Hansson 已提交
453
      end
454

455
      # Creates an object (or multiple objects) and saves it to the database, if validations pass.
456 457 458 459 460 461 462 463
      # The resulting object is returned whether the object was saved successfully to the database or not.
      #
      # The +attributes+ parameter can be either be a Hash or an Array of Hashes.  These Hashes describe the
      # attributes on the objects that are to be created.
      #
      # ==== Examples
      #   # Create a single new object
      #   User.create(:first_name => 'Jamie')
464
      #
465
      #   # Create an Array of new objects
P
Pratik Naik 已提交
466
      #   User.create([{ :first_name => 'Jamie' }, { :first_name => 'Jeremy' }])
467 468 469 470 471 472 473
      #
      #   # Create a single object and pass it into a block to set other attributes.
      #   User.create(:first_name => 'Jamie') do |u|
      #     u.is_admin = false
      #   end
      #
      #   # Creating an Array of new objects using a block, where the block is executed for each object:
P
Pratik Naik 已提交
474
      #   User.create([{ :first_name => 'Jamie' }, { :first_name => 'Jeremy' }]) do |u|
475
      #     u.is_admin = false
476
      #   end
477
      def create(attributes = nil, &block)
478
        if attributes.is_a?(Array)
479
          attributes.collect { |attr| create(attr, &block) }
480 481
        else
          object = new(attributes)
482
          yield(object) if block_given?
483 484 485
          object.save
          object
        end
D
Initial  
David Heinemeier Hansson 已提交
486 487 488
      end

      # Returns the result of an SQL statement that should only include a COUNT(*) in the SELECT part.
489
      # The use of this method should be restricted to complicated SQL queries that can't be executed
490 491
      # using the ActiveRecord::Calculations class methods.  Look into those before using this.
      #
P
Pratik Naik 已提交
492
      # ==== Parameters
493
      #
P
Pratik Naik 已提交
494
      # * +sql+ - An SQL statement which should return a count query from the database, see the example below.
495 496 497
      #
      # ==== Examples
      #
498
      #   Product.count_by_sql "SELECT COUNT(*) FROM sales s, customers c WHERE s.customer_id = c.id"
D
Initial  
David Heinemeier Hansson 已提交
499
      def count_by_sql(sql)
500
        sql = sanitize_conditions(sql)
501
        connection.select_value(sql, "#{name} Count").to_i
D
Initial  
David Heinemeier Hansson 已提交
502
      end
503

N
Neeraj Singh 已提交
504 505
      # Attributes listed as readonly will be used to create a new record but update operations will
      # ignore these fields.
506
      def attr_readonly(*attributes)
507
        write_inheritable_attribute(:attr_readonly, Set.new(attributes.map { |a| a.to_s }) + (readonly_attributes || []))
D
Initial  
David Heinemeier Hansson 已提交
508
      end
509

510 511 512
      # Returns an array of all the attributes that have been specified as readonly.
      def readonly_attributes
        read_inheritable_attribute(:attr_readonly) || []
D
Initial  
David Heinemeier Hansson 已提交
513 514
      end

515 516 517
      # If you have an attribute that needs to be saved to the database as an object, and retrieved as the same object,
      # then specify the name of that attribute using this method and it will be handled automatically.
      # The serialization is done through YAML. If +class_name+ is specified, the serialized object must be of that
P
Pratik Naik 已提交
518
      # class on retrieval or SerializationTypeMismatch will be raised.
D
David Heinemeier Hansson 已提交
519
      #
P
Pratik Naik 已提交
520
      # ==== Parameters
D
David Heinemeier Hansson 已提交
521
      #
P
Pratik Naik 已提交
522 523
      # * +attr_name+ - The field name that should be serialized.
      # * +class_name+ - Optional, class name that the object type should be equal to.
D
David Heinemeier Hansson 已提交
524 525 526 527 528 529
      #
      # ==== Example
      #   # Serialize a preferences attribute
      #   class User
      #     serialize :preferences
      #   end
D
Initial  
David Heinemeier Hansson 已提交
530
      def serialize(attr_name, class_name = Object)
531
        serialized_attributes[attr_name.to_s] = class_name
D
Initial  
David Heinemeier Hansson 已提交
532
      end
533

534 535
      # Guesses the table name (in forced lower-case) based on the name of the class in the
      # inheritance hierarchy descending directly from ActiveRecord::Base. So if the hierarchy
536
      # looks like: Reply < Message < ActiveRecord::Base, then Message is used
537 538
      # to guess the table name even when called on Reply. The rules used to do the guess
      # are handled by the Inflector class in Active Support, which knows almost all common
539
      # English inflections. You can add new inflections in config/initializers/inflections.rb.
D
Initial  
David Heinemeier Hansson 已提交
540
      #
541
      # Nested classes are given table names prefixed by the singular form of
P
Pratik Naik 已提交
542 543 544
      # the parent's table name. Enclosing modules are not considered.
      #
      # ==== Examples
545 546
      #
      #   class Invoice < ActiveRecord::Base; end;
547 548
      #   file                  class               table_name
      #   invoice.rb            Invoice             invoices
549 550 551 552 553 554 555 556
      #
      #   class Invoice < ActiveRecord::Base; class Lineitem < ActiveRecord::Base; end; end;
      #   file                  class               table_name
      #   invoice.rb            Invoice::Lineitem   invoice_lineitems
      #
      #   module Invoice; class Lineitem < ActiveRecord::Base; end; end;
      #   file                  class               table_name
      #   invoice/lineitem.rb   Invoice::Lineitem   lineitems
D
Initial  
David Heinemeier Hansson 已提交
557
      #
P
Pratik Naik 已提交
558 559
      # Additionally, the class-level +table_name_prefix+ is prepended and the
      # +table_name_suffix+ is appended.  So if you have "myapp_" as a prefix,
560 561 562 563 564
      # the table name guess for an Invoice class becomes "myapp_invoices".
      # Invoice::Lineitem becomes "myapp_invoice_lineitems".
      #
      # You can also overwrite this class method to allow for unguessable
      # links, such as a Mouse class with a link to a "mice" table. Example:
D
Initial  
David Heinemeier Hansson 已提交
565 566
      #
      #   class Mouse < ActiveRecord::Base
567
      #     set_table_name "mice"
D
Initial  
David Heinemeier Hansson 已提交
568
      #   end
569
      def table_name
570 571 572
        reset_table_name
      end

573
      # Returns a quoted version of the table name, used to construct SQL statements.
P
Pratik Naik 已提交
574 575 576 577
      def quoted_table_name
        @quoted_table_name ||= connection.quote_table_name(table_name)
      end

578
      # Computes the table name, (re)sets it internally, and returns it.
D
David Heinemeier Hansson 已提交
579
      def reset_table_name #:nodoc:
580
        self.table_name = compute_table_name
D
Initial  
David Heinemeier Hansson 已提交
581 582
      end

583
      def full_table_name_prefix #:nodoc:
584 585 586
        (parents.detect{ |p| p.respond_to?(:table_name_prefix) } || self).table_name_prefix
      end

587
      # Defines the column name for use with single table inheritance. Use
N
Neeraj Singh 已提交
588
      # <tt>set_inheritance_column</tt> to set a different value.
D
Initial  
David Heinemeier Hansson 已提交
589
      def inheritance_column
A
Aaron Patterson 已提交
590
        @inheritance_column ||= "type"
D
Initial  
David Heinemeier Hansson 已提交
591 592
      end

593 594
      # Lazy-set the sequence name to the connection's default.  This method
      # is only ever called once since set_sequence_name overrides it.
D
David Heinemeier Hansson 已提交
595
      def sequence_name #:nodoc:
596 597 598
        reset_sequence_name
      end

D
David Heinemeier Hansson 已提交
599
      def reset_sequence_name #:nodoc:
600 601 602
        default = connection.default_sequence_name(table_name, primary_key)
        set_sequence_name(default)
        default
603 604
      end

605
      # Sets the table name. If the value is nil or false  then the value returned by the given
N
Neeraj Singh 已提交
606
      # block is used.
607 608 609 610
      #
      #   class Project < ActiveRecord::Base
      #     set_table_name "project"
      #   end
611
      def set_table_name(value = nil, &block)
612
        @quoted_table_name = nil
613 614 615 616 617 618
        define_attr_method :table_name, value, &block
      end
      alias :table_name= :set_table_name

      # Sets the name of the inheritance column to use to the given value,
      # or (if the value # is nil or false) to the value returned by the
619
      # given block.
620 621 622 623 624 625
      #
      #   class Project < ActiveRecord::Base
      #     set_inheritance_column do
      #       original_inheritance_column + "_id"
      #     end
      #   end
626
      def set_inheritance_column(value = nil, &block)
627 628 629 630
        define_attr_method :inheritance_column, value, &block
      end
      alias :inheritance_column= :set_inheritance_column

631 632
      # Sets the name of the sequence to use when generating ids to the given
      # value, or (if the value is nil or false) to the value returned by the
633 634
      # given block. This is required for Oracle and is useful for any
      # database which relies on sequences for primary key generation.
635
      #
636 637 638 639 640
      # If a sequence name is not explicitly set when using Oracle or Firebird,
      # it will default to the commonly used pattern of: #{table_name}_seq
      #
      # If a sequence name is not explicitly set when using PostgreSQL, it
      # will discover the sequence corresponding to your primary key for you.
641 642 643 644
      #
      #   class Project < ActiveRecord::Base
      #     set_sequence_name "projectseq"   # default would have been "project_seq"
      #   end
645
      def set_sequence_name(value = nil, &block)
646 647 648 649
        define_attr_method :sequence_name, value, &block
      end
      alias :sequence_name= :set_sequence_name

650 651
      # Indicates whether the table associated with this class exists
      def table_exists?
652
        connection.table_exists?(table_name)
653 654
      end

D
Initial  
David Heinemeier Hansson 已提交
655 656
      # Returns an array of column objects for the table associated with this class.
      def columns
657
        unless defined?(@columns) && @columns
658
          @columns = connection.columns(table_name, "#{name} Columns")
659
          @columns.each { |column| column.primary = column.name == primary_key }
660 661
        end
        @columns
D
Initial  
David Heinemeier Hansson 已提交
662
      end
663

664
      # Returns a hash of column objects for the table associated with this class.
D
Initial  
David Heinemeier Hansson 已提交
665
      def columns_hash
666
        @columns_hash ||= Hash[columns.map { |column| [column.name, column] }]
D
Initial  
David Heinemeier Hansson 已提交
667
      end
668

D
David Heinemeier Hansson 已提交
669
      # Returns an array of column names as strings.
670
      def column_names
671
        @column_names ||= columns.map { |column| column.name }
672
      end
D
Initial  
David Heinemeier Hansson 已提交
673

674 675
      # Returns an array of column objects where the primary id, all columns ending in "_id" or "_count",
      # and columns used for single table inheritance have been removed.
D
Initial  
David Heinemeier Hansson 已提交
676
      def content_columns
677
        @content_columns ||= columns.reject { |c| c.primary || c.name =~ /(_id|_count)$/ || c.name == inheritance_column }
D
Initial  
David Heinemeier Hansson 已提交
678 679 680 681
      end

      # Returns a hash of all the methods added to query each of the columns in the table with the name of the method as the key
      # and true as the value. This makes it possible to do O(1) lookups in respond_to? to check if a given method for attribute
682
      # is available.
D
David Heinemeier Hansson 已提交
683
      def column_methods_hash #:nodoc:
684
        @dynamic_methods_hash ||= column_names.inject(Hash.new(false)) do |methods, attr|
685 686 687 688 689
          attr_name = attr.to_s
          methods[attr.to_sym]       = attr_name
          methods["#{attr}=".to_sym] = attr_name
          methods["#{attr}?".to_sym] = attr_name
          methods["#{attr}_before_type_cast".to_sym] = attr_name
D
Initial  
David Heinemeier Hansson 已提交
690 691 692
          methods
        end
      end
693

P
Pratik Naik 已提交
694 695 696 697 698 699 700 701 702 703 704 705 706 707 708 709 710 711 712 713 714 715 716 717 718 719
      # Resets all the cached information about columns, which will cause them
      # to be reloaded on the next request.
      #
      # The most common usage pattern for this method is probably in a migration,
      # when just after creating a table you want to populate it with some default
      # values, eg:
      #
      #  class CreateJobLevels < ActiveRecord::Migration
      #    def self.up
      #      create_table :job_levels do |t|
      #        t.integer :id
      #        t.string :name
      #
      #        t.timestamps
      #      end
      #
      #      JobLevel.reset_column_information
      #      %w{assistant executive manager director}.each do |type|
      #        JobLevel.create(:name => type)
      #      end
      #    end
      #
      #    def self.down
      #      drop_table :job_levels
      #    end
      #  end
720
      def reset_column_information
A
Aaron Patterson 已提交
721
        connection.clear_cache!
722
        undefine_attribute_methods
723
        @column_names = @columns = @columns_hash = @content_columns = @dynamic_methods_hash = @inheritance_column = nil
724
        @arel_engine = @relation = @arel_table = nil
725 726
      end

727
      def reset_column_information_and_inheritable_attributes_for_all_subclasses#:nodoc:
728
        descendants.each { |klass| klass.reset_inheritable_attributes; klass.reset_column_information }
729
      end
D
Initial  
David Heinemeier Hansson 已提交
730

731
      def attribute_method?(attribute)
J
José Valim 已提交
732
        super || (table_exists? && column_names.include?(attribute.to_s.sub(/=$/, '')))
733 734
      end

735 736
      # Set the lookup ancestors for ActiveModel.
      def lookup_ancestors #:nodoc:
737 738
        klass = self
        classes = [klass]
739 740
        return classes if klass == ActiveRecord::Base

E
Emilio Tagua 已提交
741
        while klass != klass.base_class
742 743 744 745 746
          classes << klass = klass.superclass
        end
        classes
      end

747 748 749
      # Set the i18n scope to overwrite ActiveModel.
      def i18n_scope #:nodoc:
        :activerecord
D
Initial  
David Heinemeier Hansson 已提交
750
      end
751

752 753 754 755 756 757 758
      # True if this isn't a concrete subclass needing a STI type condition.
      def descends_from_active_record?
        if superclass.abstract_class?
          superclass.descends_from_active_record?
        else
          superclass == Base || !columns_hash.include?(inheritance_column)
        end
D
Initial  
David Heinemeier Hansson 已提交
759 760
      end

761 762 763 764 765
      def finder_needs_type_condition? #:nodoc:
        # This is like this because benchmarking justifies the strange :false stuff
        :true == (@finder_needs_type_condition ||= descends_from_active_record? ? :false : :true)
      end

766
      # Returns a string like 'Post id:integer, title:string, body:text'
767
      def inspect
768 769 770 771
        if self == Base
          super
        elsif abstract_class?
          "#{super}(abstract)"
772
        elsif table_exists?
773 774
          attr_list = columns.map { |c| "#{c.name}: #{c.type}" } * ', '
          "#{super}(#{attr_list})"
775 776
        else
          "#{super}(Table doesn't exist)"
777
        end
778 779
      end

780
      def quote_value(value, column = nil) #:nodoc:
781
        connection.quote(value,column)
782 783
      end

784
      # Used to sanitize objects before they're used in an SQL SELECT statement. Delegates to <tt>connection.quote</tt>.
785
      def sanitize(object) #:nodoc:
786
        connection.quote(object)
D
Initial  
David Heinemeier Hansson 已提交
787 788
      end

789 790 791
      # Overwrite the default class equality method to provide support for association proxies.
      def ===(object)
        object.is_a?(self)
792
      end
793

794 795 796
      # Returns the base AR subclass that this class descends from. If A
      # extends AR::Base, A.base_class will return A. If B descends from A
      # through some arbitrarily deep hierarchy, B.base_class will return A.
797 798 799
      #
      # If B < A and C < B and if A is an abstract_class then both B.base_class
      # and C.base_class would return B as the answer since A is an abstract_class.
800 801 802 803
      def base_class
        class_of_active_record_descendant(self)
      end

P
Pratik Naik 已提交
804
      # Set this to true if this is an abstract class (see <tt>abstract_class?</tt>).
805 806
      attr_accessor :abstract_class

807
      # Returns whether this class is an abstract class or not.
808
      def abstract_class?
809
        defined?(@abstract_class) && @abstract_class == true
810 811
      end

812
      def respond_to?(method_id, include_private = false)
813 814
        if match = DynamicFinderMatch.match(method_id)
          return true if all_attributes_exists?(match.attribute_names)
815 816
        elsif match = DynamicScopeMatch.match(method_id)
          return true if all_attributes_exists?(match.attribute_names)
817
        end
818

819 820 821
        super
      end

822 823 824 825
      def sti_name
        store_full_sti_class ? name : name.demodulize
      end

P
Pratik Naik 已提交
826
      def arel_table
827
        @arel_table ||= Arel::Table.new(table_name, arel_engine)
828 829
      end

830 831
      def arel_engine
        @arel_engine ||= begin
832 833 834
          if self == ActiveRecord::Base
            Arel::Table.engine
          else
835
            connection_handler.connection_pools[name] ? self : superclass.arel_engine
836 837
          end
        end
838 839
      end

840 841 842 843 844 845 846 847 848 849 850 851 852 853 854 855
      # Returns a scope for this class without taking into account the default_scope.
      #
      #   class Post < ActiveRecord::Base
      #     default_scope :published => true
      #   end
      #
      #   Post.all          # Fires "SELECT * FROM posts WHERE published = true"
      #   Post.unscoped.all # Fires "SELECT * FROM posts"
      #
      # This method also accepts a block meaning that all queries inside the block will
      # not use the default_scope:
      #
      #   Post.unscoped {
      #     limit(10) # Fires "SELECT * FROM posts LIMIT 10"
      #   }
      #
856 857 858
      # It is recommended to use block form of unscoped because chaining unscoped with <tt>named_scope</tt>
      # does not work. Assuming that <tt>published</tt> is a <tt>named_scope</tt> following two statements are same.
      #
859 860
      # Post.unscoped.published
      # Post.published
861
      def unscoped #:nodoc:
862 863 864 865 866 867 868 869
        block_given? ? relation.scoping { yield } : relation
      end

      def scoped_methods #:nodoc:
        key = :"#{self}_scoped_methods"
        Thread.current[key] = Thread.current[key].presence || self.default_scoping.dup
      end

870 871 872 873
      def before_remove_const #:nodoc:
        reset_scoped_methods
      end

D
Initial  
David Heinemeier Hansson 已提交
874
      private
J
José Valim 已提交
875 876 877 878 879 880

        def relation #:nodoc:
          @relation ||= Relation.new(self, arel_table)
          finder_needs_type_condition? ? @relation.where(type_condition) : @relation
        end

881 882 883
        # Finder methods must instantiate through this method to work with the
        # single-table inheritance model that makes it possible to create
        # objects of different types from the same table.
D
Initial  
David Heinemeier Hansson 已提交
884
        def instantiate(record)
885 886 887
          model = find_sti_class(record[inheritance_column]).allocate
          model.init_with('attributes' => record)
          model
D
Initial  
David Heinemeier Hansson 已提交
888
        end
889

890
        def find_sti_class(type_name)
891
          if type_name.blank? || !columns_hash.include?(inheritance_column)
892 893 894
            self
          else
            begin
895 896 897 898 899
              if store_full_sti_class
                ActiveSupport::Dependencies.constantize(type_name)
              else
                compute_type(type_name)
              end
900 901 902 903 904 905 906 907 908 909
            rescue NameError
              raise SubclassNotFound,
                "The single-table inheritance mechanism failed to locate the subclass: '#{type_name}'. " +
                "This error is raised because the column '#{inheritance_column}' is reserved for storing the class in case of inheritance. " +
                "Please rename this column if you didn't intend it to be used for storing the inheritance class " +
                "or overwrite #{name}.inheritance_column to use another column for that information."
            end
          end
        end

910
        def construct_finder_arel(options = {}, scope = nil)
911
          relation = options.is_a?(Hash) ? unscoped.apply_finder_options(options) : options
912
          relation = scope.merge(relation) if scope
913
          relation
914 915
        end

916
        def type_condition
917
          sti_column = arel_table[inheritance_column]
918
          condition = sti_column.eq(sti_name)
919
          descendants.each { |subclass| condition = condition.or(sti_column.eq(subclass.sti_name)) }
920

921
          condition
D
Initial  
David Heinemeier Hansson 已提交
922 923 924
        end

        # Guesses the table name, but does not decorate it with prefix and suffix information.
925
        def undecorated_table_name(class_name = base_class.name)
926 927
          table_name = class_name.to_s.demodulize.underscore
          table_name = table_name.pluralize if pluralize_table_names
928
          table_name
D
Initial  
David Heinemeier Hansson 已提交
929 930
        end

931 932 933 934 935 936 937 938 939 940 941 942 943 944 945 946 947
        # Computes and returns a table name according to default conventions.
        def compute_table_name
          base = base_class
          if self == base
            # Nested classes are prefixed with singular parent table name.
            if parent < ActiveRecord::Base && !parent.abstract_class?
              contained = parent.table_name
              contained = contained.singularize if parent.pluralize_table_names
              contained << '_'
            end
            "#{full_table_name_prefix}#{contained}#{undecorated_table_name(name)}#{table_name_suffix}"
          else
            # STI subclasses always use their superclass' table.
            base.table_name
          end
        end

948
        # Enables dynamic finders like <tt>User.find_by_user_name(user_name)</tt> and
949 950
        # <tt>User.scoped_by_user_name(user_name). Refer to Dynamic attribute-based finders
        # section at the top of this file for more detailed information.
951
        #
952
        # It's even possible to use all the additional parameters to +find+. For example, the
953
        # full interface for +find_all_by_amount+ is actually <tt>find_all_by_amount(amount, options)</tt>.
954
        #
955
        # Each dynamic finder using <tt>scoped_by_*</tt> is also defined in the class after it
956
        # is first invoked, so that future attempts to use it do not run through method_missing.
957
        def method_missing(method_id, *arguments, &block)
958 959
          if match = DynamicFinderMatch.match(method_id)
            attribute_names = match.attribute_names
960
            super unless all_attributes_exists?(attribute_names)
961
            if match.finder?
962
              options = arguments.extract_options!
963
              relation = options.any? ? construct_finder_arel(options, current_scoped_methods) : scoped
964
              relation.send :find_by_attributes, match, attribute_names, *arguments
965
            elsif match.instantiator?
966
              scoped.send :find_or_instantiator_by_attributes, match, attribute_names, *arguments, &block
967
            end
968 969 970 971
          elsif match = DynamicScopeMatch.match(method_id)
            attribute_names = match.attribute_names
            super unless all_attributes_exists?(attribute_names)
            if match.scope?
972
              self.class_eval <<-METHOD, __FILE__, __LINE__ + 1
973 974 975 976 977
                def self.#{method_id}(*args)                                    # def self.scoped_by_user_name_and_password(*args)
                  attributes = Hash[[:#{attribute_names.join(',:')}].zip(args)] #   attributes = Hash[[:user_name, :password].zip(args)]
                                                                                #
                  scoped(:conditions => attributes)                             #   scoped(:conditions => attributes)
                end                                                             # end
978
              METHOD
979 980
              send(method_id, *arguments)
            end
981 982 983 984
          else
            super
          end
        end
D
Initial  
David Heinemeier Hansson 已提交
985

986 987
        # Similar in purpose to +expand_hash_conditions_for_aggregates+.
        def expand_attribute_names_for_aggregates(attribute_names)
988
          attribute_names.map { |attribute_name|
989
            unless (aggregation = reflect_on_aggregation(attribute_name.to_sym)).nil?
990 991
              aggregate_mapping(aggregation).map do |field_attr, _|
                field_attr.to_sym
992 993
              end
            else
994
              attribute_name.to_sym
995
            end
996
          }.flatten
997 998
        end

999
        def all_attributes_exists?(attribute_names)
1000 1001
          (expand_attribute_names_for_aggregates(attribute_names) -
           column_methods_hash.keys).empty?
1002
        end
1003

D
Initial  
David Heinemeier Hansson 已提交
1004
      protected
N
Neeraj Singh 已提交
1005 1006
        # with_scope lets you apply options to inner block incrementally. It takes a hash and the keys must be
        # <tt>:find</tt> or <tt>:create</tt>. <tt>:find</tt> parameter is <tt>Relation</tt> while
1007
        # <tt>:create</tt> parameters are an attributes hash.
1008 1009 1010
        #
        #   class Article < ActiveRecord::Base
        #     def self.create_with_scope
1011
        #       with_scope(:find => where(:blog_id => 1), :create => { :blog_id => 1 }) do
1012 1013 1014 1015 1016 1017 1018
        #         find(1) # => SELECT * from articles WHERE blog_id = 1 AND id = 1
        #         a = create(1)
        #         a.blog_id # => 1
        #       end
        #     end
        #   end
        #
D
David Heinemeier Hansson 已提交
1019
        # In nested scopings, all previous parameters are overwritten by the innermost rule, with the exception of
1020
        # <tt>where</tt>, <tt>includes</tt>, and <tt>joins</tt> operations in <tt>Relation</tt>, which are merged.
P
Pratik Naik 已提交
1021
        #
1022
        # <tt>joins</tt> operations are uniqued so multiple scopes can join in the same table without table aliasing
P
Pratik Naik 已提交
1023 1024
        # problems.  If you need to join multiple tables, but still want one of the tables to be uniqued, use the
        # array of strings format for your joins.
1025 1026 1027
        #
        #   class Article < ActiveRecord::Base
        #     def self.find_with_scope
1028 1029 1030
        #       with_scope(:find => where(:blog_id => 1).limit(1), :create => { :blog_id => 1 }) do
        #         with_scope(:find => limit(10)) do
        #           all # => SELECT * from articles WHERE blog_id = 1 LIMIT 10
1031
        #         end
1032 1033
        #         with_scope(:find => where(:author_id => 3)) do
        #           all # => SELECT * from articles WHERE blog_id = 1 AND author_id = 3 LIMIT 1
1034 1035 1036 1037 1038
        #         end
        #       end
        #     end
        #   end
        #
1039
        # You can ignore any previous scopings by using the <tt>with_exclusive_scope</tt> method.
1040 1041 1042
        #
        #   class Article < ActiveRecord::Base
        #     def self.find_with_exclusive_scope
1043
        #       with_scope(:find => where(:blog_id => 1).limit(1)) do
1044
        #         with_exclusive_scope(:find => limit(10)) do
1045
        #           all # => SELECT * from articles LIMIT 10
1046 1047 1048 1049
        #         end
        #       end
        #     end
        #   end
P
Pratik Naik 已提交
1050
        #
N
Neeraj Singh 已提交
1051
        # *Note*: the +:find+ scope also has effect on update and deletion methods, like +update_all+ and +delete_all+.
1052 1053 1054
        def with_scope(method_scoping = {}, action = :merge, &block)
          method_scoping = method_scoping.method_scoping if method_scoping.respond_to?(:method_scoping)

1055 1056
          if method_scoping.is_a?(Hash)
            # Dup first and second level of hash (method and params).
1057 1058 1059
            method_scoping = method_scoping.dup
            method_scoping.each do |method, params|
              method_scoping[method] = params.dup unless params == true
1060
            end
1061

1062 1063 1064 1065
            method_scoping.assert_valid_keys([ :find, :create ])
            relation = construct_finder_arel(method_scoping[:find] || {})

            if current_scoped_methods && current_scoped_methods.create_with_value && method_scoping[:create]
1066
              scope_for_create = if action == :merge
1067 1068 1069
                current_scoped_methods.create_with_value.merge(method_scoping[:create])
              else
                method_scoping[:create]
1070
              end
1071 1072 1073 1074 1075 1076 1077 1078 1079 1080 1081

              relation = relation.create_with(scope_for_create)
            else
              scope_for_create = method_scoping[:create]
              scope_for_create ||= current_scoped_methods.create_with_value if current_scoped_methods
              relation = relation.create_with(scope_for_create) if scope_for_create
            end

            method_scoping = relation
          end

1082
          method_scoping = current_scoped_methods.merge(method_scoping) if current_scoped_methods && action ==  :merge
1083 1084 1085 1086 1087 1088 1089 1090 1091 1092 1093

          self.scoped_methods << method_scoping
          begin
            yield
          ensure
            self.scoped_methods.pop
          end
        end

        # Works like with_scope, but discards any nested properties.
        def with_exclusive_scope(method_scoping = {}, &block)
1094
          if method_scoping.values.any? { |e| e.is_a?(ActiveRecord::Relation) }
1095 1096
            raise ArgumentError, <<-MSG
New finder API can not be used with_exclusive_scope. You can either call unscoped to get an anonymous scope not bound to the default_scope:
1097

1098 1099 1100 1101 1102 1103 1104 1105 1106
  User.unscoped.where(:active => true)

Or call unscoped with a block:

  User.unscoped do
    User.where(:active => true).all
  end

MSG
1107
          end
1108 1109 1110
          with_scope(method_scoping, :overwrite, &block)
        end

1111
        # Sets the default options for the model. The format of the
1112
        # <tt>options</tt> argument is the same as in find.
1113 1114
        #
        #   class Person < ActiveRecord::Base
1115
        #     default_scope order('last_name, first_name')
1116
        #   end
1117 1118 1119 1120 1121 1122 1123 1124
        #
        # <tt>default_scope</tt> is also applied while creating/building a record. It is not
        # applied while updating a record.
        #
        #   class Article < ActiveRecord::Base
        #     default_scope where(:published => true)
        #   end
        #
1125 1126
        #   Article.new.published    # => true
        #   Article.create.published # => true
1127
        def default_scope(options = {})
1128
          reset_scoped_methods
1129 1130 1131
          self.default_scoping << construct_finder_arel(options, default_scoping.pop)
        end

D
David Heinemeier Hansson 已提交
1132
        def current_scoped_methods #:nodoc:
T
Tim Morgan 已提交
1133 1134
          method = scoped_methods.last
          if method.respond_to?(:call)
1135
            relation.scoping { method.call }
T
Tim Morgan 已提交
1136 1137 1138
          else
            method
          end
1139
        end
1140

1141 1142 1143 1144
        def reset_scoped_methods #:nodoc:
          Thread.current[:"#{self}_scoped_methods"] = nil
        end

P
Pratik Naik 已提交
1145
        # Returns the class type of the record using the current module as a prefix. So descendants of
1146
        # MyApp::Business::Account would appear as MyApp::Business::AccountSubclass.
D
Initial  
David Heinemeier Hansson 已提交
1147
        def compute_type(type_name)
1148 1149 1150
          if type_name.match(/^::/)
            # If the type is prefixed with a scope operator then we assume that
            # the type_name is an absolute reference.
1151
            ActiveSupport::Dependencies.constantize(type_name)
1152 1153 1154 1155 1156 1157 1158 1159
          else
            # Build a list of candidates to search for
            candidates = []
            name.scan(/::|$/) { candidates.unshift "#{$`}::#{type_name}" }
            candidates << type_name

            candidates.each do |candidate|
              begin
1160
                constant = ActiveSupport::Dependencies.constantize(candidate)
1161
                return constant if candidate == constant.to_s
1162 1163 1164
              rescue NameError => e
                # We don't want to swallow NoMethodError < NameError errors
                raise e unless e.instance_of?(NameError)
1165 1166
              rescue ArgumentError
              end
1167
            end
1168 1169

            raise NameError, "uninitialized constant #{candidates.first}"
D
Initial  
David Heinemeier Hansson 已提交
1170 1171 1172
          end
        end

P
Pratik Naik 已提交
1173 1174
        # Returns the class descending directly from ActiveRecord::Base or an
        # abstract class, if any, in the inheritance hierarchy.
1175
        def class_of_active_record_descendant(klass)
1176
          if klass.superclass == Base || klass.superclass.abstract_class?
1177
            klass
D
Initial  
David Heinemeier Hansson 已提交
1178 1179 1180
          elsif klass.superclass.nil?
            raise ActiveRecordError, "#{name} doesn't belong in a hierarchy descending from ActiveRecord"
          else
1181
            class_of_active_record_descendant(klass.superclass)
D
Initial  
David Heinemeier Hansson 已提交
1182 1183 1184
          end
        end

P
Pratik Naik 已提交
1185
        # Accepts an array, hash, or string of SQL conditions and sanitizes
1186
        # them into a valid SQL fragment for a WHERE clause.
1187 1188 1189
        #   ["name='%s' and group_id='%s'", "foo'bar", 4]  returns  "name='foo''bar' and group_id='4'"
        #   { :name => "foo'bar", :group_id => 4 }  returns "name='foo''bar' and group_id='4'"
        #   "name='foo''bar' and group_id='4'" returns "name='foo''bar' and group_id='4'"
1190
        def sanitize_sql_for_conditions(condition, table_name = self.table_name)
1191 1192
          return nil if condition.blank?

1193 1194
          case condition
            when Array; sanitize_sql_array(condition)
1195
            when Hash;  sanitize_sql_hash_for_conditions(condition, table_name)
1196 1197
            else        condition
          end
1198
        end
1199
        alias_method :sanitize_sql, :sanitize_sql_for_conditions
1200

P
Pratik Naik 已提交
1201
        # Accepts an array, hash, or string of SQL conditions and sanitizes
1202 1203 1204 1205 1206 1207 1208 1209 1210 1211
        # them into a valid SQL fragment for a SET clause.
        #   { :name => nil, :group_id => 4 }  returns "name = NULL , group_id='4'"
        def sanitize_sql_for_assignment(assignments)
          case assignments
            when Array; sanitize_sql_array(assignments)
            when Hash;  sanitize_sql_hash_for_assignment(assignments)
            else        assignments
          end
        end

1212 1213 1214 1215 1216
        def aggregate_mapping(reflection)
          mapping = reflection.options[:mapping] || [reflection.name, reflection.name]
          mapping.first.is_a?(Array) ? mapping : [mapping]
        end

P
Pratik Naik 已提交
1217
        # Accepts a hash of SQL conditions and replaces those attributes
1218 1219 1220 1221 1222 1223 1224 1225 1226 1227 1228 1229 1230 1231 1232 1233 1234 1235 1236 1237 1238 1239 1240 1241 1242 1243 1244 1245 1246
        # that correspond to a +composed_of+ relationship with their expanded
        # aggregate attribute values.
        # Given:
        #     class Person < ActiveRecord::Base
        #       composed_of :address, :class_name => "Address",
        #         :mapping => [%w(address_street street), %w(address_city city)]
        #     end
        # Then:
        #     { :address => Address.new("813 abc st.", "chicago") }
        #       # => { :address_street => "813 abc st.", :address_city => "chicago" }
        def expand_hash_conditions_for_aggregates(attrs)
          expanded_attrs = {}
          attrs.each do |attr, value|
            unless (aggregation = reflect_on_aggregation(attr.to_sym)).nil?
              mapping = aggregate_mapping(aggregation)
              mapping.each do |field_attr, aggregate_attr|
                if mapping.size == 1 && !value.respond_to?(aggregate_attr)
                  expanded_attrs[field_attr] = value
                else
                  expanded_attrs[field_attr] = value.send(aggregate_attr)
                end
              end
            else
              expanded_attrs[attr] = value
            end
          end
          expanded_attrs
        end

1247
        # Sanitizes a hash of attribute/value pairs into SQL conditions for a WHERE clause.
1248 1249 1250 1251
        #   { :name => "foo'bar", :group_id => 4 }
        #     # => "name='foo''bar' and group_id= 4"
        #   { :status => nil, :group_id => [1,2,3] }
        #     # => "status IS NULL and group_id IN (1,2,3)"
1252 1253
        #   { :age => 13..18 }
        #     # => "age BETWEEN 13 AND 18"
1254 1255
        #   { 'other_records.id' => 7 }
        #     # => "`other_records`.`id` = 7"
1256 1257
        #   { :other_records => { :id => 7 } }
        #     # => "`other_records`.`id` = 7"
1258 1259 1260
        # And for value objects on a composed_of relationship:
        #   { :address => Address.new("123 abc st.", "chicago") }
        #     # => "address_street='123 abc st.' and address_city='chicago'"
1261
        def sanitize_sql_hash_for_conditions(attrs, default_table_name = self.table_name)
1262 1263
          attrs = expand_hash_conditions_for_aggregates(attrs)

1264
          table = Arel::Table.new(self.table_name, :engine => arel_engine, :as => default_table_name)
1265 1266 1267 1268
          viz = Arel::Visitors.for(arel_engine)
          PredicateBuilder.build_from_hash(arel_engine, attrs, table).map { |b|
            viz.accept b
          }.join(' AND ')
1269
        end
1270 1271 1272 1273 1274 1275
        alias_method :sanitize_sql_hash, :sanitize_sql_hash_for_conditions

        # Sanitizes a hash of attribute/value pairs into SQL conditions for a SET clause.
        #   { :status => nil, :group_id => 1 }
        #     # => "status = NULL , group_id = 1"
        def sanitize_sql_hash_for_assignment(attrs)
1276
          attrs.map do |attr, value|
1277 1278 1279
            "#{connection.quote_column_name(attr)} = #{quote_bound_value(value)}"
          end.join(', ')
        end
1280

1281
        # Accepts an array of conditions.  The array has each value
P
Pratik Naik 已提交
1282
        # sanitized and interpolated into the SQL statement.
1283
        #   ["name='%s' and group_id='%s'", "foo'bar", 4]  returns  "name='foo''bar' and group_id='4'"
1284
        def sanitize_sql_array(ary)
1285 1286 1287 1288
          statement, *values = ary
          if values.first.is_a?(Hash) and statement =~ /:\w+/
            replace_named_bind_variables(statement, values.first)
          elsif statement.include?('?')
1289
            replace_bind_variables(statement, values)
1290 1291
          elsif statement.blank?
            statement
1292
          else
1293
            statement % values.collect { |value| connection.quote_string(value.to_s) }
1294
          end
1295 1296
        end

1297 1298
        alias_method :sanitize_conditions, :sanitize_sql

D
David Heinemeier Hansson 已提交
1299
        def replace_bind_variables(statement, values) #:nodoc:
1300
          raise_if_bind_arity_mismatch(statement, statement.count('?'), values.size)
1301
          bound = values.dup
C
Comron Sattari 已提交
1302 1303
          c = connection
          statement.gsub('?') { quote_bound_value(bound.shift, c) }
1304 1305
        end

D
David Heinemeier Hansson 已提交
1306
        def replace_named_bind_variables(statement, bind_vars) #:nodoc:
1307 1308 1309 1310
          statement.gsub(/(:?):([a-zA-Z]\w*)/) do
            if $1 == ':' # skip postgresql casts
              $& # return the whole match
            elsif bind_vars.include?(match = $2.to_sym)
1311
              quote_bound_value(bind_vars[match])
1312 1313
            else
              raise PreparedStatementInvalid, "missing value for :#{match} in #{statement}"
1314 1315
            end
          end
1316 1317
        end

1318
        def expand_range_bind_variables(bind_vars) #:nodoc:
1319 1320 1321
          expanded = []

          bind_vars.each do |var|
1322 1323
            next if var.is_a?(Hash)

1324
            if var.is_a?(Range)
1325 1326
              expanded << var.first
              expanded << var.last
1327
            else
1328
              expanded << var
1329
            end
1330
          end
1331 1332

          expanded
1333 1334
        end

C
Comron Sattari 已提交
1335
        def quote_bound_value(value, c = connection) #:nodoc:
1336
          if value.respond_to?(:map) && !value.acts_like?(:string)
1337
            if value.respond_to?(:empty?) && value.empty?
C
Comron Sattari 已提交
1338
              c.quote(nil)
1339
            else
C
Comron Sattari 已提交
1340
              value.map { |v| c.quote(v) }.join(',')
1341
            end
1342
          else
C
Comron Sattari 已提交
1343
            c.quote(value)
1344 1345 1346
          end
        end

D
David Heinemeier Hansson 已提交
1347
        def raise_if_bind_arity_mismatch(statement, expected, provided) #:nodoc:
1348 1349 1350
          unless expected == provided
            raise PreparedStatementInvalid, "wrong number of bind variables (#{provided} for #{expected}) in: #{statement}"
          end
1351
        end
1352

D
David Heinemeier Hansson 已提交
1353
        def encode_quoted_value(value) #:nodoc:
1354
          quoted_value = connection.quote(value)
1355 1356
          quoted_value = "'#{quoted_value[1..-2].gsub(/\'/, "\\\\'")}'" if quoted_value.include?("\\\'") # (for ruby mode) "
          quoted_value
1357
        end
D
Initial  
David Heinemeier Hansson 已提交
1358 1359 1360 1361 1362
    end

    public
      # New objects can be instantiated as either empty (pass no construction parameter) or pre-set with
      # attributes but not yet saved (pass a hash with key names matching the associated table column names).
1363
      # In both instances, valid attribute keys are determined by the column names of the associated table --
D
Initial  
David Heinemeier Hansson 已提交
1364 1365
      # hence you can't have attributes that aren't part of the table columns.
      def initialize(attributes = nil)
1366
        @attributes = attributes_from_column_definition
1367
        @attributes_cache = {}
1368
        @persisted = false
1369
        @readonly = false
1370 1371
        @destroyed = false
        @marked_for_destruction = false
1372 1373
        @previously_changed = {}
        @changed_attributes = {}
1374

D
Initial  
David Heinemeier Hansson 已提交
1375
        ensure_proper_type
P
Pratik Naik 已提交
1376

1377
        populate_with_current_scope_attributes
1378
        self.attributes = attributes unless attributes.nil?
P
Pratik Naik 已提交
1379

1380
        result = yield self if block_given?
1381
        _run_initialize_callbacks
1382
        result
D
Initial  
David Heinemeier Hansson 已提交
1383
      end
1384

1385 1386 1387 1388
      # Cloned objects have no id assigned and are treated as new records. Note that this is a "shallow" clone
      # as it copies the object's attributes only, not its associations. The extent of a "deep" clone is
      # application specific and is therefore left to the application to implement according to its need.
      def initialize_copy(other)
1389
        _run_after_initialize_callbacks if respond_to?(:_run_after_initialize_callbacks)
1390 1391
        cloned_attributes = other.clone_attributes(:read_attribute_before_type_cast)
        cloned_attributes.delete(self.class.primary_key)
1392

1393
        @attributes = cloned_attributes
1394 1395 1396 1397 1398 1399

        @changed_attributes = {}
        attributes_from_column_definition.each do |attr, orig_value|
          @changed_attributes[attr] = orig_value if field_changed?(attr, orig_value, @attributes[attr])
        end

1400
        clear_aggregation_cache
1401
        clear_association_cache
1402
        @attributes_cache = {}
1403
        @persisted = false
1404
        ensure_proper_type
1405

1406
        populate_with_current_scope_attributes
1407 1408
      end

1409 1410 1411 1412 1413 1414 1415 1416 1417 1418 1419 1420 1421
      # Initialize an empty model object from +coder+.  +coder+ must contain
      # the attributes necessary for initializing an empty model object.  For
      # example:
      #
      #   class Post < ActiveRecord::Base
      #   end
      #
      #   post = Post.allocate
      #   post.init_with('attributes' => { 'title' => 'hello world' })
      #   post.title # => 'hello world'
      def init_with(coder)
        @attributes = coder['attributes']
        @attributes_cache, @previously_changed, @changed_attributes = {}, {}, {}
1422 1423
        @readonly = @destroyed = @marked_for_destruction = false
        @persisted = true
1424 1425 1426 1427
        _run_find_callbacks
        _run_initialize_callbacks
      end

P
Pratik Naik 已提交
1428 1429 1430 1431
      # Returns a String, which Action Pack uses for constructing an URL to this
      # object. The default implementation returns this record's id as a String,
      # or nil if this record's unsaved.
      #
P
Pratik Naik 已提交
1432
      # For example, suppose that you have a User model, and that you have a
1433
      # <tt>resources :users</tt> route. Normally, +user_path+ will
P
Pratik Naik 已提交
1434
      # construct a path with the user object's 'id' in it:
P
Pratik Naik 已提交
1435 1436
      #
      #   user = User.find_by_name('Phusion')
1437
      #   user_path(user)  # => "/users/1"
P
Pratik Naik 已提交
1438
      #
P
Pratik Naik 已提交
1439 1440
      # You can override +to_param+ in your model to make +user_path+ construct
      # a path using the user's name instead of the user's id:
P
Pratik Naik 已提交
1441 1442 1443 1444 1445 1446
      #
      #   class User < ActiveRecord::Base
      #     def to_param  # overridden
      #       name
      #     end
      #   end
E
Emilio Tagua 已提交
1447
      #
P
Pratik Naik 已提交
1448
      #   user = User.find_by_name('Phusion')
1449
      #   user_path(user)  # => "/users/Phusion"
1450
      def to_param
1451
        # We can't use alias_method here, because method 'id' optimizes itself on the fly.
1452
        id && id.to_s # Be sure to stringify the id for routes
1453
      end
1454

P
Pratik Naik 已提交
1455 1456 1457
      # Returns a cache key that can be used to identify this record.
      #
      # ==== Examples
1458 1459 1460 1461 1462
      #
      #   Product.new.cache_key     # => "products/new"
      #   Product.find(5).cache_key # => "products/5" (updated_at not available)
      #   Person.find(5).cache_key  # => "people/5-20071224150000" (updated_at available)
      def cache_key
1463
        case
1464
        when !persisted?
J
Jeremy Kemper 已提交
1465 1466 1467
          "#{self.class.model_name.cache_key}/new"
        when timestamp = self[:updated_at]
          "#{self.class.model_name.cache_key}/#{id}-#{timestamp.to_s(:number)}"
1468
        else
J
Jeremy Kemper 已提交
1469
          "#{self.class.model_name.cache_key}/#{id}"
1470 1471
        end
      end
1472

1473
      def quoted_id #:nodoc:
1474
        quote_value(id, column_for_attribute(self.class.primary_key))
1475
      end
1476

1477 1478 1479 1480 1481 1482 1483 1484 1485 1486
      # Returns true if the given attribute is in the attributes hash
      def has_attribute?(attr_name)
        @attributes.has_key?(attr_name.to_s)
      end

      # Returns an array of names for the attributes available on this object sorted alphabetically.
      def attribute_names
        @attributes.keys.sort
      end

1487
      # Returns the value of the attribute identified by <tt>attr_name</tt> after it has been typecast (for example,
D
Initial  
David Heinemeier Hansson 已提交
1488 1489
      # "2004-12-12" in a data column is cast to a date object, like Date.new(2004, 12, 12)).
      # (Alias for the protected read_attribute method).
1490
      def [](attr_name)
1491
        read_attribute(attr_name)
D
Initial  
David Heinemeier Hansson 已提交
1492
      end
1493

D
Initial  
David Heinemeier Hansson 已提交
1494 1495
      # Updates the attribute identified by <tt>attr_name</tt> with the specified +value+.
      # (Alias for the protected write_attribute method).
1496
      def []=(attr_name, value)
1497
        write_attribute(attr_name, value)
D
Initial  
David Heinemeier Hansson 已提交
1498 1499 1500
      end

      # Allows you to set all the attributes at once by passing in a hash with keys
P
Pratik Naik 已提交
1501 1502 1503 1504 1505 1506
      # matching the attribute names (which again matches the column names).
      #
      # If +guard_protected_attributes+ is true (the default), then sensitive
      # attributes can be protected from this form of mass-assignment by using
      # the +attr_protected+ macro. Or you can alternatively specify which
      # attributes *can* be accessed with the +attr_accessible+ macro. Then all the
1507
      # attributes not included in that won't be allowed to be mass-assigned.
P
Pratik Naik 已提交
1508 1509 1510 1511
      #
      #   class User < ActiveRecord::Base
      #     attr_protected :is_admin
      #   end
E
Emilio Tagua 已提交
1512
      #
P
Pratik Naik 已提交
1513 1514 1515 1516
      #   user = User.new
      #   user.attributes = { :username => 'Phusion', :is_admin => true }
      #   user.username   # => "Phusion"
      #   user.is_admin?  # => false
E
Emilio Tagua 已提交
1517
      #
P
Pratik Naik 已提交
1518 1519
      #   user.send(:attributes=, { :username => 'Phusion', :is_admin => true }, false)
      #   user.is_admin?  # => true
1520
      def attributes=(new_attributes, guard_protected_attributes = true)
P
Pratik Naik 已提交
1521
        return unless new_attributes.is_a?(Hash)
1522
        attributes = new_attributes.stringify_keys
D
Initial  
David Heinemeier Hansson 已提交
1523 1524

        multi_parameter_attributes = []
1525
        attributes = sanitize_for_mass_assignment(attributes) if guard_protected_attributes
1526

1527
        attributes.each do |k, v|
1528 1529 1530 1531 1532
          if k.include?("(")
            multi_parameter_attributes << [ k, v ]
          else
            respond_to?(:"#{k}=") ? send(:"#{k}=", v) : raise(UnknownAttributeError, "unknown attribute: #{k}")
          end
D
Initial  
David Heinemeier Hansson 已提交
1533
        end
D
David Heinemeier Hansson 已提交
1534

D
Initial  
David Heinemeier Hansson 已提交
1535 1536 1537
        assign_multiparameter_attributes(multi_parameter_attributes)
      end

1538
      # Returns a hash of all the attributes with their names as keys and the values of the attributes as values.
1539
      def attributes
1540 1541 1542
        attrs = {}
        attribute_names.each { |name| attrs[name] = read_attribute(name) }
        attrs
1543 1544
      end

P
Pratik Naik 已提交
1545 1546 1547 1548 1549 1550 1551 1552 1553 1554 1555 1556 1557
      # Returns an <tt>#inspect</tt>-like string for the value of the
      # attribute +attr_name+. String attributes are elided after 50
      # characters, and Date and Time attributes are returned in the
      # <tt>:db</tt> format. Other attributes return the value of
      # <tt>#inspect</tt> without modification.
      #
      #   person = Person.create!(:name => "David Heinemeier Hansson " * 3)
      #
      #   person.attribute_for_inspect(:name)
      #   # => '"David Heinemeier Hansson David Heinemeier Hansson D..."'
      #
      #   person.attribute_for_inspect(:created_at)
      #   # => '"2009-01-12 04:48:57"'
1558 1559 1560
      def attribute_for_inspect(attr_name)
        value = read_attribute(attr_name)

1561
        if value.is_a?(String) && value.length > 50
1562
          "#{value[0..50]}...".inspect
1563 1564
        elsif value.is_a?(Date) || value.is_a?(Time)
          %("#{value.to_s(:db)}")
1565 1566 1567 1568 1569
        else
          value.inspect
        end
      end

D
Initial  
David Heinemeier Hansson 已提交
1570
      # Returns true if the specified +attribute+ has been set by the user or by a database load and is neither
1571
      # nil nor empty? (the latter only applies to objects that respond to empty?, most notably Strings).
D
Initial  
David Heinemeier Hansson 已提交
1572
      def attribute_present?(attribute)
1573
        value = read_attribute(attribute)
1574
        !value.blank?
D
Initial  
David Heinemeier Hansson 已提交
1575 1576 1577 1578
      end

      # Returns the column object for the named attribute.
      def column_for_attribute(name)
1579
        self.class.columns_hash[name.to_s]
D
Initial  
David Heinemeier Hansson 已提交
1580
      end
1581

X
Xavier Noria 已提交
1582 1583 1584 1585 1586 1587 1588 1589 1590
      # Returns true if +comparison_object+ is the same exact object, or +comparison_object+ 
      # is of the same type and +self+ has an ID and it is equal to +comparison_object.id+.
      #
      # Note that new records are different from any other record by definition, unless the
      # other record is the receiver itself. Besides, if you fetch existing records with
      # +select+ and leave the ID out, you're on your own, this predicate will return false.
      #
      # Note also that destroying a record preserves its ID in the model instance, so deleted
      # models are still comparable.
D
Initial  
David Heinemeier Hansson 已提交
1591
      def ==(comparison_object)
1592
        comparison_object.equal?(self) ||
1593 1594 1595
          comparison_object.instance_of?(self.class) &&
          id.present? &&
          comparison_object.id == id
D
Initial  
David Heinemeier Hansson 已提交
1596 1597 1598 1599 1600 1601
      end

      # Delegates to ==
      def eql?(comparison_object)
        self == (comparison_object)
      end
1602

D
Initial  
David Heinemeier Hansson 已提交
1603 1604 1605
      # Delegates to id in order to allow two records of the same type and id to work with something like:
      #   [ Person.find(1), Person.find(2), Person.find(3) ] & [ Person.find(1), Person.find(4) ] # => [ Person.find(1) ]
      def hash
1606
        id.hash
D
Initial  
David Heinemeier Hansson 已提交
1607 1608
      end

1609
      # Freeze the attributes hash such that associations are still accessible, even on destroyed records.
1610
      def freeze
1611
        @attributes.freeze; self
1612
      end
1613

1614
      # Returns +true+ if the attributes hash has been frozen.
1615 1616 1617
      def frozen?
        @attributes.frozen?
      end
1618

1619 1620 1621
      # Returns duplicated record with unfreezed attributes.
      def dup
        obj = super
1622
        obj.instance_variable_set('@attributes', @attributes.dup)
1623 1624 1625
        obj
      end

1626 1627
      # Returns +true+ if the record is read only. Records loaded through joins with piggy-back
      # attributes will be marked as read only since they cannot be saved.
1628
      def readonly?
1629
        @readonly
1630 1631
      end

1632 1633
      # Marks this record as read only.
      def readonly!
1634 1635
        @readonly = true
      end
1636

1637
      # Returns the contents of the record as a nicely formatted string.
1638
      def inspect
1639
        attributes_as_nice_string = self.class.column_names.collect { |name|
1640
          if has_attribute?(name) || !persisted?
1641 1642 1643
            "#{name}: #{attribute_for_inspect(name)}"
          end
        }.compact.join(", ")
1644
        "#<#{self.class} #{attributes_as_nice_string}>"
1645
      end
1646

1647 1648
    protected
      def clone_attributes(reader_method = :read_attribute, attributes = {})
1649 1650
        attribute_names.each do |name|
          attributes[name] = clone_attribute_value(reader_method, name)
1651
        end
1652
        attributes
1653 1654 1655 1656 1657 1658 1659 1660 1661
      end

      def clone_attribute_value(reader_method, attribute_name)
        value = send(reader_method, attribute_name)
        value.duplicable? ? value.clone : value
      rescue TypeError, NoMethodError
        value
      end

D
Initial  
David Heinemeier Hansson 已提交
1662 1663
    private

1664
      # Sets the attribute used for single table inheritance to this class name if this is not the
1665
      # ActiveRecord::Base descendant.
1666 1667
      # Considering the hierarchy Reply < Message < ActiveRecord::Base, this makes it possible to
      # do Reply.new without having to set <tt>Reply[Reply.inheritance_column] = "Reply"</tt> yourself.
1668
      # No such attribute would be set for objects of the Message class in that example.
D
Initial  
David Heinemeier Hansson 已提交
1669 1670
      def ensure_proper_type
        unless self.class.descends_from_active_record?
1671
          write_attribute(self.class.inheritance_column, self.class.sti_name)
D
Initial  
David Heinemeier Hansson 已提交
1672 1673 1674
        end
      end

1675
      # The primary key and inheritance column can never be set by mass-assignment for security reasons.
1676 1677 1678
      def self.attributes_protected_by_default
        default = [ primary_key, inheritance_column ]
        default << 'id' unless primary_key.eql? 'id'
1679
        default
1680 1681
      end

1682 1683
      # Returns a copy of the attributes hash where all the values have been safely quoted for use in
      # an Arel insert/update method.
1684 1685 1686 1687 1688 1689
      def arel_attributes_values(include_primary_key = true, include_readonly_attributes = true, attribute_names = @attributes.keys)
        attrs = {}
        attribute_names.each do |name|
          if (column = column_for_attribute(name)) && (include_primary_key || !column.primary)

            if include_readonly_attributes || (!include_readonly_attributes && !self.class.readonly_attributes.include?(name))
1690 1691
              value = read_attribute(name)

1692 1693
              if value && self.class.serialized_attributes.key?(name)
                value = YAML.dump value
1694
              end
1695
              attrs[self.class.arel_table[name]] = value
1696 1697 1698 1699 1700 1701
            end
          end
        end
        attrs
      end

D
Initial  
David Heinemeier Hansson 已提交
1702
      # Quote strings appropriately for SQL statements.
1703
      def quote_value(value, column = nil)
1704
        self.class.connection.quote(value, column)
D
Initial  
David Heinemeier Hansson 已提交
1705 1706
      end

P
Pratik Naik 已提交
1707
      # Interpolate custom SQL string in instance context.
D
Initial  
David Heinemeier Hansson 已提交
1708 1709
      # Optional record argument is meant for custom insert_sql.
      def interpolate_sql(sql, record = nil)
1710
        instance_eval("%@#{sql.gsub('@', '\@')}@", __FILE__, __LINE__)
D
Initial  
David Heinemeier Hansson 已提交
1711 1712 1713 1714 1715 1716
      end

      # Instantiates objects for all attribute classes that needs more than one constructor parameter. This is done
      # by calling new on the column type or aggregation type (through composed_of) object with these parameters.
      # So having the pairs written_on(1) = "2004", written_on(2) = "6", written_on(3) = "24", will instantiate
      # written_on (a date type) with Date.new("2004", "6", "24"). You can also specify a typecast character in the
1717 1718
      # parentheses to have the parameters typecasted before they're used in the constructor. Use i for Fixnum,
      # f for Float, s for String, and a for Array. If all the values for a given attribute are empty, the
1719
      # attribute will be set to nil.
D
Initial  
David Heinemeier Hansson 已提交
1720 1721 1722 1723 1724
      def assign_multiparameter_attributes(pairs)
        execute_callstack_for_multiparameter_attributes(
          extract_callstack_for_multiparameter_attributes(pairs)
        )
      end
1725

1726
      def instantiate_time_object(name, values)
1727
        if self.class.send(:create_time_zone_conversion_attribute?, name, column_for_attribute(name))
1728
          Time.zone.local(*values)
1729
        else
1730
          Time.time_with_datetime_fallback(@@default_timezone, *values)
1731
        end
1732 1733
      end

D
Initial  
David Heinemeier Hansson 已提交
1734
      def execute_callstack_for_multiparameter_attributes(callstack)
1735
        errors = []
1736
        callstack.each do |name, values_with_empty_parameters|
1737 1738
          begin
            klass = (self.class.reflect_on_aggregation(name.to_sym) || column_for_attribute(name)).klass
1739 1740
            # in order to allow a date to be set without a year, we must keep the empty values.
            # Otherwise, we wouldn't be able to distinguish it from a date with an empty day.
1741
            values = values_with_empty_parameters.reject { |v| v.nil? }
1742

1743 1744 1745
            if values.empty?
              send(name + "=", nil)
            else
1746

1747
              value = if Time == klass
1748
                instantiate_time_object(name, values)
1749 1750
              elsif Date == klass
                begin
1751
                  values = values_with_empty_parameters.collect do |v| v.nil? ? 1 : v end
1752 1753
                  Date.new(*values)
                rescue ArgumentError => ex # if Date.new raises an exception on an invalid date
1754
                  instantiate_time_object(name, values).to_date # we instantiate Time object and convert it back to a date thus using Time's logic in handling invalid dates
1755 1756 1757 1758 1759 1760
                end
              else
                klass.new(*values)
              end

              send(name + "=", value)
1761
            end
1762 1763
          rescue => ex
            errors << AttributeAssignmentError.new("error on assignment #{values.inspect} to #{name}", ex, name)
D
Initial  
David Heinemeier Hansson 已提交
1764 1765
          end
        end
1766 1767 1768
        unless errors.empty?
          raise MultiparameterAssignmentErrors.new(errors), "#{errors.size} error(s) on assignment of multiparameter attributes"
        end
D
Initial  
David Heinemeier Hansson 已提交
1769
      end
1770

D
Initial  
David Heinemeier Hansson 已提交
1771 1772 1773 1774 1775 1776 1777 1778
      def extract_callstack_for_multiparameter_attributes(pairs)
        attributes = { }

        for pair in pairs
          multiparameter_name, value = pair
          attribute_name = multiparameter_name.split("(").first
          attributes[attribute_name] = [] unless attributes.include?(attribute_name)

1779 1780
          parameter_value = value.empty? ? nil : type_cast_attribute_value(multiparameter_name, value)
          attributes[attribute_name] << [ find_parameter_position(multiparameter_name), parameter_value ]
D
Initial  
David Heinemeier Hansson 已提交
1781 1782 1783 1784
        end

        attributes.each { |name, values| attributes[name] = values.sort_by{ |v| v.first }.collect { |v| v.last } }
      end
1785

D
Initial  
David Heinemeier Hansson 已提交
1786
      def type_cast_attribute_value(multiparameter_name, value)
1787
        multiparameter_name =~ /\([0-9]*([if])\)/ ? value.send("to_" + $1) : value
D
Initial  
David Heinemeier Hansson 已提交
1788
      end
1789

D
Initial  
David Heinemeier Hansson 已提交
1790 1791 1792
      def find_parameter_position(multiparameter_name)
        multiparameter_name.scan(/\(([0-9]*).*\)/).first.first
      end
1793

D
Initial  
David Heinemeier Hansson 已提交
1794 1795
      # Returns a comma-separated pair list, like "key1 = val1, key2 = val2".
      def comma_pair_list(hash)
1796
        hash.map { |k,v| "#{k} = #{v}" }.join(", ")
D
Initial  
David Heinemeier Hansson 已提交
1797 1798
      end

1799
      def quote_columns(quoter, hash)
1800
        Hash[hash.map { |name, value| [quoter.quote_column_name(name), value] }]
D
Initial  
David Heinemeier Hansson 已提交
1801 1802
      end

1803 1804
      def quoted_comma_pair_list(quoter, hash)
        comma_pair_list(quote_columns(quoter, hash))
D
Initial  
David Heinemeier Hansson 已提交
1805 1806
      end

1807 1808 1809 1810 1811 1812 1813 1814 1815 1816 1817 1818 1819 1820 1821 1822
      def convert_number_column_value(value)
        if value == false
          0
        elsif value == true
          1
        elsif value.is_a?(String) && value.blank?
          nil
        else
          value
        end
      end

      def object_from_yaml(string)
        return string unless string.is_a?(String) && string =~ /^---/
        YAML::load(string) rescue string
      end
1823 1824 1825 1826 1827 1828 1829

      def populate_with_current_scope_attributes
        if scope = self.class.send(:current_scoped_methods)
          create_with = scope.scope_for_create
          create_with.each { |att,value| self.respond_to?(:"#{att}=") && self.send("#{att}=", value) } if create_with
        end
      end
D
Initial  
David Heinemeier Hansson 已提交
1830
  end
J
Joshua Peek 已提交
1831 1832

  Base.class_eval do
1833
    include ActiveRecord::Persistence
J
Joshua Peek 已提交
1834
    extend ActiveModel::Naming
J
Joshua Peek 已提交
1835
    extend QueryCache::ClassMethods
J
José Valim 已提交
1836
    extend ActiveSupport::Benchmarkable
1837
    extend ActiveSupport::DescendantsTracker
J
José Valim 已提交
1838

1839
    include ActiveModel::Conversion
J
Joshua Peek 已提交
1840
    include Validations
1841
    extend CounterCache
J
Joshua Peek 已提交
1842 1843
    include Locking::Optimistic, Locking::Pessimistic
    include AttributeMethods
J
Joshua Peek 已提交
1844
    include AttributeMethods::Read, AttributeMethods::Write, AttributeMethods::BeforeTypeCast, AttributeMethods::Query
1845
    include AttributeMethods::PrimaryKey
J
Joshua Peek 已提交
1846 1847
    include AttributeMethods::TimeZoneConversion
    include AttributeMethods::Dirty
1848
    include ActiveModel::MassAssignmentSecurity
1849
    include Callbacks, ActiveModel::Observing, Timestamp
J
Joshua Peek 已提交
1850
    include Associations, AssociationPreload, NamedScope
1851 1852 1853 1854

    # AutosaveAssociation needs to be included before Transactions, because we want
    # #save_with_autosave_associations to be wrapped inside a transaction.
    include AutosaveAssociation, NestedAttributes
P
Pratik Naik 已提交
1855
    include Aggregations, Transactions, Reflection, Serialization
1856

1857
    NilClass.add_whiner(self) if NilClass.respond_to?(:add_whiner)
J
Joshua Peek 已提交
1858
  end
1859
end
1860 1861 1862

# TODO: Remove this and make it work with LAZY flag
require 'active_record/connection_adapters/abstract_adapter'
1863
ActiveSupport.run_load_hooks(:active_record, ActiveRecord::Base)