query_methods.rb 35.6 KB
Newer Older
1
require 'active_support/core_ext/array/wrap'
2

3 4
module ActiveRecord
  module QueryMethods
5 6
    extend ActiveSupport::Concern

7
    # WhereChain objects act as placeholder for queries in which #where does not have any parameter.
8
    # In this case, #where must be chained with #not to return a new relation.
9 10 11 12 13
    class WhereChain
      def initialize(scope)
        @scope = scope
      end

14 15
      # Returns a new relation expressing WHERE + NOT condition according to
      # the conditions in the arguments.
16
      #
17 18
      # +not+ accepts conditions as a string, array, or hash. See #where for
      # more details on each format.
19 20
      #
      #    User.where.not("name = 'Jon'")
21
      #    # SELECT * FROM users WHERE NOT (name = 'Jon')
22 23
      #
      #    User.where.not(["name = ?", "Jon"])
24
      #    # SELECT * FROM users WHERE NOT (name = 'Jon')
25
      #
26
      #    User.where.not(name: "Jon")
27
      #    # SELECT * FROM users WHERE name != 'Jon'
28 29 30 31
      #
      #    User.where.not(name: nil)
      #    # SELECT * FROM users WHERE name IS NOT NULL
      #
32
      #    User.where.not(name: %w(Ko1 Nobu))
33
      #    # SELECT * FROM users WHERE name NOT IN ('Ko1', 'Nobu')
34 35 36
      #
      #    User.where.not(name: "Jon", role: "admin")
      #    # SELECT * FROM users WHERE name != 'Jon' AND role != 'admin'
37 38 39
      def not(opts, *rest)
        where_value = @scope.send(:build_where, opts, rest).map do |rel|
          case rel
40 41
          when NilClass
            raise ArgumentError, 'Invalid argument for .where.not(), got nil.'
42 43
          when Arel::Nodes::In
            Arel::Nodes::NotIn.new(rel.left, rel.right)
44 45
          when Arel::Nodes::Equality
            Arel::Nodes::NotEqual.new(rel.left, rel.right)
46 47 48 49 50 51 52 53 54 55 56
          when String
            Arel::Nodes::Not.new(Arel::Nodes::SqlLiteral.new(rel))
          else
            Arel::Nodes::Not.new(rel)
          end
        end
        @scope.where_values += where_value
        @scope
      end
    end

57 58
    Relation::MULTI_VALUE_METHODS.each do |name|
      class_eval <<-CODE, __FILE__, __LINE__ + 1
59 60 61 62 63 64 65 66
        def #{name}_values                   # def select_values
          @values[:#{name}] || []            #   @values[:select] || []
        end                                  # end
                                             #
        def #{name}_values=(values)          # def select_values=(values)
          raise ImmutableRelation if @loaded #   raise ImmutableRelation if @loaded
          @values[:#{name}] = values         #   @values[:select] = values
        end                                  # end
67 68 69 70 71
      CODE
    end

    (Relation::SINGLE_VALUE_METHODS - [:create_with]).each do |name|
      class_eval <<-CODE, __FILE__, __LINE__ + 1
72 73 74
        def #{name}_value                    # def readonly_value
          @values[:#{name}]                  #   @values[:readonly]
        end                                  # end
75 76 77
      CODE
    end

78 79 80 81 82 83 84
    Relation::SINGLE_VALUE_METHODS.each do |name|
      class_eval <<-CODE, __FILE__, __LINE__ + 1
        def #{name}_value=(value)            # def readonly_value=(value)
          raise ImmutableRelation if @loaded #   raise ImmutableRelation if @loaded
          @values[:#{name}] = value          #   @values[:readonly] = value
        end                                  # end
      CODE
85 86
    end

O
Oscar Del Ben 已提交
87
    def create_with_value # :nodoc:
88
      @values[:create_with] || {}
89
    end
90 91

    alias extensions extending_values
92

O
Oscar Del Ben 已提交
93 94 95 96 97 98 99 100 101 102
    # Specify relationships to be included in the result set. For
    # example:
    #
    #   users = User.includes(:address)
    #   users.each do |user|
    #     user.address.city
    #   end
    #
    # allows you to access the +address+ attribute of the +User+ model without
    # firing an additional query. This will often result in a
103 104
    # performance improvement over a simple +join+.
    #
105 106 107 108 109 110 111 112
    # You can also specify multiple relationships, like this:
    #
    #   users = User.includes(:address, :friends)
    #
    # Loading nested relationships is possible using a Hash:
    #
    #   users = User.includes(:address, friends: [:address, :followers])
    #
113 114 115 116 117 118 119 120 121 122
    # === conditions
    #
    # If you want to add conditions to your included models you'll have
    # to explicitly reference them. For example:
    #
    #   User.includes(:posts).where('posts.name = ?', 'example')
    #
    # Will throw an error, but this will work:
    #
    #   User.includes(:posts).where('posts.name = ?', 'example').references(:posts)
123
    def includes(*args)
124
      check_if_method_has_arguments!(:includes, args)
125
      spawn.includes!(*args)
126
    end
127

J
Jon Leighton 已提交
128
    def includes!(*args) # :nodoc:
129 130
      args.reject!(&:blank?)
      args.flatten!
A
Aaron Patterson 已提交
131

132
      self.includes_values |= args
133
      self
134
    end
135

136 137 138 139 140 141
    # Forces eager loading by performing a LEFT OUTER JOIN on +args+:
    #
    #   User.eager_load(:posts)
    #   => SELECT "users"."id" AS t0_r0, "users"."name" AS t0_r1, ...
    #   FROM "users" LEFT OUTER JOIN "posts" ON "posts"."user_id" =
    #   "users"."id"
142
    def eager_load(*args)
143
      check_if_method_has_arguments!(:eager_load, args)
144
      spawn.eager_load!(*args)
145
    end
146

J
Jon Leighton 已提交
147
    def eager_load!(*args) # :nodoc:
148 149
      self.eager_load_values += args
      self
150 151
    end

152 153 154 155
    # Allows preloading of +args+, in the same way that +includes+ does:
    #
    #   User.preload(:posts)
    #   => SELECT "posts".* FROM "posts" WHERE "posts"."user_id" IN (1, 2, 3)
156
    def preload(*args)
157
      check_if_method_has_arguments!(:preload, args)
158
      spawn.preload!(*args)
159
    end
160

J
Jon Leighton 已提交
161
    def preload!(*args) # :nodoc:
162 163
      self.preload_values += args
      self
164
    end
165

166 167 168 169 170 171 172 173 174
    # Used to indicate that an association is referenced by an SQL string, and should
    # therefore be JOINed in any query rather than loaded separately.
    #
    #   User.includes(:posts).where("posts.name = 'foo'")
    #   # => Doesn't JOIN the posts table, resulting in an error.
    #
    #   User.includes(:posts).where("posts.name = 'foo'").references(:posts)
    #   # => Query now knows the string references posts, so adds a JOIN
    def references(*args)
175
      check_if_method_has_arguments!(:references, args)
176
      spawn.references!(*args)
177
    end
178

J
Jon Leighton 已提交
179
    def references!(*args) # :nodoc:
180
      args.flatten!
181
      args.map!(&:to_s)
182

183
      self.references_values |= args
184
      self
185 186
    end

187
    # Works in two unique ways.
188
    #
189 190
    # First: takes a block so it can be used just like Array#select.
    #
191
    #   Model.all.select { |m| m.field == value }
192 193 194 195 196
    #
    # This will build an array of objects from the database for the scope,
    # converting them into an array and iterating through them using Array#select.
    #
    # Second: Modifies the SELECT statement for the query so that only certain
V
Vijay Dev 已提交
197
    # fields are retrieved:
198
    #
199 200
    #   Model.select(:field)
    #   # => [#<Model field:value>]
201 202
    #
    # Although in the above example it looks as though this method returns an
V
Vijay Dev 已提交
203
    # array, it actually returns a relation object and can have other query
204 205
    # methods appended to it, such as the other methods in ActiveRecord::QueryMethods.
    #
206
    # The argument to the method can also be an array of fields.
207
    #
208 209
    #   Model.select(:field, :other_field, :and_one_more)
    #   # => [#<Model field: "value", other_field: "value", and_one_more: "value">]
210
    #
211 212 213 214 215 216 217 218 219 220
    # You can also use one or more strings, which will be used unchanged as SELECT fields.
    #
    #   Model.select('field AS field_one', 'other_field AS field_two')
    #   # => [#<Model field: "value", other_field: "value">]
    #
    # If an alias was specified, it will be accessible from the resulting objects:
    #
    #   Model.select('field AS field_one').first.field_one
    #   # => "value"
    #
221 222
    # Accessing attributes of an object that do not have fields retrieved by a select
    # will throw <tt>ActiveModel::MissingAttributeError</tt>:
223
    #
224 225 226
    #   Model.select(:field).first.other_field
    #   # => ActiveModel::MissingAttributeError: missing attribute: other_field
    def select(*fields)
227
      if block_given?
228
        to_a.select { |*block_args| yield(*block_args) }
229
      else
230 231
        raise ArgumentError, 'Call this with at least one field' if fields.empty?
        spawn.select!(*fields)
232 233 234
      end
    end

J
Jon Leighton 已提交
235
    def select!(*fields) # :nodoc:
236 237 238
      fields.flatten!

      self.select_values += fields
239
      self
240
    end
S
Santiago Pastorino 已提交
241

O
Oscar Del Ben 已提交
242 243 244 245 246
    # Allows to specify a group attribute:
    #
    #   User.group(:name)
    #   => SELECT "users".* FROM "users" GROUP BY name
    #
247
    # Returns an array with distinct records based on the +group+ attribute:
O
Oscar Del Ben 已提交
248 249 250 251 252 253
    #
    #   User.select([:id, :name])
    #   => [#<User id: 1, name: "Oscar">, #<User id: 2, name: "Oscar">, #<User id: 3, name: "Foo">
    #
    #   User.group(:name)
    #   => [#<User id: 3, name: "Foo", ...>, #<User id: 2, name: "Oscar", ...>]
254 255 256
    #
    #   User.group('name AS grouped_name, age')
    #   => [#<User id: 3, name: "Foo", age: 21, ...>, #<User id: 2, name: "Oscar", age: 21, ...>, #<User id: 5, name: "Foo", age: 23, ...>]
257
    def group(*args)
258
      check_if_method_has_arguments!(:group, args)
259
      spawn.group!(*args)
260
    end
261

J
Jon Leighton 已提交
262
    def group!(*args) # :nodoc:
263 264 265
      args.flatten!

      self.group_values += args
266
      self
267
    end
268

O
Oscar Del Ben 已提交
269 270 271 272 273 274 275 276 277 278
    # Allows to specify an order attribute:
    #
    #   User.order('name')
    #   => SELECT "users".* FROM "users" ORDER BY name
    #
    #   User.order('name DESC')
    #   => SELECT "users".* FROM "users" ORDER BY name DESC
    #
    #   User.order('name DESC, email')
    #   => SELECT "users".* FROM "users" ORDER BY name DESC, email
279
    #
280 281
    #   User.order(:name)
    #   => SELECT "users".* FROM "users" ORDER BY "users"."name" ASC
282
    #
283 284
    #   User.order(email: :desc)
    #   => SELECT "users".* FROM "users" ORDER BY "users"."email" DESC
285
    #
286 287
    #   User.order(:name, email: :desc)
    #   => SELECT "users".* FROM "users" ORDER BY "users"."name" ASC, "users"."email" DESC
288
    def order(*args)
289
      check_if_method_has_arguments!(:order, args)
290
      spawn.order!(*args)
291
    end
292

J
Jon Leighton 已提交
293
    def order!(*args) # :nodoc:
294
      preprocess_order_args(args)
295

296
      self.order_values += args
297
      self
298
    end
299

300 301 302 303 304 305 306 307
    # Replaces any existing order defined on the relation with the specified order.
    #
    #   User.order('email DESC').reorder('id ASC') # generated SQL has 'ORDER BY id ASC'
    #
    # Subsequent calls to order on the same relation will be appended. For example:
    #
    #   User.order('email DESC').reorder('id ASC').order('name ASC')
    #
308
    # generates a query with 'ORDER BY id ASC, name ASC'.
S
Sebastian Martinez 已提交
309
    def reorder(*args)
310
      check_if_method_has_arguments!(:reorder, args)
311
      spawn.reorder!(*args)
312
    end
313

J
Jon Leighton 已提交
314
    def reorder!(*args) # :nodoc:
315
      preprocess_order_args(args)
316

317
      self.reordering_value = true
318
      self.order_values = args
319
      self
S
Sebastian Martinez 已提交
320 321
    end

322 323 324 325 326 327 328 329
    VALID_UNSCOPING_VALUES = Set.new([:where, :select, :group, :order, :lock,
                                     :limit, :offset, :joins, :includes, :from,
                                     :readonly, :having])

    # Removes an unwanted relation that is already defined on a chain of relations.
    # This is useful when passing around chains of relations and would like to
    # modify the relations without reconstructing the entire chain.
    #
330
    #   User.order('email DESC').unscope(:order) == User.all
331 332 333 334 335
    #
    # The method arguments are symbols which correspond to the names of the methods
    # which should be unscoped. The valid arguments are given in VALID_UNSCOPING_VALUES.
    # The method can also be called with multiple arguments. For example:
    #
336
    #   User.order('email DESC').select('id').where(name: "John")
337 338 339 340 341 342
    #       .unscope(:order, :select, :where) == User.all
    #
    # One can additionally pass a hash as an argument to unscope specific :where values.
    # This is done by passing a hash with a single key-value pair. The key should be
    # :where and the value should be the where value to unscope. For example:
    #
343 344
    #   User.where(name: "John", active: true).unscope(where: :name)
    #       == User.where(active: true)
345
    #
J
Jon Leighton 已提交
346 347
    # This method is similar to <tt>except</tt>, but unlike
    # <tt>except</tt>, it persists across merges:
348
    #
J
Jon Leighton 已提交
349 350
    #   User.order('email').merge(User.except(:order))
    #       == User.order('email')
351
    #
J
Jon Leighton 已提交
352 353 354 355 356 357
    #   User.order('email').merge(User.unscope(:order))
    #       == User.all
    #
    # This means it can be used in association definitions:
    #
    #   has_many :comments, -> { unscope where: :trashed }
358 359
    #
    def unscope(*args)
360
      check_if_method_has_arguments!(:unscope, args)
361 362 363
      spawn.unscope!(*args)
    end

364
    def unscope!(*args) # :nodoc:
365
      args.flatten!
J
Jon Leighton 已提交
366
      self.unscope_values += args
367 368 369 370 371 372 373 374 375 376 377 378 379 380 381 382 383 384 385 386 387 388 389

      args.each do |scope|
        case scope
        when Symbol
          symbol_unscoping(scope)
        when Hash
          scope.each do |key, target_value|
            if key != :where
              raise ArgumentError, "Hash arguments in .unscope(*args) must have :where as the key."
            end

            Array(target_value).each do |val|
              where_unscoping(val)
            end
          end
        else
          raise ArgumentError, "Unrecognized scoping: #{args.inspect}. Use .unscope(where: :attribute_name) or .unscope(:order), for example."
        end
      end

      self
    end

390 391 392 393
    # Performs a joins on +args+:
    #
    #   User.joins(:posts)
    #   => SELECT "users".* FROM "users" INNER JOIN "posts" ON "posts"."user_id" = "users"."id"
394 395 396 397 398
    #
    # You can use strings in order to customize your joins:
    #
    #   User.joins("LEFT JOIN bookmarks ON bookmarks.bookmarkable_type = 'Post' AND bookmarks.user_id = users.id")
    #   => SELECT "users".* FROM "users" LEFT JOIN bookmarks ON bookmarks.bookmarkable_type = 'Post' AND bookmarks.user_id = users.id
399
    def joins(*args)
400 401 402 403 404 405
      check_if_method_has_arguments!(:joins, args)

      args.compact!
      args.flatten!

      spawn.joins!(*args)
406
    end
407

J
Jon Leighton 已提交
408
    def joins!(*args) # :nodoc:
409 410
      self.joins_values += args
      self
P
Pratik Naik 已提交
411 412
    end

A
Aaron Patterson 已提交
413
    def bind(value)
J
Jon Leighton 已提交
414
      spawn.bind!(value)
415 416
    end

J
Jon Leighton 已提交
417
    def bind!(value) # :nodoc:
418 419
      self.bind_values += [value]
      self
A
Aaron Patterson 已提交
420 421
    end

422 423 424 425 426 427 428 429 430 431
    # Returns a new relation, which is the result of filtering the current relation
    # according to the conditions in the arguments.
    #
    # #where accepts conditions in one of several formats. In the examples below, the resulting
    # SQL is given as an illustration; the actual query generated may be different depending
    # on the database adapter.
    #
    # === string
    #
    # A single string, without additional arguments, is passed to the query
432
    # constructor as an SQL fragment, and used in the where clause of the query.
433 434 435 436 437 438 439 440 441 442 443 444 445 446 447 448 449 450 451 452 453 454 455 456 457 458 459 460 461 462 463 464 465 466 467 468 469 470 471 472 473 474 475 476 477 478 479 480 481 482 483 484 485 486 487 488 489 490 491 492 493 494 495 496
    #
    #    Client.where("orders_count = '2'")
    #    # SELECT * from clients where orders_count = '2';
    #
    # Note that building your own string from user input may expose your application
    # to injection attacks if not done properly. As an alternative, it is recommended
    # to use one of the following methods.
    #
    # === array
    #
    # If an array is passed, then the first element of the array is treated as a template, and
    # the remaining elements are inserted into the template to generate the condition.
    # Active Record takes care of building the query to avoid injection attacks, and will
    # convert from the ruby type to the database type where needed. Elements are inserted
    # into the string in the order in which they appear.
    #
    #   User.where(["name = ? and email = ?", "Joe", "joe@example.com"])
    #   # SELECT * FROM users WHERE name = 'Joe' AND email = 'joe@example.com';
    #
    # Alternatively, you can use named placeholders in the template, and pass a hash as the
    # second element of the array. The names in the template are replaced with the corresponding
    # values from the hash.
    #
    #   User.where(["name = :name and email = :email", { name: "Joe", email: "joe@example.com" }])
    #   # SELECT * FROM users WHERE name = 'Joe' AND email = 'joe@example.com';
    #
    # This can make for more readable code in complex queries.
    #
    # Lastly, you can use sprintf-style % escapes in the template. This works slightly differently
    # than the previous methods; you are responsible for ensuring that the values in the template
    # are properly quoted. The values are passed to the connector for quoting, but the caller
    # is responsible for ensuring they are enclosed in quotes in the resulting SQL. After quoting,
    # the values are inserted using the same escapes as the Ruby core method <tt>Kernel::sprintf</tt>.
    #
    #   User.where(["name = '%s' and email = '%s'", "Joe", "joe@example.com"])
    #   # SELECT * FROM users WHERE name = 'Joe' AND email = 'joe@example.com';
    #
    # If #where is called with multiple arguments, these are treated as if they were passed as
    # the elements of a single array.
    #
    #   User.where("name = :name and email = :email", { name: "Joe", email: "joe@example.com" })
    #   # SELECT * FROM users WHERE name = 'Joe' AND email = 'joe@example.com';
    #
    # When using strings to specify conditions, you can use any operator available from
    # the database. While this provides the most flexibility, you can also unintentionally introduce
    # dependencies on the underlying database. If your code is intended for general consumption,
    # test with multiple database backends.
    #
    # === hash
    #
    # #where will also accept a hash condition, in which the keys are fields and the values
    # are values to be searched for.
    #
    # Fields can be symbols or strings. Values can be single values, arrays, or ranges.
    #
    #    User.where({ name: "Joe", email: "joe@example.com" })
    #    # SELECT * FROM users WHERE name = 'Joe' AND email = 'joe@example.com'
    #
    #    User.where({ name: ["Alice", "Bob"]})
    #    # SELECT * FROM users WHERE name IN ('Alice', 'Bob')
    #
    #    User.where({ created_at: (Time.now.midnight - 1.day)..Time.now.midnight })
    #    # SELECT * FROM users WHERE (created_at BETWEEN '2012-06-09 07:00:00.000000' AND '2012-06-10 07:00:00.000000')
    #
497 498 499 500 501 502
    # In the case of a belongs_to relationship, an association key can be used
    # to specify the model if an ActiveRecord object is used as the value.
    #
    #    author = Author.find(1)
    #
    #    # The following queries will be equivalent:
A
AvnerCohen 已提交
503 504
    #    Post.where(author: author)
    #    Post.where(author_id: author)
505 506 507
    #
    # This also works with polymorphic belongs_to relationships:
    #
A
AvnerCohen 已提交
508 509
    #    treasure = Treasure.create(name: 'gold coins')
    #    treasure.price_estimates << PriceEstimate.create(price: 125)
510 511
    #
    #    # The following queries will be equivalent:
A
AvnerCohen 已提交
512 513
    #    PriceEstimate.where(estimate_of: treasure)
    #    PriceEstimate.where(estimate_of_type: 'Treasure', estimate_of_id: treasure)
514
    #
515 516 517 518 519 520 521 522 523 524
    # === Joins
    #
    # If the relation is the result of a join, you may create a condition which uses any of the
    # tables in the join. For string and array conditions, use the table name in the condition.
    #
    #    User.joins(:posts).where("posts.created_at < ?", Time.now)
    #
    # For hash conditions, you can either use the table name in the key, or use a sub-hash.
    #
    #    User.joins(:posts).where({ "posts.published" => true })
A
AvnerCohen 已提交
525
    #    User.joins(:posts).where({ posts: { published: true } })
526
    #
527
    # === no argument
528
    #
529 530
    # If no argument is passed, #where returns a new instance of WhereChain, that
    # can be chained with #not to return a new relation that negates the where clause.
531 532
    #
    #    User.where.not(name: "Jon")
533
    #    # SELECT * FROM users WHERE name != 'Jon'
534
    #
535
    # See WhereChain for more details on #not.
536
    #
537
    # === blank condition
538
    #
539
    # If the condition is any blank-ish object, then #where is a no-op and returns
540
    # the current relation.
541 542
    def where(opts = :chain, *rest)
      if opts == :chain
543 544 545 546 547 548
        WhereChain.new(spawn)
      elsif opts.blank?
        self
      else
        spawn.where!(opts, *rest)
      end
549 550
    end

551 552
    def where!(opts = :chain, *rest) # :nodoc:
      if opts == :chain
553 554 555
        WhereChain.new(self)
      else
        references!(PredicateBuilder.references(opts)) if Hash === opts
556

557 558 559
        self.where_values += build_where(opts, rest)
        self
      end
560
    end
P
Pratik Naik 已提交
561

562 563
    # Allows you to change a previously set where condition for a given attribute, instead of appending to that condition.
    #
564 565 566
    #   Post.where(trashed: true).where(trashed: false)                       # => WHERE `trashed` = 1 AND `trashed` = 0
    #   Post.where(trashed: true).rewhere(trashed: false)                     # => WHERE `trashed` = 0
    #   Post.where(active: true).where(trashed: true).rewhere(trashed: false) # => WHERE `active` = 1 AND `trashed` = 0
567 568 569 570 571 572 573
    #
    # This is short-hand for unscope(where: conditions.keys).where(conditions). Note that unlike reorder, we're only unscoping
    # the named conditions -- not the entire where statement.
    def rewhere(conditions)
      unscope(where: conditions.keys).where(conditions)
    end

574 575 576 577
    # Allows to specify a HAVING clause. Note that you can't use HAVING
    # without also specifying a GROUP clause.
    #
    #   Order.having('SUM(price) > 30').group('user_id')
578
    def having(opts, *rest)
579
      opts.blank? ? self : spawn.having!(opts, *rest)
580 581
    end

J
Jon Leighton 已提交
582
    def having!(opts, *rest) # :nodoc:
583
      references!(PredicateBuilder.references(opts)) if Hash === opts
584

585 586
      self.having_values += build_where(opts, rest)
      self
587 588
    end

589
    # Specifies a limit for the number of records to retrieve.
590 591 592 593
    #
    #   User.limit(10) # generated SQL has 'LIMIT 10'
    #
    #   User.limit(10).limit(20) # generated SQL has 'LIMIT 20'
594
    def limit(value)
J
Jon Leighton 已提交
595
      spawn.limit!(value)
596 597
    end

J
Jon Leighton 已提交
598
    def limit!(value) # :nodoc:
599 600
      self.limit_value = value
      self
601 602
    end

603 604 605 606
    # Specifies the number of rows to skip before returning rows.
    #
    #   User.offset(10) # generated SQL has "OFFSET 10"
    #
607
    # Should be used with order.
608
    #
609
    #   User.offset(10).order("name ASC")
610
    def offset(value)
J
Jon Leighton 已提交
611
      spawn.offset!(value)
612 613
    end

J
Jon Leighton 已提交
614
    def offset!(value) # :nodoc:
615 616
      self.offset_value = value
      self
617 618
    end

619
    # Specifies locking settings (default to +true+). For more information
620
    # on locking, please see +ActiveRecord::Locking+.
621
    def lock(locks = true)
J
Jon Leighton 已提交
622
      spawn.lock!(locks)
623
    end
624

J
Jon Leighton 已提交
625
    def lock!(locks = true) # :nodoc:
626
      case locks
627
      when String, TrueClass, NilClass
628
        self.lock_value = locks || true
629
      else
630
        self.lock_value = false
631
      end
632

633
      self
634 635
    end

636
    # Returns a chainable relation with zero records.
637
    #
638 639 640
    # The returned relation implements the Null Object pattern. It is an
    # object with defined null behavior and always returns an empty array of
    # records without querying the database.
641 642 643 644
    #
    # Any subsequent condition chained to the returned relation will continue
    # generating an empty relation and will not fire any query to the database.
    #
645 646
    # Used in cases where a method or scope could return zero records but the
    # result needs to be chainable.
647 648 649
    #
    # For example:
    #
A
AvnerCohen 已提交
650
    #   @posts = current_user.visible_posts.where(name: params[:name])
651
    #   # => the visible_posts method is expected to return a chainable Relation
652 653 654
    #
    #   def visible_posts
    #     case role
655
    #     when 'Country Manager'
A
AvnerCohen 已提交
656
    #       Post.where(country: country)
657
    #     when 'Reviewer'
658
    #       Post.published
659
    #     when 'Bad User'
660
    #       Post.none # It can't be chained if [] is returned.
661 662 663 664
    #     end
    #   end
    #
    def none
665
      extending(NullRelation)
666 667
    end

J
Jon Leighton 已提交
668
    def none! # :nodoc:
669 670 671
      extending!(NullRelation)
    end

672 673 674 675 676 677
    # Sets readonly attributes for the returned relation. If value is
    # true (default), attempting to update a record will result in an error.
    #
    #   users = User.readonly
    #   users.first.save
    #   => ActiveRecord::ReadOnlyRecord: ActiveRecord::ReadOnlyRecord
678
    def readonly(value = true)
J
Jon Leighton 已提交
679
      spawn.readonly!(value)
680 681
    end

J
Jon Leighton 已提交
682
    def readonly!(value = true) # :nodoc:
683 684
      self.readonly_value = value
      self
685 686
    end

687 688 689 690 691 692 693 694 695 696 697 698 699
    # Sets attributes to be used when creating new records from a
    # relation object.
    #
    #   users = User.where(name: 'Oscar')
    #   users.new.name # => 'Oscar'
    #
    #   users = users.create_with(name: 'DHH')
    #   users.new.name # => 'DHH'
    #
    # You can pass +nil+ to +create_with+ to reset attributes:
    #
    #   users = users.create_with(nil)
    #   users.new.name # => 'Oscar'
700
    def create_with(value)
J
Jon Leighton 已提交
701
      spawn.create_with!(value)
702 703
    end

J
Jon Leighton 已提交
704
    def create_with!(value) # :nodoc:
705 706
      self.create_with_value = value ? create_with_value.merge(value) : {}
      self
707 708
    end

709 710 711
    # Specifies table from which the records will be fetched. For example:
    #
    #   Topic.select('title').from('posts')
712
    #   # => SELECT title FROM posts
713 714 715
    #
    # Can accept other relation objects. For example:
    #
716
    #   Topic.select('title').from(Topic.approved)
717 718
    #   # => SELECT title FROM (SELECT * FROM topics WHERE approved = 't') subquery
    #
719
    #   Topic.select('a.title').from(Topic.approved, :a)
720 721 722 723
    #   # => SELECT a.title FROM (SELECT * FROM topics WHERE approved = 't') a
    #
    def from(value, subquery_name = nil)
      spawn.from!(value, subquery_name)
724 725
    end

J
Jon Leighton 已提交
726
    def from!(value, subquery_name = nil) # :nodoc:
727
      self.from_value = [value, subquery_name]
728
      self
729 730
    end

731 732 733 734 735
    # Specifies whether the records should be unique or not. For example:
    #
    #   User.select(:name)
    #   # => Might return two records with the same name
    #
736 737
    #   User.select(:name).distinct
    #   # => Returns 1 record per distinct name
738
    #
739
    #   User.select(:name).distinct.distinct(false)
740
    #   # => You can also remove the uniqueness
741 742
    def distinct(value = true)
      spawn.distinct!(value)
743
    end
744
    alias uniq distinct
745

746 747 748
    # Like #distinct, but modifies relation in place.
    def distinct!(value = true) # :nodoc:
      self.distinct_value = value
749
      self
750
    end
751
    alias uniq! distinct!
752

753
    # Used to extend a scope with additional methods, either through
754 755
    # a module or through a block provided.
    #
756 757 758 759 760 761 762 763 764 765
    # The object returned is a relation, which can be further extended.
    #
    # === Using a module
    #
    #   module Pagination
    #     def page(number)
    #       # pagination code goes here
    #     end
    #   end
    #
766
    #   scope = Model.all.extending(Pagination)
767 768
    #   scope.page(params[:page])
    #
V
Vijay Dev 已提交
769
    # You can also pass a list of modules:
770
    #
771
    #   scope = Model.all.extending(Pagination, SomethingElse)
772 773 774
    #
    # === Using a block
    #
775
    #   scope = Model.all.extending do
776
    #     def page(number)
777
    #       # pagination code goes here
778 779 780 781 782 783
    #     end
    #   end
    #   scope.page(params[:page])
    #
    # You can also use a block and a module list:
    #
784
    #   scope = Model.all.extending(Pagination) do
785
    #     def per_page(number)
786
    #       # pagination code goes here
787 788
    #     end
    #   end
789 790
    def extending(*modules, &block)
      if modules.any? || block
J
Jon Leighton 已提交
791
        spawn.extending!(*modules, &block)
792 793 794 795
      else
        self
      end
    end
796

J
Jon Leighton 已提交
797
    def extending!(*modules, &block) # :nodoc:
798 799
      modules << Module.new(&block) if block
      modules.flatten!
800

801
      self.extending_values += modules
802
      extend(*extending_values) if extending_values.any?
803

804
      self
805 806
    end

807 808 809
    # Reverse the existing order clause on the relation.
    #
    #   User.order('name ASC').reverse_order # generated SQL has 'ORDER BY name DESC'
810
    def reverse_order
J
Jon Leighton 已提交
811
      spawn.reverse_order!
812 813
    end

J
Jon Leighton 已提交
814
    def reverse_order! # :nodoc:
815 816
      self.reverse_order_value = !reverse_order_value
      self
817 818
    end

819
    # Returns the Arel object associated with the relation.
820
    def arel
821
      @arel ||= build_arel
822 823
    end

824
    # Like #arel, but ignores the default scope of the model.
825
    def build_arel
826
      arel = Arel::SelectManager.new(table.engine, table)
827

828
      build_joins(arel, joins_values.flatten) unless joins_values.empty?
829

830
      collapse_wheres(arel, (where_values - [''])) #TODO: Add uniq with real value comparison / ignore uniqs that have binds
831

832
      arel.having(*having_values.uniq.reject(&:blank?)) unless having_values.empty?
833

834 835
      arel.take(connection.sanitize_limit(limit_value)) if limit_value
      arel.skip(offset_value.to_i) if offset_value
A
Aaron Patterson 已提交
836

837
      arel.group(*group_values.uniq.reject(&:blank?)) unless group_values.empty?
838

839
      build_order(arel)
840

841
      build_select(arel, select_values.uniq)
842

843
      arel.distinct(distinct_value)
844
      arel.from(build_from) if from_value
845
      arel.lock(lock_value) if lock_value
846 847

      arel
848 849
    end

850 851
    private

852 853 854 855 856 857
    def symbol_unscoping(scope)
      if !VALID_UNSCOPING_VALUES.include?(scope)
        raise ArgumentError, "Called unscope() with invalid unscoping argument ':#{scope}'. Valid arguments are :#{VALID_UNSCOPING_VALUES.to_a.join(", :")}."
      end

      single_val_method = Relation::SINGLE_VALUE_METHODS.include?(scope)
A
Aaron Patterson 已提交
858
      unscope_code = "#{scope}_value#{'s' unless single_val_method}="
859 860 861

      case scope
      when :order
862
        self.reverse_order_value = false
863 864 865 866 867 868 869 870 871
        result = []
      else
        result = [] unless single_val_method
      end

      self.send(unscope_code, result)
    end

    def where_unscoping(target_value)
A
Aaron Patterson 已提交
872
      target_value = target_value.to_s
873 874 875

      where_values.reject! do |rel|
        case rel
876
        when Arel::Nodes::In, Arel::Nodes::NotIn, Arel::Nodes::Equality, Arel::Nodes::NotEqual
877
          subrelation = (rel.left.kind_of?(Arel::Attributes::Attribute) ? rel.left : rel.right)
A
Aaron Patterson 已提交
878
          subrelation.name == target_value
879 880 881 882
        else
          raise "unscope(where: #{target_value.inspect}) failed: unscoping #{rel.class} is unimplemented."
        end
      end
883 884

      bind_values.reject! { |col,_| col.name == target_value }
885 886
    end

887
    def custom_join_ast(table, joins)
888
      joins = joins.reject(&:blank?)
889

890
      return [] if joins.empty?
891

892
      joins.map! do |join|
893 894 895 896 897 898
        case join
        when Array
          join = Arel.sql(join.join(' ')) if array_of_strings?(join)
        when String
          join = Arel.sql(join)
        end
899
        table.create_string_join(join)
900 901 902
      end
    end

903
    def collapse_wheres(arel, wheres)
904 905
      predicates = wheres.map do |where|
        next where if ::Arel::Nodes::Equality === where
906
        where = Arel.sql(where) if String === where
907
        Arel::Nodes::Grouping.new(where)
908
      end
909 910

      arel.where(Arel::Nodes::And.new(predicates)) if predicates.present?
911 912
    end

913
    def build_where(opts, other = [])
A
Aaron Patterson 已提交
914 915
      case opts
      when String, Array
916 917 918 919 920 921 922
        #TODO: Remove duplication with: /activerecord/lib/active_record/sanitization.rb:113
        values = Hash === other.first ? other.first.values : other

        values.grep(ActiveRecord::Relation) do |rel|
          self.bind_values += rel.bind_values
        end

923
        [@klass.send(:sanitize_sql, other.empty? ? opts : ([opts] + other))]
A
Aaron Patterson 已提交
924
      when Hash
925
        temp_opts = opts.dup
926
        opts = PredicateBuilder.resolve_column_aliases(klass, opts)
927
        attributes = @klass.send(:expand_hash_conditions_for_aggregates, opts)
928

929 930 931 932
        create_binds(temp_opts)
        temp_opts = substitute_opts(temp_opts)

        attributes = @klass.send(:expand_hash_conditions_for_aggregates, temp_opts)
933 934 935 936
        attributes.values.grep(ActiveRecord::Relation) do |rel|
          self.bind_values += rel.bind_values
        end

937
        PredicateBuilder.build_from_hash(klass, attributes, table)
938
      else
939
        [opts]
940 941 942
      end
    end

943 944 945 946 947 948 949 950 951 952 953 954 955 956 957 958 959 960 961 962 963 964 965 966
    def create_binds(temp_opts)
      binds = []
      temp_opts.map do |column, value| 
        case value
          when String, Integer
            if @klass.column_names.include? column.to_s
              binds.push([@klass.columns_hash[column.to_s], value])
            end
        end
      end
      self.bind_values += binds
    end

    def substitute_opts(temp_opts)
      temp_opts = temp_opts.each_with_index do |(column,value), index|
        if @klass.columns_hash[column.to_s] != nil        
          case value
            when String, Integer
              temp_opts[column] = connection.substitute_at(column, index) 
          end
        end
      end
    end

967 968 969 970 971
    def build_from
      opts, name = from_value
      case opts
      when Relation
        name ||= 'subquery'
972
        self.bind_values = opts.bind_values + self.bind_values
973 974 975 976 977 978
        opts.arel.as(name.to_s)
      else
        opts
      end
    end

979
    def build_joins(manager, joins)
A
Aaron Patterson 已提交
980 981 982
      buckets = joins.group_by do |join|
        case join
        when String
983
          :string_join
A
Aaron Patterson 已提交
984
        when Hash, Symbol, Array
985
          :association_join
986
        when ActiveRecord::Associations::JoinDependency
987
          :stashed_join
988
        when Arel::Nodes::Join
989
          :join_node
A
Aaron Patterson 已提交
990 991 992
        else
          raise 'unknown class: %s' % join.class.name
        end
993 994
      end

995 996 997
      association_joins         = buckets[:association_join] || []
      stashed_association_joins = buckets[:stashed_join] || []
      join_nodes                = (buckets[:join_node] || []).uniq
998
      string_joins              = (buckets[:string_join] || []).map(&:strip).uniq
999

1000
      join_list = join_nodes + custom_join_ast(manager, string_joins)
1001

1002
      join_dependency = ActiveRecord::Associations::JoinDependency.new(
1003 1004 1005 1006
        @klass,
        association_joins,
        join_list
      )
1007

1008
      join_infos = join_dependency.join_constraints stashed_association_joins
1009

1010 1011
      join_infos.each do |info|
        info.joins.each { |join| manager.from(join) }
1012
        manager.bind_values.concat info.binds
1013
      end
1014

1015
      manager.join_sources.concat(join_list)
1016 1017

      manager
1018 1019
    end

1020
    def build_select(arel, selects)
C
Cody Cutrer 已提交
1021
      if !selects.empty?
1022
        arel.project(*selects)
C
Cody Cutrer 已提交
1023 1024
      elsif from_value
        arel.project(Arel.star)
1025
      else
1026
        arel.project(@klass.arel_table[Arel.star])
1027 1028 1029
      end
    end

1030
    def reverse_sql_order(order_query)
B
Brian Mathiyakom 已提交
1031 1032
      order_query = ["#{quoted_table_name}.#{quoted_primary_key} ASC"] if order_query.empty?

1033
      order_query.flat_map do |o|
1034
        case o
1035
        when Arel::Nodes::Ordering
1036
          o.reverse
1037
        when String
1038
          o.to_s.split(',').map! do |s|
1039 1040 1041
            s.strip!
            s.gsub!(/\sasc\Z/i, ' DESC') || s.gsub!(/\sdesc\Z/i, ' ASC') || s.concat(' DESC')
          end
1042 1043 1044
        else
          o
        end
1045
      end
1046 1047
    end

P
Pratik Naik 已提交
1048
    def array_of_strings?(o)
1049
      o.is_a?(Array) && o.all? { |obj| obj.is_a?(String) }
P
Pratik Naik 已提交
1050
    end
1051

1052
    def build_order(arel)
1053 1054
      orders = order_values.uniq
      orders.reject!(&:blank?)
1055
      orders = reverse_sql_order(orders) if reverse_order_value
1056

1057 1058
      arel.order(*orders) unless orders.empty?
    end
1059

1060
    def validate_order_args(args)
1061
      args.grep(Hash) do |h|
1062 1063 1064 1065 1066
        unless (h.values - [:asc, :desc]).empty?
          raise ArgumentError, 'Direction should be :asc or :desc'
        end
      end
    end
P
Pratik Naik 已提交
1067

1068 1069 1070 1071 1072 1073 1074 1075 1076 1077
    def preprocess_order_args(order_args)
      order_args.flatten!
      validate_order_args(order_args)

      references = order_args.grep(String)
      references.map! { |arg| arg =~ /^([a-zA-Z]\w*)\.(\w+)/ && $1 }.compact!
      references!(references) if references.any?

      # if a symbol is given we prepend the quoted table name
      order_args.map! do |arg|
1078 1079 1080 1081 1082 1083 1084 1085 1086 1087 1088
        case arg
        when Symbol
          table[arg].asc
        when Hash
          arg.map { |field, dir|
            table[field].send(dir)
          }
        else
          arg
        end
      end.flatten!
1089 1090
    end

1091 1092 1093 1094 1095 1096 1097 1098 1099 1100 1101 1102 1103
    # Checks to make sure that the arguments are not blank. Note that if some
    # blank-like object were initially passed into the query method, then this
    # method will not raise an error.
    #
    # Example:
    #
    #    Post.references()   # => raises an error
    #    Post.references([]) # => does not raise an error
    #
    # This particular method should be called with a method_name and the args
    # passed into that method as an input. For example:
    #
    # def references(*args)
1104
    #   check_if_method_has_arguments!("references", args)
1105 1106
    #   ...
    # end
1107
    def check_if_method_has_arguments!(method_name, args)
1108 1109 1110 1111
      if args.blank?
        raise ArgumentError, "The method .#{method_name}() must contain arguments."
      end
    end
1112 1113
  end
end