module ActiveRecord::ConnectionAdapters::QueryCache

# File lib/active_record/connection_adapters/abstract/query_cache.rb, line 20
        def dirties_query_cache(base, *method_names)
          method_names.each do |method_name|
            base.class_eval <<-end_code, __FILE__, __LINE__ + 1
              def #{method_name}(...)
                if pool.dirties_query_cache
                  ActiveRecord::Base.clear_query_caches_for_current_thread
                end
                super
              end
            end_code
          end
        end

# File lib/active_record/connection_adapters/abstract/query_cache.rb, line 194
def initialize(*)
  super
  @query_cache = nil
end

# File lib/active_record/connection_adapters/abstract/query_cache.rb, line 217
def cache(&block)
  pool.enable_query_cache(&block)
end

# File lib/active_record/connection_adapters/abstract/query_cache.rb, line 243
def clear_query_cache
  pool.clear_query_cache
end

# File lib/active_record/connection_adapters/abstract/query_cache.rb, line 233
def disable_query_cache!
  pool.disable_query_cache!
end

# File lib/active_record/connection_adapters/abstract/query_cache.rb, line 221
def enable_query_cache!
  pool.enable_query_cache!
end

# File lib/active_record/connection_adapters/abstract/query_cache.rb, line 201
def query_cache
  if @pinned && @owner != ActiveSupport::IsolatedExecutionState.context
    # With transactional tests, if the connection is pinned, any thread
    # other than the one that pinned the connection need to go through the
    # query cache pool, so each thread get a different cache.
    pool.query_cache
  else
    @query_cache
  end
end

# File lib/active_record/connection_adapters/abstract/query_cache.rb, line 212
def query_cache_enabled
  query_cache&.enabled?
end

# File lib/active_record/connection_adapters/abstract/query_cache.rb, line 229
def uncached(dirties: true, &block)
  pool.disable_query_cache(dirties: dirties, &block)
end

# File lib/active_record/connection_adapters/abstract/query_cache.rb, line 319
def cache_notification_info(sql, name, binds)
  {
    sql: sql,
    binds: binds,
    type_casted_binds: -> { type_casted_binds(binds) },
    name: name,
    connection: self,
    transaction: current_transaction.user_transaction.presence,
    cached: true
  }
end

# File lib/active_record/connection_adapters/abstract/query_cache.rb, line 311
def cache_notification_info_result(sql, name, binds, result)
  payload = cache_notification_info(sql, name, binds)
  payload[:row_count] = result.length
  payload
end

# File lib/active_record/connection_adapters/abstract/query_cache.rb, line 289
def cache_sql(sql, name, binds)
  key = binds.empty? ? sql : [sql, binds]
  result = nil
  hit = true

  @lock.synchronize do
    result = query_cache.compute_if_absent(key) do
      hit = false
      yield
    end
  end

  if hit
    ActiveSupport::Notifications.instrument(
      "sql.active_record",
      cache_notification_info_result(sql, name, binds, result)
    )
  end

  result.dup
end

# File lib/active_record/connection_adapters/abstract/query_cache.rb, line 271
def lookup_sql_cache(sql, name, binds)
  key = binds.empty? ? sql : [sql, binds]

  result = nil
  @lock.synchronize do
    result = query_cache[key]
  end

  if result
    ActiveSupport::Notifications.instrument(
      "sql.active_record",
      cache_notification_info_result(sql, name, binds, result)
    )
  end

  result
end

# File lib/active_record/connection_adapters/abstract/query_cache.rb, line 267
def unset_query_cache!
  @query_cache = nil
end