diff --git a/Cargo.lock b/Cargo.lock index c42e760..921262f 100644 --- a/Cargo.lock +++ b/Cargo.lock @@ -747,7 +747,7 @@ dependencies = [ [[package]] name = "k-core" -version = "0.1.4" +version = "0.1.5" dependencies = [ "anyhow", "chrono", diff --git a/Cargo.toml b/Cargo.toml index 8b0d5a4..8158d2b 100644 --- a/Cargo.toml +++ b/Cargo.toml @@ -1,6 +1,6 @@ [package] name = "k-core" -version = "0.1.4" +version = "0.1.5" edition = "2024" [features] diff --git a/src/db.rs b/src/db.rs index 8f8a18c..484df5a 100644 --- a/src/db.rs +++ b/src/db.rs @@ -110,7 +110,25 @@ pub async fn connect(config: &DatabaseConfig) -> Result Option<&Pool> { + if let DatabasePool::Sqlite(pool) = self { + Some(pool) + } else { + None + } + } + + #[cfg(feature = "postgres")] + pub fn postgres_pool(&self) -> Option<&Pool> { + if let DatabasePool::Postgres(pool) = self { + Some(pool) + } else { + None + } + } +} #[cfg(feature = "sqlite")] pub async fn connect_sqlite(url: &str) -> Result, sqlx::Error> { sqlx::sqlite::SqlitePoolOptions::new().connect(url).await