module ActiveRecord::Import

def self.base_adapter(adapter)
  case adapter
  when 'mysql2_makara' then 'mysql2'
  when 'mysql2spatial' then 'mysql2'
  when 'spatialite' then 'sqlite3'
  when 'postgresql_makara' then 'postgresql'
  when 'makara_postgis' then 'postgresql'
  when 'postgis' then 'postgresql'
  when 'cockroachdb' then 'postgresql'
  else adapter
  end
end

def self.load_from_connection_pool(connection_pool)
  adapter =
    if connection_pool.respond_to?(:db_config) # ActiveRecord >= 6.1
      connection_pool.db_config.adapter
    else
      connection_pool.spec.config[:adapter]
    end
  require_adapter adapter
end

def self.require_adapter(adapter)
  require File.join(ADAPTER_PATH, "/#{base_adapter(adapter)}_adapter")
rescue LoadError
  # fallback
end