query_methods.rb 40.2 KB
Newer Older
1
require "active_record/relation/from_clause"
S
Sean Griffin 已提交
2
require "active_record/relation/query_attribute"
S
Sean Griffin 已提交
3
require "active_record/relation/where_clause"
4
require "active_record/relation/where_clause_factory"
5 6
require "active_model/forbidden_attributes_protection"
require "active_support/core_ext/string/filters"
7

8 9
module ActiveRecord
  module QueryMethods
10 11
    extend ActiveSupport::Concern

12 13
    include ActiveModel::ForbiddenAttributesProtection

14
    # WhereChain objects act as placeholder for queries in which #where does not have any parameter.
15
    # In this case, #where must be chained with #not to return a new relation.
16
    class WhereChain
17 18
      include ActiveModel::ForbiddenAttributesProtection

19 20 21 22
      def initialize(scope)
        @scope = scope
      end

23 24
      # Returns a new relation expressing WHERE + NOT condition according to
      # the conditions in the arguments.
25
      #
26
      # #not accepts conditions as a string, array, or hash. See QueryMethods#where for
27
      # more details on each format.
28 29
      #
      #    User.where.not("name = 'Jon'")
30
      #    # SELECT * FROM users WHERE NOT (name = 'Jon')
31 32
      #
      #    User.where.not(["name = ?", "Jon"])
33
      #    # SELECT * FROM users WHERE NOT (name = 'Jon')
34
      #
35
      #    User.where.not(name: "Jon")
36
      #    # SELECT * FROM users WHERE name != 'Jon'
37 38 39 40
      #
      #    User.where.not(name: nil)
      #    # SELECT * FROM users WHERE name IS NOT NULL
      #
41
      #    User.where.not(name: %w(Ko1 Nobu))
42
      #    # SELECT * FROM users WHERE name NOT IN ('Ko1', 'Nobu')
43 44 45
      #
      #    User.where.not(name: "Jon", role: "admin")
      #    # SELECT * FROM users WHERE name != 'Jon' AND role != 'admin'
46
      def not(opts, *rest)
47 48
        opts = sanitize_forbidden_attributes(opts)

49
        where_clause = @scope.send(:where_clause_factory).build(opts, rest)
50 51

        @scope.references!(PredicateBuilder.references(opts)) if Hash === opts
52
        @scope.where_clause += where_clause.invert
53 54 55 56
        @scope
      end
    end

57
    FROZEN_EMPTY_ARRAY = [].freeze
58
    FROZEN_EMPTY_HASH = {}.freeze
59

60
    Relation::VALUE_METHODS.each do |name|
61 62 63 64 65 66
      method_name = \
        case name
        when *Relation::MULTI_VALUE_METHODS then "#{name}_values"
        when *Relation::SINGLE_VALUE_METHODS then "#{name}_value"
        when *Relation::CLAUSE_METHODS then "#{name}_clause"
        end
67
      class_eval <<-CODE, __FILE__, __LINE__ + 1
68 69
        def #{method_name}                   # def includes_values
          get_value(#{name.inspect})         #   get_value(:includes)
70
        end                                  # end
71

72 73
        def #{method_name}=(value)           # def includes_values=(value)
          set_value(#{name.inspect}, value)  #   set_value(:includes, value)
74 75
        end                                  # end
      CODE
76 77
    end

78
    def bound_attributes
79
      if limit_value && !string_containing_comma?(limit_value)
80
        limit_bind = Attribute.with_cast_value(
81 82
          "LIMIT".freeze,
          connection.sanitize_limit(limit_value),
83
          Type.default_value,
84 85 86
        )
      end
      if offset_value
87
        offset_bind = Attribute.with_cast_value(
88 89
          "OFFSET".freeze,
          offset_value.to_i,
90
          Type.default_value,
91 92
        )
      end
93 94 95 96 97 98 99 100
      connection.combine_bind_parameters(
        from_clause: from_clause.binds,
        join_clause: arel.bind_values,
        where_clause: where_clause.binds,
        having_clause: having_clause.binds,
        limit: limit_bind,
        offset: offset_bind,
      )
S
Sean Griffin 已提交
101 102
    end

103
    alias extensions extending_values
104

O
Oscar Del Ben 已提交
105 106 107 108 109 110 111 112 113 114
    # 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
115
    # performance improvement over a simple join.
116
    #
117 118 119 120 121 122 123 124
    # 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])
    #
125 126 127 128 129 130 131 132 133 134
    # === 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)
135
    #
136
    # Note that #includes works with association names while #references needs
137
    # the actual table name.
138
    def includes(*args)
139
      check_if_method_has_arguments!(:includes, args)
140
      spawn.includes!(*args)
141
    end
142

J
Jon Leighton 已提交
143
    def includes!(*args) # :nodoc:
144 145
      args.reject!(&:blank?)
      args.flatten!
A
Aaron Patterson 已提交
146

147
      self.includes_values |= args
148
      self
149
    end
150

151 152 153
    # Forces eager loading by performing a LEFT OUTER JOIN on +args+:
    #
    #   User.eager_load(:posts)
154 155 156
    #   # SELECT "users"."id" AS t0_r0, "users"."name" AS t0_r1, ...
    #   # FROM "users" LEFT OUTER JOIN "posts" ON "posts"."user_id" =
    #   # "users"."id"
157
    def eager_load(*args)
158
      check_if_method_has_arguments!(:eager_load, args)
159
      spawn.eager_load!(*args)
160
    end
161

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

167
    # Allows preloading of +args+, in the same way that #includes does:
168 169
    #
    #   User.preload(:posts)
170
    #   # SELECT "posts".* FROM "posts" WHERE "posts"."user_id" IN (1, 2, 3)
171
    def preload(*args)
172
      check_if_method_has_arguments!(:preload, args)
173
      spawn.preload!(*args)
174
    end
175

J
Jon Leighton 已提交
176
    def preload!(*args) # :nodoc:
177 178
      self.preload_values += args
      self
179
    end
180

181 182
    # Use to indicate that the given +table_names+ are referenced by an SQL string,
    # and should therefore be JOINed in any query rather than loaded separately.
183
    # This method only works in conjunction with #includes.
184
    # See #includes for more details.
185 186
    #
    #   User.includes(:posts).where("posts.name = 'foo'")
187
    #   # Doesn't JOIN the posts table, resulting in an error.
188 189
    #
    #   User.includes(:posts).where("posts.name = 'foo'").references(:posts)
190
    #   # Query now knows the string references posts, so adds a JOIN
191 192 193
    def references(*table_names)
      check_if_method_has_arguments!(:references, table_names)
      spawn.references!(*table_names)
194
    end
195

196 197 198
    def references!(*table_names) # :nodoc:
      table_names.flatten!
      table_names.map!(&:to_s)
199

200
      self.references_values |= table_names
201
      self
202 203
    end

204
    # Works in two unique ways.
205
    #
206
    # First: takes a block so it can be used just like +Array#select+.
207
    #
208
    #   Model.all.select { |m| m.field == value }
209 210
    #
    # This will build an array of objects from the database for the scope,
211
    # converting them into an array and iterating through them using +Array#select+.
212 213
    #
    # Second: Modifies the SELECT statement for the query so that only certain
V
Vijay Dev 已提交
214
    # fields are retrieved:
215
    #
216
    #   Model.select(:field)
217
    #   # => [#<Model id: nil, field: "value">]
218 219
    #
    # Although in the above example it looks as though this method returns an
V
Vijay Dev 已提交
220
    # array, it actually returns a relation object and can have other query
221 222
    # methods appended to it, such as the other methods in ActiveRecord::QueryMethods.
    #
223
    # The argument to the method can also be an array of fields.
224
    #
225
    #   Model.select(:field, :other_field, :and_one_more)
226
    #   # => [#<Model id: nil, field: "value", other_field: "value", and_one_more: "value">]
227
    #
228 229 230
    # 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')
231
    #   # => [#<Model id: nil, field: "value", other_field: "value">]
232 233 234 235 236 237
    #
    # If an alias was specified, it will be accessible from the resulting objects:
    #
    #   Model.select('field AS field_one').first.field_one
    #   # => "value"
    #
238
    # Accessing attributes of an object that do not have fields retrieved by a select
239
    # except +id+ will throw ActiveModel::MissingAttributeError:
240
    #
241 242 243
    #   Model.select(:field).first.other_field
    #   # => ActiveModel::MissingAttributeError: missing attribute: other_field
    def select(*fields)
244 245 246 247 248 249 250 251 252 253
      if block_given?
        if fields.any?
          ActiveSupport::Deprecation.warn(<<-WARNING.squish)
            When select is called with a block, it ignores other arguments. This behavior is now deprecated and will result in an ArgumentError in Rails 5.1. You can safely remove the arguments to resolve the deprecation warning because they do not have any effect on the output of the call to the select method with a block.
          WARNING
        end

        return super()
      end

254
      raise ArgumentError, "Call this with at least one field" if fields.empty?
255
      spawn._select!(*fields)
256 257
    end

258
    def _select!(*fields) # :nodoc:
259
      fields.flatten!
260
      fields.map! do |field|
261
        klass.attribute_alias?(field) ? klass.attribute_alias(field).to_sym : field
262
      end
263
      self.select_values += fields
264
      self
265
    end
S
Santiago Pastorino 已提交
266

O
Oscar Del Ben 已提交
267 268 269
    # Allows to specify a group attribute:
    #
    #   User.group(:name)
270
    #   # SELECT "users".* FROM "users" GROUP BY name
O
Oscar Del Ben 已提交
271
    #
272
    # Returns an array with distinct records based on the +group+ attribute:
O
Oscar Del Ben 已提交
273 274
    #
    #   User.select([:id, :name])
275
    #   # => [#<User id: 1, name: "Oscar">, #<User id: 2, name: "Oscar">, #<User id: 3, name: "Foo">]
O
Oscar Del Ben 已提交
276 277
    #
    #   User.group(:name)
278
    #   # => [#<User id: 3, name: "Foo", ...>, #<User id: 2, name: "Oscar", ...>]
279 280
    #
    #   User.group('name AS grouped_name, age')
281
    #   # => [#<User id: 3, name: "Foo", age: 21, ...>, #<User id: 2, name: "Oscar", age: 21, ...>, #<User id: 5, name: "Foo", age: 23, ...>]
282 283
    #
    # Passing in an array of attributes to group by is also supported.
284
    #
285
    #   User.select([:id, :first_name]).group(:id, :first_name).first(3)
286
    #   # => [#<User id: 1, first_name: "Bill">, #<User id: 2, first_name: "Earl">, #<User id: 3, first_name: "Beto">]
287
    def group(*args)
288
      check_if_method_has_arguments!(:group, args)
289
      spawn.group!(*args)
290
    end
291

J
Jon Leighton 已提交
292
    def group!(*args) # :nodoc:
293 294 295
      args.flatten!

      self.group_values += args
296
      self
297
    end
298

O
Oscar Del Ben 已提交
299 300
    # Allows to specify an order attribute:
    #
301
    #   User.order(:name)
302
    #   # SELECT "users".* FROM "users" ORDER BY "users"."name" ASC
303
    #
304
    #   User.order(email: :desc)
305
    #   # SELECT "users".* FROM "users" ORDER BY "users"."email" DESC
306
    #
307
    #   User.order(:name, email: :desc)
308
    #   # SELECT "users".* FROM "users" ORDER BY "users"."name" ASC, "users"."email" DESC
309 310
    #
    #   User.order('name')
311
    #   # SELECT "users".* FROM "users" ORDER BY name
312 313
    #
    #   User.order('name DESC')
314
    #   # SELECT "users".* FROM "users" ORDER BY name DESC
315 316
    #
    #   User.order('name DESC, email')
317
    #   # SELECT "users".* FROM "users" ORDER BY name DESC, email
318
    def order(*args)
319
      check_if_method_has_arguments!(:order, args)
320
      spawn.order!(*args)
321
    end
322

J
Jon Leighton 已提交
323
    def order!(*args) # :nodoc:
324
      preprocess_order_args(args)
325

326
      self.order_values += args
327
      self
328
    end
329

330 331 332 333 334 335 336 337
    # 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')
    #
338
    # generates a query with 'ORDER BY id ASC, name ASC'.
S
Sebastian Martinez 已提交
339
    def reorder(*args)
340
      check_if_method_has_arguments!(:reorder, args)
341
      spawn.reorder!(*args)
342
    end
343

J
Jon Leighton 已提交
344
    def reorder!(*args) # :nodoc:
345
      preprocess_order_args(args)
346

347
      self.reordering_value = true
348
      self.order_values = args
349
      self
S
Sebastian Martinez 已提交
350 351
    end

352 353 354 355 356 357 358 359
    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.
    #
360
    #   User.order('email DESC').unscope(:order) == User.all
361 362 363 364 365
    #
    # 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:
    #
366
    #   User.order('email DESC').select('id').where(name: "John")
367 368
    #       .unscope(:order, :select, :where) == User.all
    #
369
    # One can additionally pass a hash as an argument to unscope specific +:where+ values.
370
    # This is done by passing a hash with a single key-value pair. The key should be
371
    # +:where+ and the value should be the where value to unscope. For example:
372
    #
373 374
    #   User.where(name: "John", active: true).unscope(where: :name)
    #       == User.where(active: true)
375
    #
376 377
    # This method is similar to #except, but unlike
    # #except, it persists across merges:
378
    #
J
Jon Leighton 已提交
379 380
    #   User.order('email').merge(User.except(:order))
    #       == User.order('email')
381
    #
J
Jon Leighton 已提交
382 383 384 385 386
    #   User.order('email').merge(User.unscope(:order))
    #       == User.all
    #
    # This means it can be used in association definitions:
    #
R
Ryuta Kamizono 已提交
387
    #   has_many :comments, -> { unscope(where: :trashed) }
388 389
    #
    def unscope(*args)
390
      check_if_method_has_arguments!(:unscope, args)
391 392 393
      spawn.unscope!(*args)
    end

394
    def unscope!(*args) # :nodoc:
395
      args.flatten!
J
Jon Leighton 已提交
396
      self.unscope_values += args
397 398 399 400

      args.each do |scope|
        case scope
        when Symbol
401 402 403 404
          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
          set_value(scope, nil)
405 406 407 408 409 410
        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

411 412
            target_values = Array(target_value).map(&:to_s)
            self.where_clause = where_clause.except(*target_values)
413 414 415 416 417 418 419 420 421
          end
        else
          raise ArgumentError, "Unrecognized scoping: #{args.inspect}. Use .unscope(where: :attribute_name) or .unscope(:order), for example."
        end
      end

      self
    end

422 423
    # Performs a joins on +args+. The given symbol(s) should match the name of
    # the association(s).
424 425
    #
    #   User.joins(:posts)
426 427 428 429 430 431 432 433 434 435 436 437 438 439 440 441 442 443 444 445
    #   # SELECT "users".*
    #   # FROM "users"
    #   # INNER JOIN "posts" ON "posts"."user_id" = "users"."id"
    #
    # Multiple joins:
    #
    #   User.joins(:posts, :account)
    #   # SELECT "users".*
    #   # FROM "users"
    #   # INNER JOIN "posts" ON "posts"."user_id" = "users"."id"
    #   # INNER JOIN "accounts" ON "accounts"."id" = "users"."account_id"
    #
    # Nested joins:
    #
    #   User.joins(posts: [:comments])
    #   # SELECT "users".*
    #   # FROM "users"
    #   # INNER JOIN "posts" ON "posts"."user_id" = "users"."id"
    #   # INNER JOIN "comments" "comments_posts"
    #   #   ON "comments_posts"."post_id" = "posts"."id"
446 447 448 449
    #
    # 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")
450
    #   # SELECT "users".* FROM "users" LEFT JOIN bookmarks ON bookmarks.bookmarkable_type = 'Post' AND bookmarks.user_id = users.id
451
    def joins(*args)
452 453
      check_if_method_has_arguments!(:joins, args)
      spawn.joins!(*args)
454
    end
455

J
Jon Leighton 已提交
456
    def joins!(*args) # :nodoc:
457 458
      args.compact!
      args.flatten!
459 460
      self.joins_values += args
      self
P
Pratik Naik 已提交
461 462
    end

463 464 465 466 467 468 469 470 471 472 473 474 475
    # Performs a left outer joins on +args+:
    #
    #   User.left_outer_joins(:posts)
    #   => SELECT "users".* FROM "users" LEFT OUTER JOIN "posts" ON "posts"."user_id" = "users"."id"
    #
    def left_outer_joins(*args)
      check_if_method_has_arguments!(:left_outer_joins, args)

      args.compact!
      args.flatten!

      spawn.left_outer_joins!(*args)
    end
476
    alias :left_joins :left_outer_joins
477 478 479 480 481 482

    def left_outer_joins!(*args) # :nodoc:
      self.left_outer_joins_values += args
      self
    end

483 484 485 486 487 488 489 490 491 492
    # 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
493
    # constructor as an SQL fragment, and used in the where clause of the query.
494 495 496 497 498 499 500 501 502 503 504 505 506 507 508 509 510 511 512 513 514 515 516 517 518 519 520 521 522 523 524 525
    #
    #    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,
526
    # the values are inserted using the same escapes as the Ruby core method +Kernel::sprintf+.
527 528 529 530 531 532 533 534 535 536 537 538 539 540 541 542 543 544 545 546 547 548 549 550 551 552 553 554 555 556 557
    #
    #   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')
    #
558 559 560 561 562 563
    # 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 已提交
564 565
    #    Post.where(author: author)
    #    Post.where(author_id: author)
566 567 568
    #
    # This also works with polymorphic belongs_to relationships:
    #
A
AvnerCohen 已提交
569 570
    #    treasure = Treasure.create(name: 'gold coins')
    #    treasure.price_estimates << PriceEstimate.create(price: 125)
571 572
    #
    #    # The following queries will be equivalent:
A
AvnerCohen 已提交
573 574
    #    PriceEstimate.where(estimate_of: treasure)
    #    PriceEstimate.where(estimate_of_type: 'Treasure', estimate_of_id: treasure)
575
    #
576 577 578 579 580 581 582 583 584 585
    # === 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 已提交
586
    #    User.joins(:posts).where({ posts: { published: true } })
587
    #
588
    # === no argument
589
    #
590 591
    # 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.
592 593
    #
    #    User.where.not(name: "Jon")
594
    #    # SELECT * FROM users WHERE name != 'Jon'
595
    #
596
    # See WhereChain for more details on #not.
597
    #
598
    # === blank condition
599
    #
600
    # If the condition is any blank-ish object, then #where is a no-op and returns
601
    # the current relation.
602
    def where(opts = :chain, *rest)
603
      if :chain == opts
604 605 606 607 608 609
        WhereChain.new(spawn)
      elsif opts.blank?
        self
      else
        spawn.where!(opts, *rest)
      end
610 611
    end

612
    def where!(opts, *rest) # :nodoc:
613 614
      opts = sanitize_forbidden_attributes(opts)
      references!(PredicateBuilder.references(opts)) if Hash === opts
615
      self.where_clause += where_clause_factory.build(opts, rest)
616
      self
617
    end
P
Pratik Naik 已提交
618

619 620
    # Allows you to change a previously set where condition for a given attribute, instead of appending to that condition.
    #
621 622 623 624 625 626 627 628
    #   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
629
    #
630 631
    # This is short-hand for <tt>unscope(where: conditions.keys).where(conditions)</tt>.
    # Note that unlike reorder, we're only unscoping the named conditions -- not the entire where statement.
632 633 634 635
    def rewhere(conditions)
      unscope(where: conditions.keys).where(conditions)
    end

636 637 638 639
    # Returns a new relation, which is the logical union of this relation and the one passed as an
    # argument.
    #
    # The two relations must be structurally compatible: they must be scoping the same model, and
640 641
    # they must differ only by #where (if no #group has been defined) or #having (if a #group is
    # present). Neither relation may have a #limit, #offset, or #distinct set.
642
    #
643
    #    Post.where("id = 1").or(Post.where("author_id = 3"))
R
Ryuta Kamizono 已提交
644
    #    # SELECT `posts`.* FROM `posts` WHERE ((id = 1) OR (author_id = 3))
645 646
    #
    def or(other)
647 648 649 650
      unless other.is_a? Relation
        raise ArgumentError, "You have passed #{other.class.name} object to #or. Pass an ActiveRecord::Relation object instead."
      end

651 652 653
      spawn.or!(other)
    end

654
    def or!(other) # :nodoc:
655 656 657 658
      incompatible_values = structurally_incompatible_values_for_or(other)

      unless incompatible_values.empty?
        raise ArgumentError, "Relation passed to #or must be structurally compatible. Incompatible values: #{incompatible_values}"
659 660
      end

661 662
      self.where_clause = self.where_clause.or(other.where_clause)
      self.having_clause = self.having_clause.or(other.having_clause)
663 664 665 666

      self
    end

667 668 669 670
    # 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')
671
    def having(opts, *rest)
672
      opts.blank? ? self : spawn.having!(opts, *rest)
673 674
    end

J
Jon Leighton 已提交
675
    def having!(opts, *rest) # :nodoc:
676
      opts = sanitize_forbidden_attributes(opts)
677
      references!(PredicateBuilder.references(opts)) if Hash === opts
678

679
      self.having_clause += having_clause_factory.build(opts, rest)
680
      self
681 682
    end

683
    # Specifies a limit for the number of records to retrieve.
684 685 686 687
    #
    #   User.limit(10) # generated SQL has 'LIMIT 10'
    #
    #   User.limit(10).limit(20) # generated SQL has 'LIMIT 20'
688
    def limit(value)
J
Jon Leighton 已提交
689
      spawn.limit!(value)
690 691
    end

J
Jon Leighton 已提交
692
    def limit!(value) # :nodoc:
693 694
      if string_containing_comma?(value)
        # Remove `string_containing_comma?` when removing this deprecation
695
        ActiveSupport::Deprecation.warn(<<-WARNING.squish)
S
Sean Griffin 已提交
696 697 698 699
          Passing a string to limit in the form "1,2" is deprecated and will be
          removed in Rails 5.1. Please call `offset` explicitly instead.
        WARNING
      end
700 701
      self.limit_value = value
      self
702 703
    end

704 705 706 707
    # Specifies the number of rows to skip before returning rows.
    #
    #   User.offset(10) # generated SQL has "OFFSET 10"
    #
708
    # Should be used with order.
709
    #
710
    #   User.offset(10).order("name ASC")
711
    def offset(value)
J
Jon Leighton 已提交
712
      spawn.offset!(value)
713 714
    end

J
Jon Leighton 已提交
715
    def offset!(value) # :nodoc:
716 717
      self.offset_value = value
      self
718 719
    end

720
    # Specifies locking settings (default to +true+). For more information
721
    # on locking, please see ActiveRecord::Locking.
722
    def lock(locks = true)
J
Jon Leighton 已提交
723
      spawn.lock!(locks)
724
    end
725

J
Jon Leighton 已提交
726
    def lock!(locks = true) # :nodoc:
727
      case locks
728
      when String, TrueClass, NilClass
729
        self.lock_value = locks || true
730
      else
731
        self.lock_value = false
732
      end
733

734
      self
735 736
    end

737
    # Returns a chainable relation with zero records.
738
    #
739 740 741
    # 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.
742 743 744 745
    #
    # Any subsequent condition chained to the returned relation will continue
    # generating an empty relation and will not fire any query to the database.
    #
746 747
    # Used in cases where a method or scope could return zero records but the
    # result needs to be chainable.
748 749 750
    #
    # For example:
    #
A
AvnerCohen 已提交
751
    #   @posts = current_user.visible_posts.where(name: params[:name])
752
    #   # the visible_posts method is expected to return a chainable Relation
753 754 755
    #
    #   def visible_posts
    #     case role
756
    #     when 'Country Manager'
A
AvnerCohen 已提交
757
    #       Post.where(country: country)
758
    #     when 'Reviewer'
759
    #       Post.published
760
    #     when 'Bad User'
761
    #       Post.none # It can't be chained if [] is returned.
762 763 764 765
    #     end
    #   end
    #
    def none
766
      spawn.none!
767 768
    end

J
Jon Leighton 已提交
769
    def none! # :nodoc:
770
      where!("1=0").extending!(NullRelation)
771 772
    end

773 774 775 776 777
    # 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
778
    #   => ActiveRecord::ReadOnlyRecord: User is marked as readonly
779
    def readonly(value = true)
J
Jon Leighton 已提交
780
      spawn.readonly!(value)
781 782
    end

J
Jon Leighton 已提交
783
    def readonly!(value = true) # :nodoc:
784 785
      self.readonly_value = value
      self
786 787
    end

788 789 790 791 792 793 794 795 796
    # 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'
    #
797
    # You can pass +nil+ to #create_with to reset attributes:
798 799 800
    #
    #   users = users.create_with(nil)
    #   users.new.name # => 'Oscar'
801
    def create_with(value)
J
Jon Leighton 已提交
802
      spawn.create_with!(value)
803 804
    end

J
Jon Leighton 已提交
805
    def create_with!(value) # :nodoc:
806 807 808 809 810 811 812
      if value
        value = sanitize_forbidden_attributes(value)
        self.create_with_value = create_with_value.merge(value)
      else
        self.create_with_value = {}
      end

813
      self
814 815
    end

816 817 818
    # Specifies table from which the records will be fetched. For example:
    #
    #   Topic.select('title').from('posts')
819
    #   # SELECT title FROM posts
820 821 822
    #
    # Can accept other relation objects. For example:
    #
823
    #   Topic.select('title').from(Topic.approved)
824
    #   # SELECT title FROM (SELECT * FROM topics WHERE approved = 't') subquery
825
    #
826
    #   Topic.select('a.title').from(Topic.approved, :a)
827
    #   # SELECT a.title FROM (SELECT * FROM topics WHERE approved = 't') a
828 829 830
    #
    def from(value, subquery_name = nil)
      spawn.from!(value, subquery_name)
831 832
    end

J
Jon Leighton 已提交
833
    def from!(value, subquery_name = nil) # :nodoc:
834
      self.from_clause = Relation::FromClause.new(value, subquery_name)
835
      self
836 837
    end

838 839 840
    # Specifies whether the records should be unique or not. For example:
    #
    #   User.select(:name)
841
    #   # Might return two records with the same name
842
    #
843
    #   User.select(:name).distinct
844
    #   # Returns 1 record per distinct name
845
    #
846
    #   User.select(:name).distinct.distinct(false)
847
    #   # You can also remove the uniqueness
848 849
    def distinct(value = true)
      spawn.distinct!(value)
850
    end
851
    alias uniq distinct
852
    deprecate uniq: :distinct
853

854 855 856
    # Like #distinct, but modifies relation in place.
    def distinct!(value = true) # :nodoc:
      self.distinct_value = value
857
      self
858
    end
859
    alias uniq! distinct!
860
    deprecate uniq!: :distinct!
861

862
    # Used to extend a scope with additional methods, either through
863 864
    # a module or through a block provided.
    #
865 866 867 868 869 870 871 872 873 874
    # 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
    #
875
    #   scope = Model.all.extending(Pagination)
876 877
    #   scope.page(params[:page])
    #
V
Vijay Dev 已提交
878
    # You can also pass a list of modules:
879
    #
880
    #   scope = Model.all.extending(Pagination, SomethingElse)
881 882 883
    #
    # === Using a block
    #
884
    #   scope = Model.all.extending do
885
    #     def page(number)
886
    #       # pagination code goes here
887 888 889 890 891 892
    #     end
    #   end
    #   scope.page(params[:page])
    #
    # You can also use a block and a module list:
    #
893
    #   scope = Model.all.extending(Pagination) do
894
    #     def per_page(number)
895
    #       # pagination code goes here
896 897
    #     end
    #   end
898 899
    def extending(*modules, &block)
      if modules.any? || block
J
Jon Leighton 已提交
900
        spawn.extending!(*modules, &block)
901 902 903 904
      else
        self
      end
    end
905

J
Jon Leighton 已提交
906
    def extending!(*modules, &block) # :nodoc:
907 908
      modules << Module.new(&block) if block
      modules.flatten!
909

910
      self.extending_values += modules
911
      extend(*extending_values) if extending_values.any?
912

913
      self
914 915
    end

916 917 918
    # Reverse the existing order clause on the relation.
    #
    #   User.order('name ASC').reverse_order # generated SQL has 'ORDER BY name DESC'
919
    def reverse_order
J
Jon Leighton 已提交
920
      spawn.reverse_order!
921 922
    end

J
Jon Leighton 已提交
923
    def reverse_order! # :nodoc:
924 925 926
      orders = order_values.uniq
      orders.reject!(&:blank?)
      self.order_values = reverse_sql_order(orders)
927
      self
928 929
    end

930
    # Returns the Arel object associated with the relation.
931
    def arel # :nodoc:
932
      @arel ||= build_arel
933 934
    end

935 936 937 938 939 940 941 942 943 944 945
    # Returns a relation value with a given name
    def get_value(name) # :nodoc:
      @values[name] || default_value_for(name)
    end

    # Sets the relation value with the given name
    def set_value(name, value) # :nodoc:
      assert_mutability!
      @values[name] = value
    end

946 947
    private

948 949 950 951
      def assert_mutability!
        raise ImmutableRelation if @loaded
        raise ImmutableRelation if defined?(@arel) && @arel
      end
952

953 954
      def build_arel
        arel = Arel::SelectManager.new(table)
955

956 957
        build_joins(arel, joins_values.flatten) unless joins_values.empty?
        build_left_outer_joins(arel, left_outer_joins_values.flatten) unless left_outer_joins_values.empty?
958

959 960 961 962 963 964 965 966
        arel.where(where_clause.ast) unless where_clause.empty?
        arel.having(having_clause.ast) unless having_clause.empty?
        if limit_value
          if string_containing_comma?(limit_value)
            arel.take(connection.sanitize_limit(limit_value))
          else
            arel.take(Arel::Nodes::BindParam.new)
          end
967
        end
968 969
        arel.skip(Arel::Nodes::BindParam.new) if offset_value
        arel.group(*arel_columns(group_values.uniq.reject(&:blank?))) unless group_values.empty?
970

971
        build_order(arel)
972

973
        build_select(arel)
974

975 976 977
        arel.distinct(distinct_value)
        arel.from(build_from) unless from_clause.empty?
        arel.lock(lock_value) if lock_value
978

979
        arel
980 981
      end

982 983 984 985 986 987 988
      def build_from
        opts = from_clause.value
        name = from_clause.name
        case opts
        when Relation
          name ||= "subquery"
          opts.arel.as(name.to_s)
989
        else
990
          opts
991 992 993
        end
      end

994 995 996 997 998 999 1000 1001
      def build_left_outer_joins(manager, outer_joins)
        buckets = outer_joins.group_by do |join|
          case join
          when Hash, Symbol, Array
            :association_join
          else
            raise ArgumentError, "only Hash, Symbol and Array are allowed"
          end
A
Aaron Patterson 已提交
1002
        end
1003 1004

        build_join_query(manager, buckets, Arel::Nodes::OuterJoin)
1005
      end
1006

1007 1008 1009 1010 1011 1012 1013 1014 1015 1016 1017 1018 1019 1020 1021
      def build_joins(manager, joins)
        buckets = joins.group_by do |join|
          case join
          when String
            :string_join
          when Hash, Symbol, Array
            :association_join
          when ActiveRecord::Associations::JoinDependency
            :stashed_join
          when Arel::Nodes::Join
            :join_node
          else
            raise "unknown class: %s" % join.class.name
          end
        end
1022

1023 1024
        build_join_query(manager, buckets, Arel::Nodes::InnerJoin)
      end
1025

1026 1027
      def build_join_query(manager, buckets, join_type)
        buckets.default = []
1028

1029 1030 1031 1032
        association_joins         = buckets[:association_join]
        stashed_association_joins = buckets[:stashed_join]
        join_nodes                = buckets[:join_node].uniq
        string_joins              = buckets[:string_join].map(&:strip).uniq
1033

1034
        join_list = join_nodes + convert_join_strings_to_ast(manager, string_joins)
1035

1036 1037 1038 1039 1040
        join_dependency = ActiveRecord::Associations::JoinDependency.new(
          @klass,
          association_joins,
          join_list
        )
1041

1042
        join_infos = join_dependency.join_constraints stashed_association_joins, join_type
1043

1044 1045 1046 1047
        join_infos.each do |info|
          info.joins.each { |join| manager.from(join) }
          manager.bind_values.concat info.binds
        end
1048

1049
        manager.join_sources.concat(join_list)
1050

1051 1052
        manager
      end
1053

1054 1055 1056 1057 1058
      def convert_join_strings_to_ast(table, joins)
        joins
          .flatten
          .reject(&:blank?)
          .map { |join| table.create_string_join(Arel.sql(join)) }
1059 1060
      end

1061 1062 1063
      def build_select(arel)
        if select_values.any?
          arel.project(*arel_columns(select_values.uniq))
1064
        else
1065
          arel.project(@klass.arel_table[Arel.star])
1066 1067 1068
        end
      end

1069 1070 1071 1072 1073 1074 1075 1076 1077 1078
      def arel_columns(columns)
        columns.map do |field|
          if (Symbol === field || String === field) && (klass.has_attribute?(field) || klass.attribute_alias?(field)) && !from_clause.value
            arel_attribute(field)
          elsif Symbol === field
            connection.quote_table_name(field.to_s)
          else
            field
          end
        end
1079
      end
B
Brian Mathiyakom 已提交
1080

1081 1082 1083 1084 1085 1086 1087 1088 1089 1090 1091 1092 1093 1094 1095 1096 1097 1098 1099 1100 1101 1102 1103
      def reverse_sql_order(order_query)
        if order_query.empty?
          return [arel_attribute(primary_key).desc] if primary_key
          raise IrreversibleOrderError,
            "Relation has no current order and table has no primary key to be used as default order"
        end

        order_query.flat_map do |o|
          case o
          when Arel::Attribute
            o.desc
          when Arel::Nodes::Ordering
            o.reverse
          when String
            if does_not_support_reverse?(o)
              raise IrreversibleOrderError, "Order #{o.inspect} can not be reversed automatically"
            end
            o.split(",").map! do |s|
              s.strip!
              s.gsub!(/\sasc\Z/i, " DESC") || s.gsub!(/\sdesc\Z/i, " ASC") || s.concat(" DESC")
            end
          else
            o
1104
          end
1105
        end
1106
      end
1107

1108 1109
      def does_not_support_reverse?(order)
        # Uses SQL function with multiple arguments.
R
Rafael Mendonça França 已提交
1110 1111 1112
        (order.include?(",") && order.split(",").find { |section| section.count("(") != section.count(")") }) ||
          # Uses "nulls first" like construction.
          /nulls (first|last)\Z/i.match?(order)
1113
      end
1114

1115 1116 1117
      def build_order(arel)
        orders = order_values.uniq
        orders.reject!(&:blank?)
1118

1119 1120
        arel.order(*orders) unless orders.empty?
      end
1121

1122 1123
      VALID_DIRECTIONS = [:asc, :desc, :ASC, :DESC,
                          "asc", "desc", "ASC", "DESC"] # :nodoc:
1124

1125 1126 1127 1128 1129 1130 1131
      def validate_order_args(args)
        args.each do |arg|
          next unless arg.is_a?(Hash)
          arg.each do |_key, value|
            raise ArgumentError, "Direction \"#{value}\" is invalid. Valid " \
                                 "directions are: #{VALID_DIRECTIONS.inspect}" unless VALID_DIRECTIONS.include?(value)
          end
1132 1133
        end
      end
1134

1135 1136 1137
      def preprocess_order_args(order_args)
        order_args.map! do |arg|
          klass.send(:sanitize_sql_for_order, arg)
1138
        end
1139 1140 1141 1142 1143 1144 1145 1146 1147 1148 1149 1150 1151 1152 1153 1154 1155 1156 1157 1158 1159
        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|
          case arg
          when Symbol
            arel_attribute(arg).asc
          when Hash
            arg.map { |field, dir|
              arel_attribute(field).send(dir.downcase)
            }
          else
            arg
          end
        end.flatten!
      end
1160

1161 1162 1163 1164 1165 1166 1167 1168 1169 1170 1171 1172 1173 1174 1175 1176
      # 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)
      #   check_if_method_has_arguments!("references", args)
      #   ...
      # end
1177 1178 1179 1180
      def check_if_method_has_arguments!(method_name, args)
        if args.blank?
          raise ArgumentError, "The method .#{method_name}() must contain arguments."
        end
1181
      end
S
Sean Griffin 已提交
1182

1183
      STRUCTURAL_OR_METHODS = Relation::VALUE_METHODS - [:extending, :where, :having]
1184
      def structurally_incompatible_values_for_or(other)
1185 1186 1187
        STRUCTURAL_OR_METHODS.reject do |method|
          get_value(method) == other.get_value(method)
        end
1188
      end
1189

1190 1191 1192 1193
      def where_clause_factory
        @where_clause_factory ||= Relation::WhereClauseFactory.new(klass, predicate_builder)
      end
      alias having_clause_factory where_clause_factory
1194

1195 1196 1197
      def string_containing_comma?(value)
        ::String === value && value.include?(",")
      end
1198 1199 1200 1201 1202 1203 1204 1205 1206 1207 1208 1209 1210 1211 1212 1213 1214 1215 1216

      def default_value_for(name)
        case name
        when :create_with
          FROZEN_EMPTY_HASH
        when :readonly
          false
        when :where, :having
          Relation::WhereClause.empty
        when :from
          Relation::FromClause.empty
        when *Relation::MULTI_VALUE_METHODS
          FROZEN_EMPTY_ARRAY
        when *Relation::SINGLE_VALUE_METHODS
          nil
        else
          raise ArgumentError, "unknown relation value #{name.inspect}"
        end
      end
1217 1218
  end
end