module ActiveRecord::ConnectionAdapters::QueryCache::ConnectionPoolConfiguration

def disable_query_cache!
  @query_cache_enabled.delete connection_cache_key(current_thread)
  connection.disable_query_cache! if active_connection?
end

def enable_query_cache!
  @query_cache_enabled[connection_cache_key(current_thread)] = true
  connection.enable_query_cache! if active_connection?
end

def initialize(*)
  super
  @query_cache_enabled = Concurrent::Map.new { false }
end

def query_cache_enabled
  @query_cache_enabled[connection_cache_key(current_thread)]
end