Make the database kind queryable from AnyConnectionKind and AnyConnection

Author: 05storm26

sqlx-core/src/any/connection/mod.rs

@@ -1,6 +1,6 @@
 use futures_core::future::BoxFuture;
 
-use crate::any::{Any, AnyConnectOptions};
+use crate::any::{Any, AnyConnectOptions, AnyKind};
 use crate::connection::Connection;
 use crate::error::Error;
 
@@ -47,6 +47,30 @@ pub(crate) enum AnyConnectionKind {
     Sqlite(sqlite::SqliteConnection),
 }
 
+impl AnyConnectionKind {
+    pub fn kind(&self) -> AnyKind {
+        match self {
+            #[cfg(feature = "postgres")]
+            AnyConnectionKind::Postgres(conn) => AnyKind::Postgres,
+
+            #[cfg(feature = "mysql")]
+            AnyConnectionKind::MySql(conn) => AnyKind::MySql,
+
+            #[cfg(feature = "sqlite")]
+            AnyConnectionKind::Sqlite(conn) => AnyKind::Sqlite,
+
+            #[cfg(feature = "mssql")]
+            AnyConnectionKind::Mssql(conn) => AnyKind::Mssql,
+        }
+    }
+}
+
+impl AnyConnection {
+    pub fn kind(&self) -> AnyKind {
+        self.0.kind()
+    }
+}
+
 macro_rules! delegate_to {
     ($self:ident.$method:ident($($arg:ident),*)) => {
         match &$self.0 {