class ActiveRecord::ConnectionAdapters::ConnectionHandler

Parent:: Объект

Обработчик подключений Active Record

ConnectionHandler представляет собой коллекцию объектов ConnectionPool. Он используется для хранения отдельных пулов подключений, которые соединяются с различными базами данных.

Например, предположим, что у вас есть 5 моделей с такой иерархией:

class Author < ActiveRecord::Base
end

class BankAccount < ActiveRecord::Base
end

class Book < ActiveRecord::Base
  establish_connection :library_db
end

class ScaryBook < Book
end

class GoodBook < Book
end

И файл database.yml, который выглядит так:

development:
  database: my_application
  host: localhost

library_db:
  database: library
  host: some.library.org

Ваша основная база данных в среде разработки — “my_application”, но модель Book подключается к отдельной базе данных под названием “library_db” (это может быть даже база данных на другом компьютере).

Book, ScaryBook и GoodBook будут использовать один и тот же пул подключений к “library_db”, в то время как Author, BankAccount и любые другие созданные вами модели будут использовать стандартный пул подключений к “my_application”.

Различные пулы подключений управляются единственным экземпляром ConnectionHandler, доступным через ActiveRecord::Base.connection_handler. Все модели Active Record используют этот обработчик для определения пула подключений, который они должны использовать.

Класс ConnectionHandler не связан с моделями Active, так как он не имеет информации о модели. Модель должна передать имя спецификации подключения обработчику, чтобы найти соответствующий пул подключений.

Общедоступные методы класса

new() Показать исходный код

# File activerecord/lib/active_record/connection_adapters/abstract/connection_handler.rb, line 74
def initialize
  # These caches are keyed by pool_config.connection_name (PoolConfig#connection_name).
  @connection_name_to_pool_manager = Concurrent::Map.new(initial_capacity: 2)
end

Общедоступные методы экземпляра

active_connections?(role = nil) Показать исходный код

# File activerecord/lib/active_record/connection_adapters/abstract/connection_handler.rb, line 155
def active_connections?(role = nil)
  each_connection_pool(role).any?(&:active_connection?)
end

Возвращает true, если среди пулов подключений, которые управляет ConnectionHandler, есть активные подключения.

clear_active_connections!(role = nil) Показать исходный код

# File activerecord/lib/active_record/connection_adapters/abstract/connection_handler.rb, line 162
def clear_active_connections!(role = nil)
  each_connection_pool(role).each do |pool|
    pool.release_connection
    pool.disable_query_cache!
  end
end

Возвращает все используемые текущей нитью подключения обратно в пул, а также возвращает подключения в пул, кэшированные нитями, которые больше не активны.

clear_all_connections!(role = nil) Показать исходный код

# File activerecord/lib/active_record/connection_adapters/abstract/connection_handler.rb, line 176
def clear_all_connections!(role = nil)
  each_connection_pool(role).each(&:disconnect!)
end

clear_reloadable_connections!(role = nil) Показать исходный код

# File activerecord/lib/active_record/connection_adapters/abstract/connection_handler.rb, line 172
def clear_reloadable_connections!(role = nil)
  each_connection_pool(role).each(&:clear_reloadable_connections!)
end

Очищает кэш, который сопоставляет классы.

См. ConnectionPool#clear_reloadable_connections! для подробностей.

connected?(connection_name, role: ActiveRecord::Base.current_role, shard: ActiveRecord::Base.current_shard) Показать исходный код

# File activerecord/lib/active_record/connection_adapters/abstract/connection_handler.rb, line 198
def connected?(connection_name, role: ActiveRecord::Base.current_role, shard: ActiveRecord::Base.current_shard)
  pool = retrieve_connection_pool(connection_name, role: role, shard: shard)
  pool && pool.connected?
end

Возвращает true, если подключение, доступное для этого класса, уже открыто.

connection_pool_list(role = nil) Показать исходный код

# File activerecord/lib/active_record/connection_adapters/abstract/connection_handler.rb, line 93
def connection_pool_list(role = nil)
  if role.nil? || role == :all
    connection_name_to_pool_manager.values.flat_map { |m| m.pool_configs.map(&:pool) }
  else
    connection_name_to_pool_manager.values.flat_map { |m| m.pool_configs(role).map(&:pool) }
  end
end

Возвращает пулы для обработчика подключений и заданного роли. Если передано :all, будут возвращены все пулы, принадлежащие обработчику подключений.

Также алиас: connection_pools

connection_pools(role = nil)

Псевдоним для: connection_pool_list

establish_connection(config, owner_name: Base, role: Base.current_role, shard: Base.current_shard, clobber: false) Показать исходный код

# File activerecord/lib/active_record/connection_adapters/abstract/connection_handler.rb, line 113
def establish_connection(config, owner_name: Base, role: Base.current_role, shard: Base.current_shard, clobber: false)
  owner_name = determine_owner_name(owner_name, config)

  pool_config = resolve_pool_config(config, owner_name, role, shard)
  db_config = pool_config.db_config

  pool_manager = set_pool_manager(pool_config.connection_name)

  # If there is an existing pool with the same values as the pool_config
  # don't remove the connection. Connections should only be removed if we are
  # establishing a connection on a class that is already connected to a different
  # configuration.
  existing_pool_config = pool_manager.get_pool_config(role, shard)

  if !clobber && existing_pool_config && existing_pool_config.db_config == db_config
    # Update the pool_config's connection class if it differs. This is used
    # for ensuring that ActiveRecord::Base and the primary_abstract_class use
    # the same pool. Without this granular swapping will not work correctly.
    if owner_name.primary_class? && (existing_pool_config.connection_class != owner_name)
      existing_pool_config.connection_class = owner_name
    end

    existing_pool_config.pool
  else
    disconnect_pool_from_pool_manager(pool_manager, role, shard)
    pool_manager.set_pool_config(role, shard, pool_config)

    payload = {
      connection_name: pool_config.connection_name,
      role: role,
      shard: shard,
      config: db_config.configuration_hash
    }

    ActiveSupport::Notifications.instrumenter.instrument("!connection.active_record", payload) do
      pool_config.pool
    end
  end
end

flush_idle_connections!(role = nil) Показать исходный код

# File activerecord/lib/active_record/connection_adapters/abstract/connection_handler.rb, line 183
def flush_idle_connections!(role = nil)
  each_connection_pool(role).each(&:flush!)
end

Отключает все в настоящее время неактивные подключения.

См. ConnectionPool#flush! для подробностей.

remove_connection_pool(connection_name, role: ActiveRecord::Base.current_role, shard: ActiveRecord::Base.current_shard) Показать исходный код

# File activerecord/lib/active_record/connection_adapters/abstract/connection_handler.rb, line 203
def remove_connection_pool(connection_name, role: ActiveRecord::Base.current_role, shard: ActiveRecord::Base.current_shard)
  if pool_manager = get_pool_manager(connection_name)
    disconnect_pool_from_pool_manager(pool_manager, role, shard)
  end
end

retrieve_connection_pool(connection_name, role: ActiveRecord::Base.current_role, shard: ActiveRecord::Base.current_shard, strict: false) Показать исходный код

# File activerecord/lib/active_record/connection_adapters/abstract/connection_handler.rb, line 212
def retrieve_connection_pool(connection_name, role: ActiveRecord::Base.current_role, shard: ActiveRecord::Base.current_shard, strict: false)
  pool = get_pool_manager(connection_name)&.get_pool_config(role, shard)&.pool

  if strict && !pool
    if shard != ActiveRecord::Base.default_shard
      message = "No connection pool for '#{connection_name}' found for the '#{shard}' shard."
    elsif role != ActiveRecord::Base.default_role
      message = "No connection pool for '#{connection_name}' found for the '#{role}' role."
    else
      message = "No connection pool for '#{connection_name}' found."
    end

    raise ConnectionNotEstablished, message
  end

  pool
end

Извлечение пула подключений происходит часто, поэтому мы кэшируем его в @connection_name_to_pool_manager. Это делает извлечение пула подключений O(1) после того, как процесс прогрелся. При установлении или удалении подключения мы аннулируем кэш.