Module: ActiveRecord::ConnectionHandling

Defined in:
lib/active_record/connection_adapters/cockroachdb_adapter.rb

Instance Method Summary collapse

Instance Method Details

#cockroachdb_connection(config) ⇒ Object 



34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
 
# File 'lib/active_record/connection_adapters/cockroachdb_adapter.rb', line 34

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
  no_db_err_check1 = conn_params && conn_params[:dbname] && error.cause.message.include?(conn_params[:dbname])
  no_db_err_check2 = conn_params && conn_params[:dbname] && error.cause.message.include?("pg_type")
  if no_db_err_check1 || no_db_err_check2
    raise ActiveRecord::NoDatabaseError
  else
    raise ActiveRecord::ConnectionNotEstablished, error.message
  end
end