Merge pull request #36621 from robertomiranda/r/backport

(Backport 6-0-stable) Add database_exists? method to connection adapters

activerecord/CHANGELOG.md

+*   Add database_exists? method to connection adapters to check if a database exists.
+
+      *Guilherme Mansur*
+
 *   PostgreSQL: Fix GROUP BY with ORDER BY virtual count attribute.
 
     Fixes #36022.

activerecord/lib/active_record/connection_adapters/abstract_adapter.rb

@@ -264,6 +264,11 @@ def adapter_name
         self.class::ADAPTER_NAME
       end
 
+      # Does the database for this adapter exist?
+      def self.database_exists?(config)
+        raise NotImplementedError
+      end
+
       # Does this adapter support DDL rollbacks in transactions? That is, would
       # CREATE TABLE or ALTER TABLE get rolled back by a transaction?
       def supports_ddl_transactions?

activerecord/lib/active_record/connection_adapters/mysql2_adapter.rb

@@ -42,6 +42,12 @@ def initialize(connection, logger, connection_options, config)
         configure_connection
       end
 
+      def self.database_exists?(config)
+        !!ActiveRecord::Base.mysql2_connection(config)
+      rescue ActiveRecord::NoDatabaseError
+        false
+      end
+
       def supports_json?
         !mariadb? && database_version >= "5.7.8"
       end

activerecord/lib/active_record/connection_adapters/postgresql_adapter.rb

@@ -259,6 +259,12 @@ def initialize(connection, logger, connection_parameters, config)
         @use_insert_returning = @config.key?(:insert_returning) ? self.class.type_cast_config_to_boolean(@config[:insert_returning]) : true
       end
 
+      def self.database_exists?(config)
+        !!ActiveRecord::Base.postgresql_connection(config)
+      rescue ActiveRecord::NoDatabaseError
+        false
+      end
+
       # Is this connection alive and ready for queries?
       def active?
         @lock.synchronize do

activerecord/lib/active_record/connection_adapters/sqlite3_adapter.rb

@@ -98,6 +98,16 @@ def initialize(connection, logger, connection_options, config)
         configure_connection
       end
 
+      def self.database_exists?(config)
+        config = config.symbolize_keys
+        if config[:database] == ":memory:"
+          return true
+        else
+          database_file = defined?(Rails.root) ? File.expand_path(config[:database], Rails.root) : config[:database]
+          File.exist?(database_file)
+        end
+      end
+
       def supports_ddl_transactions?
         true
       end

activerecord/test/cases/adapters/mysql2/mysql2_adapter_test.rb

@@ -20,6 +20,18 @@ def test_exec_query_nothing_raises_with_no_result_queries
     end
   end
 
+  def test_database_exists_returns_false_if_database_does_not_exist
+    config = ActiveRecord::Base.configurations["arunit"].merge(database: "inexistent_activerecord_unittest")
+    assert_not ActiveRecord::ConnectionAdapters::Mysql2Adapter.database_exists?(config),
+      "expected database to not exist"
+  end
+
+  def test_database_exists_returns_true_when_the_database_exists
+    config = ActiveRecord::Base.configurations["arunit"]
+    assert ActiveRecord::ConnectionAdapters::Mysql2Adapter.database_exists?(config),
+     "expected database #{config[:database]} to exist"
+  end
+
   def test_columns_for_distinct_zero_orders
     assert_equal "posts.id",
       @conn.columns_for_distinct("posts.id", [])

activerecord/test/cases/adapters/postgresql/postgresql_adapter_test.rb

@@ -24,6 +24,18 @@ def test_bad_connection
         end
       end
 
+      def test_database_exists_returns_false_when_the_database_does_not_exist
+        config = { database: "non_extant_database", adapter: "postgresql" }
+        assert_not ActiveRecord::ConnectionAdapters::PostgreSQLAdapter.database_exists?(config),
+          "expected database #{config[:database]} to not exist"
+      end
+
+      def test_database_exists_returns_true_when_the_database_exists
+        config = ActiveRecord::Base.configurations["arunit"]
+        assert ActiveRecord::ConnectionAdapters::PostgreSQLAdapter.database_exists?(config),
+          "expected database #{config[:database]} to exist"
+      end
+
       def test_primary_key
         with_example_table do
           assert_equal "id", @connection.primary_key("ex")

activerecord/test/cases/adapters/sqlite3/sqlite3_adapter_test.rb

@@ -30,6 +30,17 @@ def test_bad_connection
         end
       end
 
+      def test_database_exists_returns_false_when_the_database_does_not_exist
+        assert_not SQLite3Adapter.database_exists?(adapter: "sqlite3", database: "non_extant_db"),
+          "expected non_extant_db to not exist"
+      end
+
+      def test_database_exists_returns_true_when_databae_exists
+        config = ActiveRecord::Base.configurations["arunit"]
+        assert SQLite3Adapter.database_exists?(config),
+          "expected #{config[:database]} to exist"
+      end
+
       unless in_memory_db?
         def test_connect_with_url
           original_connection = ActiveRecord::Base.remove_connection
@@ -53,6 +64,11 @@ def test_connect_memory_with_url
         end
       end
 
+      def test_database_exists_returns_true_for_an_in_memory_db
+        assert SQLite3Adapter.database_exists?(database: ":memory:"),
+          "Expected in memory database to exist"
+      end
+
       def test_column_types
         owner = Owner.create!(name: "hello".encode("ascii-8bit"))
         owner.reload