Add a get_kind method to Pool (#1228)

Author: nitnelave

sqlx-core/src/any/kind.rs

@@ -1,7 +1,7 @@
 use crate::error::Error;
 use std::str::FromStr;
 
-#[derive(Debug, Clone, Copy)]
+#[derive(Copy, Clone, Debug, PartialEq, Eq)]
 pub enum AnyKind {
     #[cfg(feature = "postgres")]
     Postgres,

sqlx-core/src/pool/mod.rs

@@ -55,6 +55,8 @@
 //! [`Pool::begin`].
 
 use self::inner::SharedPool;
+#[cfg(feature = "any")]
+use crate::any::{Any, AnyKind};
 use crate::connection::Connection;
 use crate::database::Database;
 use crate::error::Error;
@@ -290,7 +292,7 @@ impl<DB: Database> Pool<DB> {
         }
     }
 
-    /// Shut down the connection pool, waiting for all connections to be gracefully closed.    
+    /// Shut down the connection pool, waiting for all connections to be gracefully closed.
     ///
     /// Upon `.await`ing this call, any currently waiting or subsequent calls to [Pool::acquire] and
     /// the like will immediately return [Error::PoolClosed] and no new connections will be opened.
@@ -337,6 +339,17 @@ impl<DB: Database> Pool<DB> {
     }
 }
 
+#[cfg(feature = "any")]
+impl Pool<Any> {
+    /// Returns the database driver currently in-use by this `Pool`.
+    ///
+    /// Determined by the connection URI.
+    #[cfg(feature = "any")]
+    pub fn any_kind(&self) -> AnyKind {
+        self.0.connect_options.kind()
+    }
+}
+
 /// Returns a new [Pool] tied to the same shared connection pool.
 impl<DB: Database> Clone for Pool<DB> {
     fn clone(&self) -> Self {