Enable extension of connection inspection methods

Author: GromNaN

src/Illuminate/Database/Connection.php

@@ -1347,6 +1347,24 @@ public function getDriverName()
         return $this->getConfig('driver');
     }
 
+    /**
+     * Get a human-readable name for the given connection.
+     */
+    public function getConnectionName(): string
+    {
+        return $this->getName() . '(' . $this->getDriverName() . ')';
+    }
+
+    /**
+     * Get the number of open connections for a database.
+     *
+     * @return int|null
+     */
+    public function getConnectionCount(): ?int
+    {
+        return null;
+    }
+
     /**
      * Get the query grammar used by the connection.
      *

src/Illuminate/Database/Console/DatabaseInspectionCommand.php

@@ -22,15 +22,7 @@ abstract class DatabaseInspectionCommand extends Command
      */
     protected function getConnectionName(ConnectionInterface $connection, $database)
     {
-        return match (true) {
-            $connection instanceof MySqlConnection && $connection->isMaria() => 'MariaDB',
-            $connection instanceof MySqlConnection => 'MySQL',
-            $connection instanceof MariaDbConnection => 'MariaDB',
-            $connection instanceof PostgresConnection => 'PostgreSQL',
-            $connection instanceof SQLiteConnection => 'SQLite',
-            $connection instanceof SqlServerConnection => 'SQL Server',
-            default => $database,
-        };
+        return method_exists($connection, 'getConnectionName') ? $connection->getConnectionName() : $database;
     }
 
     /**
@@ -41,18 +33,7 @@ protected function getConnectionName(ConnectionInterface $connection, $database)
      */
     protected function getConnectionCount(ConnectionInterface $connection)
     {
-        $result = match (true) {
-            $connection instanceof MySqlConnection => $connection->selectOne('show status where variable_name = "threads_connected"'),
-            $connection instanceof PostgresConnection => $connection->selectOne('select count(*) as "Value" from pg_stat_activity'),
-            $connection instanceof SqlServerConnection => $connection->selectOne('select count(*) Value from sys.dm_exec_sessions where status = ?', ['running']),
-            default => null,
-        };
-
-        if (! $result) {
-            return null;
-        }
-
-        return Arr::wrap((array) $result)['Value'];
+        return method_exists($connection, 'getConnectionCount') ? $connection->getConnectionCount() : null;
     }
 
     /**

src/Illuminate/Database/MariaDbConnection.php

@@ -32,6 +32,11 @@ public function getServerVersion(): string
         return Str::between(parent::getServerVersion(), '5.5.5-', '-MariaDB');
     }
 
+    public function getConnectionName(): string
+    {
+        return $this->getName() . ' (MariaDB)';
+    }
+
     /**
      * Get the default query grammar instance.
      *

src/Illuminate/Database/MySqlConnection.php

@@ -48,6 +48,22 @@ public function isMaria()
         return str_contains($this->getPdo()->getAttribute(PDO::ATTR_SERVER_VERSION), 'MariaDB');
     }
 
+    public function getConnectionName(): string
+    {
+        return $this->getName() . ' (' . ($this->isMaria() ? 'MariaDB' : 'MySQL') . ')';
+    }
+
+    public function getConnectionCount(): ?int
+    {
+        $result = $this->selectOne('show status where variable_name = "threads_connected"');
+
+        if (! $result) {
+            return null;
+        }
+
+        return (int) $result['Value'];
+    }
+
     /**
      * Get the server version for the connection.
      *

src/Illuminate/Database/PostgresConnection.php

@@ -12,6 +12,22 @@
 
 class PostgresConnection extends Connection
 {
+    public function getConnectionName(): string
+    {
+        return $this->getName() . ' (PostgreSQL)';
+    }
+
+    public function getConnectionCount(): ?int
+    {
+        $result = $this->selectOne('select count(*) as "Value" from pg_stat_activity');
+
+        if (! $result) {
+            return null;
+        }
+
+        return (int) $result['Value'];
+    }
+
     /**
      * Escape a binary value for safe SQL embedding.
      *

src/Illuminate/Database/SQLiteConnection.php

@@ -31,6 +31,11 @@ public function __construct($pdo, $database = '', $tablePrefix = '', array $conf
         $this->configureSynchronous();
     }
 
+    public function getConnectionName(): string
+    {
+        return $this->getName() . ' (SQLite)';
+    }
+
     /**
      * Enable or disable foreign key constraints if configured.
      *

src/Illuminate/Database/SqlServerConnection.php

@@ -90,6 +90,22 @@ protected function getDefaultQueryGrammar()
         return $this->withTablePrefix($grammar);
     }
 
+    public function getConnectionName(): string
+    {
+        return $this->getName() . ' (SQL Server)';
+    }
+
+    public function getConnectionCount(): ?int
+    {
+        $result = $this->selectOne('select count(*) Value from sys.dm_exec_sessions where status = ?', ['running']);
+
+        if (! $result) {
+            return null;
+        }
+
+        return (int) $result['Value'];
+    }
+
     /**
      * Get a schema builder instance for the connection.
      *