Module: ActiveRecord::ConnectionHandling

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

Instance Method Summary collapse

Instance Method Details

#cockroachdb_connection(config) ⇒ Object 



26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
 
# File 'lib/active_record/connection_adapters/cockroachdb_adapter.rb', line 26

def cockroachdb_connection(config)
  # This is copied from the PostgreSQL adapter.
  conn_params = config.symbolize_keys

  conn_params.delete_if { |_, v| v.nil? }

  # 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 + [:sslmode, :application_name]
  conn_params.slice!(*valid_conn_param_keys)

  conn = PG.connect(conn_params)
  ConnectionAdapters::CockroachDBAdapter.new(conn, logger, conn_params, config)
rescue ::PG::Error, ActiveRecord::ActiveRecordError  => error
  if error.message.include?("does not exist")
    raise ActiveRecord::NoDatabaseError
  else
    raise
  end
end