activerecord 7.1.4 → 7.2.0.beta1

data/lib/active_record/relation.rb

@@ -3,6 +3,54 @@
 module ActiveRecord
   # = Active Record \Relation
   class Relation
+    class ExplainProxy  # :nodoc:
+      def initialize(relation, options)
+        @relation = relation
+        @options  = options
+      end
+
+      def inspect
+        exec_explain { @relation.send(:exec_queries) }
+      end
+
+      def average(column_name)
+        exec_explain { @relation.average(column_name) }
+      end
+
+      def count(column_name = nil)
+        exec_explain { @relation.count(column_name) }
+      end
+
+      def first(limit = nil)
+        exec_explain { @relation.first(limit) }
+      end
+
+      def last(limit = nil)
+        exec_explain { @relation.last(limit) }
+      end
+
+      def maximum(column_name)
+        exec_explain { @relation.maximum(column_name) }
+      end
+
+      def minimum(column_name)
+        exec_explain { @relation.minimum(column_name) }
+      end
+
+      def pluck(*column_names)
+        exec_explain { @relation.pluck(*column_names) }
+      end
+
+      def sum(identity_or_column = nil)
+        exec_explain { @relation.sum(identity_or_column) }
+      end
+
+      private
+        def exec_explain(&block)
+          @relation.exec_explain(@relation.collecting_queries_for_explain { block.call }, @options)
+        end
+    end
+
     MULTI_VALUE_METHODS  = [:includes, :eager_load, :preload, :select, :group,
                             :order, :joins, :left_outer_joins, :references,
                             :extending, :unscope, :optimizer_hints, :annotate,
@@ -12,12 +60,13 @@ module ActiveRecord
                             :reverse_order, :distinct, :create_with, :skip_query_cache]
 
     CLAUSE_METHODS = [:where, :having, :from]
-    INVALID_METHODS_FOR_DELETE_ALL = [:distinct, :with]
+    INVALID_METHODS_FOR_DELETE_ALL = [:distinct, :with, :with_recursive]
 
     VALUE_METHODS = MULTI_VALUE_METHODS + SINGLE_VALUE_METHODS + CLAUSE_METHODS
 
     include Enumerable
     include FinderMethods, Calculations, SpawnMethods, QueryMethods, Batches, Explain, Delegation
+    include SignedId::RelationMethods, TokenFor::RelationMethods
 
     attr_reader :table, :klass, :loaded, :predicate_builder
     attr_accessor :skip_preloading_value
@@ -207,18 +256,22 @@ module ActiveRecord
     #   the problem of running out of integers, if the underlying table is still stuck on a primary
     #   key of type int (note: All \Rails apps since 5.1+ have defaulted to bigint, which is not liable
     #   to this problem).
+    # * Columns with unique database constraints should not have uniqueness validations defined,
+    #   otherwise #create will fail due to validation errors and #find_by will never be called.
     #
     # This method will return a record if all given attributes are covered by unique constraints
     # (unless the INSERT -> DELETE -> SELECT race condition is triggered), but if creation was attempted
     # and failed due to validation errors it won't be persisted, you get what #create returns in
     # such situation.
     def create_or_find_by(attributes, &block)
-      transaction(requires_new: true) { create(attributes, &block) }
-    rescue ActiveRecord::RecordNotUnique
-      if connection.transaction_open?
-        where(attributes).lock.find_by!(attributes)
-      else
-        find_by!(attributes)
+      with_connection do |connection|
+        transaction(requires_new: true) { create(attributes, &block) }
+      rescue ActiveRecord::RecordNotUnique
+        if connection.transaction_open?
+          where(attributes).lock.find_by!(attributes)
+        else
+          find_by!(attributes)
+        end
       end
     end
 
@@ -226,12 +279,14 @@ module ActiveRecord
     # {create!}[rdoc-ref:Persistence::ClassMethods#create!] so an exception
     # is raised if the created record is invalid.
     def create_or_find_by!(attributes, &block)
-      transaction(requires_new: true) { create!(attributes, &block) }
-    rescue ActiveRecord::RecordNotUnique
-      if connection.transaction_open?
-        where(attributes).lock.find_by!(attributes)
-      else
-        find_by!(attributes)
+      with_connection do |connection|
+        transaction(requires_new: true) { create!(attributes, &block) }
+      rescue ActiveRecord::RecordNotUnique
+        if connection.transaction_open?
+          where(attributes).lock.find_by!(attributes)
+        else
+          find_by!(attributes)
+        end
       end
     end
 
@@ -245,13 +300,30 @@ module ActiveRecord
     # returns the result as a string. The string is formatted imitating the
     # ones printed by the database shell.
     #
+    #   User.all.explain
+    #   # EXPLAIN SELECT `users`.* FROM `users`
+    #   # ...
+    #
     # Note that this method actually runs the queries, since the results of some
     # are needed by the next ones when eager loading is going on.
     #
+    # To run EXPLAIN on queries created by +first+, +pluck+ and +count+, call
+    # these methods on +explain+:
+    #
+    #   User.all.explain.count
+    #   # EXPLAIN SELECT COUNT(*) FROM `users`
+    #   # ...
+    #
+    # The column name can be passed if required:
+    #
+    #   User.all.explain.maximum(:id)
+    #   # EXPLAIN SELECT MAX(`users`.`id`) FROM `users`
+    #   # ...
+    #
     # Please see further details in the
     # {Active Record Query Interface guide}[https://guides.rubyonrails.org/active_record_querying.html#running-explain].
     def explain(*options)
-      exec_explain(collecting_queries_for_explain { exec_queries }, options)
+      ExplainProxy.new(self, options)
     end
 
     # Converts relation objects to Array.
@@ -401,28 +473,30 @@ module ActiveRecord
       else
         collection = eager_loading? ? apply_join_dependency : self
 
-        column = connection.visitor.compile(table[timestamp_column])
-        select_values = "COUNT(*) AS #{connection.quote_column_name("size")}, MAX(%s) AS timestamp"
+        with_connection do |c|
+          column = c.visitor.compile(table[timestamp_column])
+          select_values = "COUNT(*) AS #{adapter_class.quote_column_name("size")}, MAX(%s) AS timestamp"
 
-        if collection.has_limit_or_offset?
-          query = collection.select("#{column} AS collection_cache_key_timestamp")
-          query._select!(table[Arel.star]) if distinct_value && collection.select_values.empty?
-          subquery_alias = "subquery_for_cache_key"
-          subquery_column = "#{subquery_alias}.collection_cache_key_timestamp"
-          arel = query.build_subquery(subquery_alias, select_values % subquery_column)
-        else
-          query = collection.unscope(:order)
-          query.select_values = [select_values % column]
-          arel = query.arel
-        end
+          if collection.has_limit_or_offset?
+            query = collection.select("#{column} AS collection_cache_key_timestamp")
+            query._select!(table[Arel.star]) if distinct_value && collection.select_values.empty?
+            subquery_alias = "subquery_for_cache_key"
+            subquery_column = "#{subquery_alias}.collection_cache_key_timestamp"
+            arel = query.build_subquery(subquery_alias, select_values % subquery_column)
+          else
+            query = collection.unscope(:order)
+            query.select_values = [select_values % column]
+            arel = query.arel
+          end
 
-        size, timestamp = connection.select_rows(arel, nil).first
+          size, timestamp = c.select_rows(arel, nil).first
 
-        if size
-          column_type = klass.type_for_attribute(timestamp_column)
-          timestamp = column_type.deserialize(timestamp)
-        else
-          size = 0
+          if size
+            column_type = klass.type_for_attribute(timestamp_column)
+            timestamp = column_type.deserialize(timestamp)
+          else
+            size = 0
+          end
         end
       end
 
@@ -521,18 +595,20 @@ module ActiveRecord
         values = Arel.sql(klass.sanitize_sql_for_assignment(updates, table.name))
       end
 
-      arel = eager_loading? ? apply_join_dependency.arel : build_arel
-      arel.source.left = table
+      klass.with_connection do |c|
+        arel = eager_loading? ? apply_join_dependency.arel : build_arel(c)
+        arel.source.left = table
 
-      group_values_arel_columns = arel_columns(group_values.uniq)
-      having_clause_ast = having_clause.ast unless having_clause.empty?
-      key = if klass.composite_primary_key?
-        primary_key.map { |pk| table[pk] }
-      else
-        table[primary_key]
+        group_values_arel_columns = arel_columns(group_values.uniq)
+        having_clause_ast = having_clause.ast unless having_clause.empty?
+        key = if klass.composite_primary_key?
+          primary_key.map { |pk| table[pk] }
+        else
+          table[primary_key]
+        end
+        stmt = arel.compile_update(values, key, having_clause_ast, group_values_arel_columns)
+        c.update(stmt, "#{klass} Update All").tap { reset }
       end
-      stmt = arel.compile_update(values, key, having_clause_ast, group_values_arel_columns)
-      klass.connection.update(stmt, "#{klass} Update All").tap { reset }
     end
 
     def update(id = :all, attributes) # :nodoc:
@@ -551,6 +627,283 @@ module ActiveRecord
       end
     end
 
+
+    # Inserts a single record into the database in a single SQL INSERT
+    # statement. It does not instantiate any models nor does it trigger
+    # Active Record callbacks or validations. Though passed values
+    # go through Active Record's type casting and serialization.
+    #
+    # See #insert_all for documentation.
+    def insert(attributes, returning: nil, unique_by: nil, record_timestamps: nil)
+      insert_all([ attributes ], returning: returning, unique_by: unique_by, record_timestamps: record_timestamps)
+    end
+
+    # Inserts multiple records into the database in a single SQL INSERT
+    # statement. It does not instantiate any models nor does it trigger
+    # Active Record callbacks or validations. Though passed values
+    # go through Active Record's type casting and serialization.
+    #
+    # The +attributes+ parameter is an Array of Hashes. Every Hash determines
+    # the attributes for a single row and must have the same keys.
+    #
+    # Rows are considered to be unique by every unique index on the table. Any
+    # duplicate rows are skipped.
+    # Override with <tt>:unique_by</tt> (see below).
+    #
+    # Returns an ActiveRecord::Result with its contents based on
+    # <tt>:returning</tt> (see below).
+    #
+    # ==== Options
+    #
+    # [:returning]
+    #   (PostgreSQL, SQLite3, and MariaDB only) An array of attributes to return for all successfully
+    #   inserted records, which by default is the primary key.
+    #   Pass <tt>returning: %w[ id name ]</tt> for both id and name
+    #   or <tt>returning: false</tt> to omit the underlying <tt>RETURNING</tt> SQL
+    #   clause entirely.
+    #
+    #   You can also pass an SQL string if you need more control on the return values
+    #   (for example, <tt>returning: Arel.sql("id, name as new_name")</tt>).
+    #
+    # [:unique_by]
+    #   (PostgreSQL and SQLite only) By default rows are considered to be unique
+    #   by every unique index on the table. Any duplicate rows are skipped.
+    #
+    #   To skip rows according to just one unique index pass <tt>:unique_by</tt>.
+    #
+    #   Consider a Book model where no duplicate ISBNs make sense, but if any
+    #   row has an existing id, or is not unique by another unique index,
+    #   ActiveRecord::RecordNotUnique is raised.
+    #
+    #   Unique indexes can be identified by columns or name:
+    #
+    #     unique_by: :isbn
+    #     unique_by: %i[ author_id name ]
+    #     unique_by: :index_books_on_isbn
+    #
+    # [:record_timestamps]
+    #   By default, automatic setting of timestamp columns is controlled by
+    #   the model's <tt>record_timestamps</tt> config, matching typical
+    #   behavior.
+    #
+    #   To override this and force automatic setting of timestamp columns one
+    #   way or the other, pass <tt>:record_timestamps</tt>:
+    #
+    #     record_timestamps: true  # Always set timestamps automatically
+    #     record_timestamps: false # Never set timestamps automatically
+    #
+    # Because it relies on the index information from the database
+    # <tt>:unique_by</tt> is recommended to be paired with
+    # Active Record's schema_cache.
+    #
+    # ==== Example
+    #
+    #   # Insert records and skip inserting any duplicates.
+    #   # Here "Eloquent Ruby" is skipped because its id is not unique.
+    #
+    #   Book.insert_all([
+    #     { id: 1, title: "Rework", author: "David" },
+    #     { id: 1, title: "Eloquent Ruby", author: "Russ" }
+    #   ])
+    #
+    #   # insert_all works on chained scopes, and you can use create_with
+    #   # to set default attributes for all inserted records.
+    #
+    #   author.books.create_with(created_at: Time.now).insert_all([
+    #     { id: 1, title: "Rework" },
+    #     { id: 2, title: "Eloquent Ruby" }
+    #   ])
+    def insert_all(attributes, returning: nil, unique_by: nil, record_timestamps: nil)
+      InsertAll.execute(self, attributes, on_duplicate: :skip, returning: returning, unique_by: unique_by, record_timestamps: record_timestamps)
+    end
+
+    # Inserts a single record into the database in a single SQL INSERT
+    # statement. It does not instantiate any models nor does it trigger
+    # Active Record callbacks or validations. Though passed values
+    # go through Active Record's type casting and serialization.
+    #
+    # See #insert_all! for more.
+    def insert!(attributes, returning: nil, record_timestamps: nil)
+      insert_all!([ attributes ], returning: returning, record_timestamps: record_timestamps)
+    end
+
+    # Inserts multiple records into the database in a single SQL INSERT
+    # statement. It does not instantiate any models nor does it trigger
+    # Active Record callbacks or validations. Though passed values
+    # go through Active Record's type casting and serialization.
+    #
+    # The +attributes+ parameter is an Array of Hashes. Every Hash determines
+    # the attributes for a single row and must have the same keys.
+    #
+    # Raises ActiveRecord::RecordNotUnique if any rows violate a
+    # unique index on the table. In that case, no rows are inserted.
+    #
+    # To skip duplicate rows, see #insert_all. To replace them, see #upsert_all.
+    #
+    # Returns an ActiveRecord::Result with its contents based on
+    # <tt>:returning</tt> (see below).
+    #
+    # ==== Options
+    #
+    # [:returning]
+    #   (PostgreSQL, SQLite3, and MariaDB only) An array of attributes to return for all successfully
+    #   inserted records, which by default is the primary key.
+    #   Pass <tt>returning: %w[ id name ]</tt> for both id and name
+    #   or <tt>returning: false</tt> to omit the underlying <tt>RETURNING</tt> SQL
+    #   clause entirely.
+    #
+    #   You can also pass an SQL string if you need more control on the return values
+    #   (for example, <tt>returning: Arel.sql("id, name as new_name")</tt>).
+    #
+    # [:record_timestamps]
+    #   By default, automatic setting of timestamp columns is controlled by
+    #   the model's <tt>record_timestamps</tt> config, matching typical
+    #   behavior.
+    #
+    #   To override this and force automatic setting of timestamp columns one
+    #   way or the other, pass <tt>:record_timestamps</tt>:
+    #
+    #     record_timestamps: true  # Always set timestamps automatically
+    #     record_timestamps: false # Never set timestamps automatically
+    #
+    # ==== Examples
+    #
+    #   # Insert multiple records
+    #   Book.insert_all!([
+    #     { title: "Rework", author: "David" },
+    #     { title: "Eloquent Ruby", author: "Russ" }
+    #   ])
+    #
+    #   # Raises ActiveRecord::RecordNotUnique because "Eloquent Ruby"
+    #   # does not have a unique id.
+    #   Book.insert_all!([
+    #     { id: 1, title: "Rework", author: "David" },
+    #     { id: 1, title: "Eloquent Ruby", author: "Russ" }
+    #   ])
+    def insert_all!(attributes, returning: nil, record_timestamps: nil)
+      InsertAll.execute(self, attributes, on_duplicate: :raise, returning: returning, record_timestamps: record_timestamps)
+    end
+
+    # Updates or inserts (upserts) a single record into the database in a
+    # single SQL INSERT statement. It does not instantiate any models nor does
+    # it trigger Active Record callbacks or validations. Though passed values
+    # go through Active Record's type casting and serialization.
+    #
+    # See #upsert_all for documentation.
+    def upsert(attributes, **kwargs)
+      upsert_all([ attributes ], **kwargs)
+    end
+
+    # Updates or inserts (upserts) multiple records into the database in a
+    # single SQL INSERT statement. It does not instantiate any models nor does
+    # it trigger Active Record callbacks or validations. Though passed values
+    # go through Active Record's type casting and serialization.
+    #
+    # The +attributes+ parameter is an Array of Hashes. Every Hash determines
+    # the attributes for a single row and must have the same keys.
+    #
+    # Returns an ActiveRecord::Result with its contents based on
+    # <tt>:returning</tt> (see below).
+    #
+    # By default, +upsert_all+ will update all the columns that can be updated when
+    # there is a conflict. These are all the columns except primary keys, read-only
+    # columns, and columns covered by the optional +unique_by+.
+    #
+    # ==== Options
+    #
+    # [:returning]
+    #   (PostgreSQL, SQLite3, and MariaDB only) An array of attributes to return for all successfully
+    #   inserted records, which by default is the primary key.
+    #   Pass <tt>returning: %w[ id name ]</tt> for both id and name
+    #   or <tt>returning: false</tt> to omit the underlying <tt>RETURNING</tt> SQL
+    #   clause entirely.
+    #
+    #   You can also pass an SQL string if you need more control on the return values
+    #   (for example, <tt>returning: Arel.sql("id, name as new_name")</tt>).
+    #
+    # [:unique_by]
+    #   (PostgreSQL and SQLite only) By default rows are considered to be unique
+    #   by every unique index on the table. Any duplicate rows are skipped.
+    #
+    #   To skip rows according to just one unique index pass <tt>:unique_by</tt>.
+    #
+    #   Consider a Book model where no duplicate ISBNs make sense, but if any
+    #   row has an existing id, or is not unique by another unique index,
+    #   ActiveRecord::RecordNotUnique is raised.
+    #
+    #   Unique indexes can be identified by columns or name:
+    #
+    #     unique_by: :isbn
+    #     unique_by: %i[ author_id name ]
+    #     unique_by: :index_books_on_isbn
+    #
+    # Because it relies on the index information from the database
+    # <tt>:unique_by</tt> is recommended to be paired with
+    # Active Record's schema_cache.
+    #
+    # [:on_duplicate]
+    #   Configure the SQL update sentence that will be used in case of conflict.
+    #
+    #   NOTE: If you use this option you must provide all the columns you want to update
+    #   by yourself.
+    #
+    #   Example:
+    #
+    #     Commodity.upsert_all(
+    #       [
+    #         { id: 2, name: "Copper", price: 4.84 },
+    #         { id: 4, name: "Gold", price: 1380.87 },
+    #         { id: 6, name: "Aluminium", price: 0.35 }
+    #       ],
+    #       on_duplicate: Arel.sql("price = GREATEST(commodities.price, EXCLUDED.price)")
+    #     )
+    #
+    #   See the related +:update_only+ option. Both options can't be used at the same time.
+    #
+    # [:update_only]
+    #   Provide a list of column names that will be updated in case of conflict. If not provided,
+    #   +upsert_all+ will update all the columns that can be updated. These are all the columns
+    #   except primary keys, read-only columns, and columns covered by the optional +unique_by+
+    #
+    #   Example:
+    #
+    #     Commodity.upsert_all(
+    #       [
+    #         { id: 2, name: "Copper", price: 4.84 },
+    #         { id: 4, name: "Gold", price: 1380.87 },
+    #         { id: 6, name: "Aluminium", price: 0.35 }
+    #       ],
+    #       update_only: [:price] # Only prices will be updated
+    #     )
+    #
+    #   See the related +:on_duplicate+ option. Both options can't be used at the same time.
+    #
+    # [:record_timestamps]
+    #   By default, automatic setting of timestamp columns is controlled by
+    #   the model's <tt>record_timestamps</tt> config, matching typical
+    #   behavior.
+    #
+    #   To override this and force automatic setting of timestamp columns one
+    #   way or the other, pass <tt>:record_timestamps</tt>:
+    #
+    #     record_timestamps: true  # Always set timestamps automatically
+    #     record_timestamps: false # Never set timestamps automatically
+    #
+    # ==== Examples
+    #
+    #   # Inserts multiple records, performing an upsert when records have duplicate ISBNs.
+    #   # Here "Eloquent Ruby" overwrites "Rework" because its ISBN is duplicate.
+    #
+    #   Book.upsert_all([
+    #     { title: "Rework", author: "David", isbn: "1" },
+    #     { title: "Eloquent Ruby", author: "Russ", isbn: "1" }
+    #   ], unique_by: :isbn)
+    #
+    #   Book.find_by(isbn: "1").title # => "Eloquent Ruby"
+    def upsert_all(attributes, on_duplicate: :update, update_only: nil, returning: nil, unique_by: nil, record_timestamps: nil)
+      InsertAll.execute(self, attributes, on_duplicate: on_duplicate, update_only: update_only, returning: returning, unique_by: unique_by, record_timestamps: record_timestamps)
+    end
+
     # Updates the counters of the records in the current relation.
     #
     # ==== Parameters
@@ -659,19 +1012,79 @@ module ActiveRecord
         raise ActiveRecordError.new("delete_all doesn't support #{invalid_methods.join(', ')}")
       end
 
-      arel = eager_loading? ? apply_join_dependency.arel : build_arel
-      arel.source.left = table
+      klass.with_connection do |c|
+        arel = eager_loading? ? apply_join_dependency.arel : build_arel(c)
+        arel.source.left = table
+
+        group_values_arel_columns = arel_columns(group_values.uniq)
+        having_clause_ast = having_clause.ast unless having_clause.empty?
+        key = if klass.composite_primary_key?
+          primary_key.map { |pk| table[pk] }
+        else
+          table[primary_key]
+        end
+        stmt = arel.compile_delete(key, having_clause_ast, group_values_arel_columns)
+
+        c.delete(stmt, "#{klass} Delete All").tap { reset }
+      end
+    end
+
+    # Deletes the row with a primary key matching the +id+ argument, using an
+    # SQL +DELETE+ statement, and returns the number of rows deleted. Active
+    # Record objects are not instantiated, so the object's callbacks are not
+    # executed, including any <tt>:dependent</tt> association options.
+    #
+    # You can delete multiple rows at once by passing an Array of <tt>id</tt>s.
+    #
+    # Note: Although it is often much faster than the alternative, #destroy,
+    # skipping callbacks might bypass business logic in your application
+    # that ensures referential integrity or performs other essential jobs.
+    #
+    # ==== Examples
+    #
+    #   # Delete a single row
+    #   Todo.delete(1)
+    #
+    #   # Delete multiple rows
+    #   Todo.delete([2,3,4])
+    def delete(id_or_array)
+      return 0 if id_or_array.nil? || (id_or_array.is_a?(Array) && id_or_array.empty?)
+
+      where(model.primary_key => id_or_array).delete_all
+    end
 
-      group_values_arel_columns = arel_columns(group_values.uniq)
-      having_clause_ast = having_clause.ast unless having_clause.empty?
-      key = if klass.composite_primary_key?
-        primary_key.map { |pk| table[pk] }
+
+    # Destroy an object (or multiple objects) that has the given id. The object is instantiated first,
+    # therefore all callbacks and filters are fired off before the object is deleted. This method is
+    # less efficient than #delete but allows cleanup methods and other actions to be run.
+    #
+    # This essentially finds the object (or multiple objects) with the given id, creates a new object
+    # from the attributes, and then calls destroy on it.
+    #
+    # ==== Parameters
+    #
+    # * +id+ - This should be the id or an array of ids to be destroyed.
+    #
+    # ==== Examples
+    #
+    #   # Destroy a single object
+    #   Todo.destroy(1)
+    #
+    #   # Destroy multiple objects
+    #   todos = [1,2,3]
+    #   Todo.destroy(todos)
+    def destroy(id)
+      multiple_ids = if model.composite_primary_key?
+        id.first.is_a?(Array)
       else
-        table[primary_key]
+        id.is_a?(Array)
       end
-      stmt = arel.compile_delete(key, having_clause_ast, group_values_arel_columns)
 
-      klass.connection.delete(stmt, "#{klass} Delete All").tap { reset }
+      if multiple_ids
+        find(id).each(&:destroy)
+      else
+        find(id).destroy
+      end
     end
 
     # Finds and destroys all records matching the specified conditions.
@@ -719,17 +1132,19 @@ module ActiveRecord
     #
     #   ASYNC Post Load (0.0ms) (db time 2ms)  SELECT "posts".* FROM "posts" LIMIT 100
     def load_async
-      return load if !connection.async_enabled?
+      with_connection do |c|
+        return load if !c.async_enabled?
 
-      unless loaded?
-        result = exec_main_query(async: connection.current_transaction.closed?)
+        unless loaded?
+          result = exec_main_query(async: c.current_transaction.closed?)
 
-        if result.is_a?(Array)
-          @records = result
-        else
-          @future_result = result
+          if result.is_a?(Array)
+            @records = result
+          else
+            @future_result = result
+          end
+          @loaded = true
         end
-        @loaded = true
       end
 
       self
@@ -785,8 +1200,9 @@ module ActiveRecord
           relation.to_sql
         end
       else
-        conn = klass.connection
-        conn.unprepared_statement { conn.to_sql(arel) }
+        klass.with_connection do |conn|
+          conn.unprepared_statement { conn.to_sql(arel) }
+        end
       end
     end
 
@@ -871,7 +1287,7 @@ module ActiveRecord
     end
 
     def alias_tracker(joins = [], aliases = nil) # :nodoc:
-      ActiveRecord::Associations::AliasTracker.create(connection, table.name, joins, aliases)
+      ActiveRecord::Associations::AliasTracker.create(connection_pool, table.name, joins, aliases)
     end
 
     class StrictLoadingScope # :nodoc:
@@ -947,7 +1363,11 @@ module ActiveRecord
       def _substitute_values(values)
         values.map do |name, value|
           attr = table[name]
-          unless Arel.arel_node?(value)
+          if Arel.arel_node?(value)
+            if value.is_a?(Arel::Nodes::SqlLiteral)
+              value = Arel::Nodes::Grouping.new(value)
+            end
+          else
             type = klass.type_for_attribute(attr.name)
             value = predicate_builder.build_bind_attribute(attr.name, type.cast(value))
           end
@@ -995,17 +1415,21 @@ module ActiveRecord
           if where_clause.contradiction?
             [].freeze
           elsif eager_loading?
-            apply_join_dependency do |relation, join_dependency|
-              if relation.null_relation?
-                [].freeze
-              else
-                relation = join_dependency.apply_column_aliases(relation)
-                @_join_dependency = join_dependency
-                connection.select_all(relation.arel, "SQL", async: async)
+            klass.with_connection do |c|
+              apply_join_dependency do |relation, join_dependency|
+                if relation.null_relation?
+                  [].freeze
+                else
+                  relation = join_dependency.apply_column_aliases(relation)
+                  @_join_dependency = join_dependency
+                  c.select_all(relation.arel, "SQL", async: async)
+                end
               end
             end
           else
-            klass._query_by_sql(arel, async: async)
+            klass.with_connection do |c|
+              klass._query_by_sql(c, arel, async: async)
+            end
           end
         end
       end