module ActiveRecord::ConnectionHandling

Public Instance Methods

cockroachdb_connection(config) click to toggle source
# File lib/active_record/connection_adapters/cockroachdb_adapter.rb, line 28
def cockroachdb_connection(config)
  # This is copied from the PostgreSQL adapter.
  conn_params = config.symbolize_keys.compact

  # Map ActiveRecords param names to PGs.
  conn_params[:user] = conn_params.delete(:username) if conn_params[:username]
  conn_params[:dbname] = conn_params.delete(:database) if conn_params[:database]

  # Forward only valid config params to PG::Connection.connect.
  valid_conn_param_keys = PG::Connection.conndefaults_hash.keys + [:requiressl]
  conn_params.slice!(*valid_conn_param_keys)

  ConnectionAdapters::CockroachDBAdapter.new(
    ConnectionAdapters::CockroachDBAdapter.new_client(conn_params),
    logger,
    conn_params,
    config
  )
# This rescue flow appears in new_client, but it is needed here as well
# since Cockroach will sometimes not raise until a query is made.
rescue ActiveRecord::StatementInvalid => error
  if conn_params && conn_params[:dbname] && error.cause.message.include?(conn_params[:dbname])
    raise ActiveRecord::NoDatabaseError
  else
    raise ActiveRecord::ConnectionNotEstablished, error.message
  end
end