base.rb 78.3 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]
E
Emilio Tagua 已提交
739
        while klass != klass.base_class
740 741 742 743 744
          classes << klass = klass.superclass
        end
        classes
      rescue
        # OPTIMIZE this rescue is to fix this test: ./test/cases/reflection_test.rb:56:in `test_human_name_for_column'
P
Pratik Naik 已提交
745
        # Apparently the method base_class causes some trouble.
746 747 748 749
        # It now works for sure.
        [self]
      end

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

755 756 757 758 759 760 761
      # 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 已提交
762 763
      end

764 765 766 767 768
      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

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

783
      def quote_value(value, column = nil) #:nodoc:
784
        connection.quote(value,column)
785 786
      end

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

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

797 798 799
      # 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.
800 801 802
      #
      # 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.
803 804 805 806
      def base_class
        class_of_active_record_descendant(self)
      end

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

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

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

822 823 824
        super
      end

825 826 827 828
      def sti_name
        store_full_sti_class ? name : name.demodulize
      end

P
Pratik Naik 已提交
829
      def arel_table
830
        @arel_table ||= Arel::Table.new(table_name, arel_engine)
831 832
      end

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

843 844 845 846 847 848 849 850 851 852 853 854 855 856 857 858
      # 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"
      #   }
      #
859 860 861
      # 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.
      #
862 863
      # Post.unscoped.published
      # Post.published
864
      def unscoped #:nodoc:
865 866 867 868 869 870 871 872
        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

873 874 875 876
      def before_remove_const #:nodoc:
        reset_scoped_methods
      end

D
Initial  
David Heinemeier Hansson 已提交
877
      private
J
José Valim 已提交
878 879 880 881 882 883

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

884 885 886
        # 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 已提交
887
        def instantiate(record)
888 889 890
          model = find_sti_class(record[inheritance_column]).allocate
          model.init_with('attributes' => record)
          model
D
Initial  
David Heinemeier Hansson 已提交
891
        end
892

893
        def find_sti_class(type_name)
894
          if type_name.blank? || !columns_hash.include?(inheritance_column)
895 896 897
            self
          else
            begin
898 899 900 901 902
              if store_full_sti_class
                ActiveSupport::Dependencies.constantize(type_name)
              else
                compute_type(type_name)
              end
903 904 905 906 907 908 909 910 911 912
            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

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

919
        def type_condition
920
          sti_column = arel_table[inheritance_column]
921
          condition = sti_column.eq(sti_name)
922
          descendants.each { |subclass| condition = condition.or(sti_column.eq(subclass.sti_name)) }
923

924
          condition
D
Initial  
David Heinemeier Hansson 已提交
925 926 927
        end

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

934 935 936 937 938 939 940 941 942 943 944 945 946 947 948 949 950
        # 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

951
        # Enables dynamic finders like <tt>User.find_by_user_name(user_name)</tt> and
952 953
        # <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.
954
        #
955
        # It's even possible to use all the additional parameters to +find+. For example, the
956
        # full interface for +find_all_by_amount+ is actually <tt>find_all_by_amount(amount, options)</tt>.
957
        #
958
        # Each dynamic finder using <tt>scoped_by_*</tt> is also defined in the class after it
959
        # is first invoked, so that future attempts to use it do not run through method_missing.
960
        def method_missing(method_id, *arguments, &block)
961 962
          if match = DynamicFinderMatch.match(method_id)
            attribute_names = match.attribute_names
963
            super unless all_attributes_exists?(attribute_names)
964
            if match.finder?
965
              options = arguments.extract_options!
966
              relation = options.any? ? construct_finder_arel(options, current_scoped_methods) : scoped
967
              relation.send :find_by_attributes, match, attribute_names, *arguments
968
            elsif match.instantiator?
969
              scoped.send :find_or_instantiator_by_attributes, match, attribute_names, *arguments, &block
970
            end
971 972 973 974
          elsif match = DynamicScopeMatch.match(method_id)
            attribute_names = match.attribute_names
            super unless all_attributes_exists?(attribute_names)
            if match.scope?
975
              self.class_eval <<-METHOD, __FILE__, __LINE__ + 1
976 977 978 979 980
                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
981
              METHOD
982 983
              send(method_id, *arguments)
            end
984 985 986 987
          else
            super
          end
        end
D
Initial  
David Heinemeier Hansson 已提交
988

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

1002
        def all_attributes_exists?(attribute_names)
1003 1004
          (expand_attribute_names_for_aggregates(attribute_names) -
           column_methods_hash.keys).empty?
1005
        end
1006

D
Initial  
David Heinemeier Hansson 已提交
1007
      protected
N
Neeraj Singh 已提交
1008 1009
        # 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
1010
        # <tt>:create</tt> parameters are an attributes hash.
1011 1012 1013
        #
        #   class Article < ActiveRecord::Base
        #     def self.create_with_scope
1014
        #       with_scope(:find => where(:blog_id => 1), :create => { :blog_id => 1 }) do
1015 1016 1017 1018 1019 1020 1021
        #         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 已提交
1022
        # In nested scopings, all previous parameters are overwritten by the innermost rule, with the exception of
1023
        # <tt>where</tt>, <tt>includes</tt>, and <tt>joins</tt> operations in <tt>Relation</tt>, which are merged.
P
Pratik Naik 已提交
1024
        #
1025
        # <tt>joins</tt> operations are uniqued so multiple scopes can join in the same table without table aliasing
P
Pratik Naik 已提交
1026 1027
        # 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.
1028 1029 1030
        #
        #   class Article < ActiveRecord::Base
        #     def self.find_with_scope
1031 1032 1033
        #       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
1034
        #         end
1035 1036
        #         with_scope(:find => where(:author_id => 3)) do
        #           all # => SELECT * from articles WHERE blog_id = 1 AND author_id = 3 LIMIT 1
1037 1038 1039 1040 1041
        #         end
        #       end
        #     end
        #   end
        #
1042
        # You can ignore any previous scopings by using the <tt>with_exclusive_scope</tt> method.
1043 1044 1045
        #
        #   class Article < ActiveRecord::Base
        #     def self.find_with_exclusive_scope
1046
        #       with_scope(:find => where(:blog_id => 1).limit(1)) do
1047
        #         with_exclusive_scope(:find => limit(10)) do
1048
        #           all # => SELECT * from articles LIMIT 10
1049 1050 1051 1052
        #         end
        #       end
        #     end
        #   end
P
Pratik Naik 已提交
1053
        #
N
Neeraj Singh 已提交
1054
        # *Note*: the +:find+ scope also has effect on update and deletion methods, like +update_all+ and +delete_all+.
1055 1056 1057
        def with_scope(method_scoping = {}, action = :merge, &block)
          method_scoping = method_scoping.method_scoping if method_scoping.respond_to?(:method_scoping)

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

1065 1066 1067 1068
            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]
1069
              scope_for_create = if action == :merge
1070 1071 1072
                current_scoped_methods.create_with_value.merge(method_scoping[:create])
              else
                method_scoping[:create]
1073
              end
1074 1075 1076 1077 1078 1079 1080 1081 1082 1083 1084

              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

1085
          method_scoping = current_scoped_methods.merge(method_scoping) if current_scoped_methods && action ==  :merge
1086 1087 1088 1089 1090 1091 1092 1093 1094 1095 1096

          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)
1097
          if method_scoping.values.any? { |e| e.is_a?(ActiveRecord::Relation) }
1098 1099
            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:
1100

1101 1102 1103 1104 1105 1106 1107 1108 1109
  User.unscoped.where(:active => true)

Or call unscoped with a block:

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

MSG
1110
          end
1111 1112 1113
          with_scope(method_scoping, :overwrite, &block)
        end

1114
        # Sets the default options for the model. The format of the
1115
        # <tt>options</tt> argument is the same as in find.
1116 1117
        #
        #   class Person < ActiveRecord::Base
1118
        #     default_scope order('last_name, first_name')
1119
        #   end
1120 1121 1122 1123 1124 1125 1126 1127
        #
        # <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
        #
1128 1129
        #   Article.new.published    # => true
        #   Article.create.published # => true
1130
        def default_scope(options = {})
1131
          reset_scoped_methods
1132 1133 1134
          self.default_scoping << construct_finder_arel(options, default_scoping.pop)
        end

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

1144 1145 1146 1147
        def reset_scoped_methods #:nodoc:
          Thread.current[:"#{self}_scoped_methods"] = nil
        end

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

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

            raise NameError, "uninitialized constant #{candidates.first}"
D
Initial  
David Heinemeier Hansson 已提交
1173 1174 1175
          end
        end

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

P
Pratik Naik 已提交
1188
        # Accepts an array, hash, or string of SQL conditions and sanitizes
1189
        # them into a valid SQL fragment for a WHERE clause.
1190 1191 1192
        #   ["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'"
1193
        def sanitize_sql_for_conditions(condition, table_name = self.table_name)
1194 1195
          return nil if condition.blank?

1196 1197
          case condition
            when Array; sanitize_sql_array(condition)
1198
            when Hash;  sanitize_sql_hash_for_conditions(condition, table_name)
1199 1200
            else        condition
          end
1201
        end
1202
        alias_method :sanitize_sql, :sanitize_sql_for_conditions
1203

P
Pratik Naik 已提交
1204
        # Accepts an array, hash, or string of SQL conditions and sanitizes
1205 1206 1207 1208 1209 1210 1211 1212 1213 1214
        # 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

1215 1216 1217 1218 1219
        def aggregate_mapping(reflection)
          mapping = reflection.options[:mapping] || [reflection.name, reflection.name]
          mapping.first.is_a?(Array) ? mapping : [mapping]
        end

P
Pratik Naik 已提交
1220
        # Accepts a hash of SQL conditions and replaces those attributes
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 1247 1248 1249
        # 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

1250
        # Sanitizes a hash of attribute/value pairs into SQL conditions for a WHERE clause.
1251 1252 1253 1254
        #   { :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)"
1255 1256
        #   { :age => 13..18 }
        #     # => "age BETWEEN 13 AND 18"
1257 1258
        #   { 'other_records.id' => 7 }
        #     # => "`other_records`.`id` = 7"
1259 1260
        #   { :other_records => { :id => 7 } }
        #     # => "`other_records`.`id` = 7"
1261 1262 1263
        # 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'"
1264
        def sanitize_sql_hash_for_conditions(attrs, default_table_name = self.table_name)
1265 1266
          attrs = expand_hash_conditions_for_aggregates(attrs)

1267
          table = Arel::Table.new(self.table_name, :engine => arel_engine, :as => default_table_name)
1268 1269 1270 1271
          viz = Arel::Visitors.for(arel_engine)
          PredicateBuilder.build_from_hash(arel_engine, attrs, table).map { |b|
            viz.accept b
          }.join(' AND ')
1272
        end
1273 1274 1275 1276 1277 1278
        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)
1279
          attrs.map do |attr, value|
1280 1281 1282
            "#{connection.quote_column_name(attr)} = #{quote_bound_value(value)}"
          end.join(', ')
        end
1283

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

1300 1301
        alias_method :sanitize_conditions, :sanitize_sql

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

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

1321
        def expand_range_bind_variables(bind_vars) #:nodoc:
1322 1323 1324
          expanded = []

          bind_vars.each do |var|
1325 1326
            next if var.is_a?(Hash)

1327
            if var.is_a?(Range)
1328 1329
              expanded << var.first
              expanded << var.last
1330
            else
1331
              expanded << var
1332
            end
1333
          end
1334 1335

          expanded
1336 1337
        end

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

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

D
David Heinemeier Hansson 已提交
1356
        def encode_quoted_value(value) #:nodoc:
1357
          quoted_value = connection.quote(value)
1358 1359
          quoted_value = "'#{quoted_value[1..-2].gsub(/\'/, "\\\\'")}'" if quoted_value.include?("\\\'") # (for ruby mode) "
          quoted_value
1360
        end
D
Initial  
David Heinemeier Hansson 已提交
1361 1362 1363 1364 1365
    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).
1366
      # In both instances, valid attribute keys are determined by the column names of the associated table --
D
Initial  
David Heinemeier Hansson 已提交
1367 1368
      # hence you can't have attributes that aren't part of the table columns.
      def initialize(attributes = nil)
1369
        @attributes = attributes_from_column_definition
1370
        @attributes_cache = {}
1371
        @persisted = false
1372
        @readonly = false
1373 1374
        @destroyed = false
        @marked_for_destruction = false
1375 1376
        @previously_changed = {}
        @changed_attributes = {}
1377

D
Initial  
David Heinemeier Hansson 已提交
1378
        ensure_proper_type
P
Pratik Naik 已提交
1379

1380
        populate_with_current_scope_attributes
1381
        self.attributes = attributes unless attributes.nil?
P
Pratik Naik 已提交
1382

1383
        result = yield self if block_given?
1384
        _run_initialize_callbacks
1385
        result
D
Initial  
David Heinemeier Hansson 已提交
1386
      end
1387

1388 1389 1390 1391
      # 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)
1392
        _run_after_initialize_callbacks if respond_to?(:_run_after_initialize_callbacks)
1393 1394
        cloned_attributes = other.clone_attributes(:read_attribute_before_type_cast)
        cloned_attributes.delete(self.class.primary_key)
1395

1396
        @attributes = cloned_attributes
1397 1398 1399 1400 1401 1402

        @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

1403
        clear_aggregation_cache
1404
        clear_association_cache
1405
        @attributes_cache = {}
1406
        @persisted = false
1407
        ensure_proper_type
1408

1409
        populate_with_current_scope_attributes
1410 1411
      end

1412 1413 1414 1415 1416 1417 1418 1419 1420 1421 1422 1423 1424
      # 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 = {}, {}, {}
1425 1426
        @readonly = @destroyed = @marked_for_destruction = false
        @persisted = true
1427 1428 1429 1430
        _run_find_callbacks
        _run_initialize_callbacks
      end

P
Pratik Naik 已提交
1431 1432 1433 1434
      # 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 已提交
1435
      # For example, suppose that you have a User model, and that you have a
1436
      # <tt>resources :users</tt> route. Normally, +user_path+ will
P
Pratik Naik 已提交
1437
      # construct a path with the user object's 'id' in it:
P
Pratik Naik 已提交
1438 1439
      #
      #   user = User.find_by_name('Phusion')
1440
      #   user_path(user)  # => "/users/1"
P
Pratik Naik 已提交
1441
      #
P
Pratik Naik 已提交
1442 1443
      # 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 已提交
1444 1445 1446 1447 1448 1449
      #
      #   class User < ActiveRecord::Base
      #     def to_param  # overridden
      #       name
      #     end
      #   end
E
Emilio Tagua 已提交
1450
      #
P
Pratik Naik 已提交
1451
      #   user = User.find_by_name('Phusion')
1452
      #   user_path(user)  # => "/users/Phusion"
1453
      def to_param
1454
        # We can't use alias_method here, because method 'id' optimizes itself on the fly.
1455
        id && id.to_s # Be sure to stringify the id for routes
1456
      end
1457

P
Pratik Naik 已提交
1458 1459 1460
      # Returns a cache key that can be used to identify this record.
      #
      # ==== Examples
1461 1462 1463 1464 1465
      #
      #   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
1466
        case
1467
        when !persisted?
J
Jeremy Kemper 已提交
1468 1469 1470
          "#{self.class.model_name.cache_key}/new"
        when timestamp = self[:updated_at]
          "#{self.class.model_name.cache_key}/#{id}-#{timestamp.to_s(:number)}"
1471
        else
J
Jeremy Kemper 已提交
1472
          "#{self.class.model_name.cache_key}/#{id}"
1473 1474
        end
      end
1475

1476
      def quoted_id #:nodoc:
1477
        quote_value(id, column_for_attribute(self.class.primary_key))
1478
      end
1479

1480 1481 1482 1483 1484 1485 1486 1487 1488 1489
      # 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

1490
      # Returns the value of the attribute identified by <tt>attr_name</tt> after it has been typecast (for example,
D
Initial  
David Heinemeier Hansson 已提交
1491 1492
      # "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).
1493
      def [](attr_name)
1494
        read_attribute(attr_name)
D
Initial  
David Heinemeier Hansson 已提交
1495
      end
1496

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

      # Allows you to set all the attributes at once by passing in a hash with keys
P
Pratik Naik 已提交
1504 1505 1506 1507 1508 1509
      # 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
1510
      # attributes not included in that won't be allowed to be mass-assigned.
P
Pratik Naik 已提交
1511 1512 1513 1514
      #
      #   class User < ActiveRecord::Base
      #     attr_protected :is_admin
      #   end
E
Emilio Tagua 已提交
1515
      #
P
Pratik Naik 已提交
1516 1517 1518 1519
      #   user = User.new
      #   user.attributes = { :username => 'Phusion', :is_admin => true }
      #   user.username   # => "Phusion"
      #   user.is_admin?  # => false
E
Emilio Tagua 已提交
1520
      #
P
Pratik Naik 已提交
1521 1522
      #   user.send(:attributes=, { :username => 'Phusion', :is_admin => true }, false)
      #   user.is_admin?  # => true
1523
      def attributes=(new_attributes, guard_protected_attributes = true)
P
Pratik Naik 已提交
1524
        return unless new_attributes.is_a?(Hash)
1525
        attributes = new_attributes.stringify_keys
D
Initial  
David Heinemeier Hansson 已提交
1526 1527

        multi_parameter_attributes = []
1528
        attributes = sanitize_for_mass_assignment(attributes) if guard_protected_attributes
1529

1530
        attributes.each do |k, v|
1531 1532 1533 1534 1535
          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 已提交
1536
        end
D
David Heinemeier Hansson 已提交
1537

D
Initial  
David Heinemeier Hansson 已提交
1538 1539 1540
        assign_multiparameter_attributes(multi_parameter_attributes)
      end

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

P
Pratik Naik 已提交
1548 1549 1550 1551 1552 1553 1554 1555 1556 1557 1558 1559 1560
      # 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"'
1561 1562 1563
      def attribute_for_inspect(attr_name)
        value = read_attribute(attr_name)

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

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

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

X
Xavier Noria 已提交
1585 1586 1587 1588 1589 1590 1591 1592 1593
      # 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 已提交
1594
      def ==(comparison_object)
1595
        comparison_object.equal?(self) ||
1596 1597 1598
          comparison_object.instance_of?(self.class) &&
          id.present? &&
          comparison_object.id == id
D
Initial  
David Heinemeier Hansson 已提交
1599 1600 1601 1602 1603 1604
      end

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

D
Initial  
David Heinemeier Hansson 已提交
1606 1607 1608
      # 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
1609
        id.hash
D
Initial  
David Heinemeier Hansson 已提交
1610 1611
      end

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

1617
      # Returns +true+ if the attributes hash has been frozen.
1618 1619 1620
      def frozen?
        @attributes.frozen?
      end
1621

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

1629 1630
      # 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.
1631
      def readonly?
1632
        @readonly
1633 1634
      end

1635 1636
      # Marks this record as read only.
      def readonly!
1637 1638
        @readonly = true
      end
1639

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

1650 1651
    protected
      def clone_attributes(reader_method = :read_attribute, attributes = {})
1652 1653
        attribute_names.each do |name|
          attributes[name] = clone_attribute_value(reader_method, name)
1654
        end
1655
        attributes
1656 1657 1658 1659 1660 1661 1662 1663 1664
      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 已提交
1665 1666
    private

1667
      # Sets the attribute used for single table inheritance to this class name if this is not the
1668
      # ActiveRecord::Base descendant.
1669 1670
      # 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.
1671
      # No such attribute would be set for objects of the Message class in that example.
D
Initial  
David Heinemeier Hansson 已提交
1672 1673
      def ensure_proper_type
        unless self.class.descends_from_active_record?
1674
          write_attribute(self.class.inheritance_column, self.class.sti_name)
D
Initial  
David Heinemeier Hansson 已提交
1675 1676 1677
        end
      end

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

1685 1686
      # Returns a copy of the attributes hash where all the values have been safely quoted for use in
      # an Arel insert/update method.
1687 1688 1689 1690 1691 1692
      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))
1693 1694
              value = read_attribute(name)

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

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

P
Pratik Naik 已提交
1710
      # Interpolate custom SQL string in instance context.
D
Initial  
David Heinemeier Hansson 已提交
1711 1712
      # Optional record argument is meant for custom insert_sql.
      def interpolate_sql(sql, record = nil)
1713
        instance_eval("%@#{sql.gsub('@', '\@')}@", __FILE__, __LINE__)
D
Initial  
David Heinemeier Hansson 已提交
1714 1715 1716 1717 1718 1719
      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
1720 1721
      # 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
1722
      # attribute will be set to nil.
D
Initial  
David Heinemeier Hansson 已提交
1723 1724 1725 1726 1727
      def assign_multiparameter_attributes(pairs)
        execute_callstack_for_multiparameter_attributes(
          extract_callstack_for_multiparameter_attributes(pairs)
        )
      end
1728

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

D
Initial  
David Heinemeier Hansson 已提交
1737
      def execute_callstack_for_multiparameter_attributes(callstack)
1738
        errors = []
1739
        callstack.each do |name, values_with_empty_parameters|
1740 1741
          begin
            klass = (self.class.reflect_on_aggregation(name.to_sym) || column_for_attribute(name)).klass
1742 1743
            # 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.
1744
            values = values_with_empty_parameters.reject { |v| v.nil? }
1745

1746 1747 1748
            if values.empty?
              send(name + "=", nil)
            else
1749

1750
              value = if Time == klass
1751
                instantiate_time_object(name, values)
1752 1753
              elsif Date == klass
                begin
1754
                  values = values_with_empty_parameters.collect do |v| v.nil? ? 1 : v end
1755 1756
                  Date.new(*values)
                rescue ArgumentError => ex # if Date.new raises an exception on an invalid date
1757
                  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
1758 1759 1760 1761 1762 1763
                end
              else
                klass.new(*values)
              end

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

D
Initial  
David Heinemeier Hansson 已提交
1774 1775 1776 1777 1778 1779 1780 1781
      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)

1782 1783
          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 已提交
1784 1785 1786 1787
        end

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

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

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

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

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

1806 1807
      def quoted_comma_pair_list(quoter, hash)
        comma_pair_list(quote_columns(quoter, hash))
D
Initial  
David Heinemeier Hansson 已提交
1808 1809
      end

1810 1811 1812 1813 1814 1815 1816 1817 1818 1819 1820 1821 1822 1823 1824 1825
      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
1826 1827 1828 1829 1830 1831 1832

      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 已提交
1833
  end
J
Joshua Peek 已提交
1834 1835

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

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

    # 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 已提交
1858
    include Aggregations, Transactions, Reflection, Serialization
1859

1860
    NilClass.add_whiner(self) if NilClass.respond_to?(:add_whiner)
J
Joshua Peek 已提交
1861
  end
1862
end
1863 1864 1865

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