craken-database 0.1.1

SQLx-powered database abstraction and ORM for the Craken framework.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47

pub use async_trait;
use craken_http::error::CrakenError;
pub use craken_macros::Model;
use sqlx::postgres::PgPool;

pub mod migration;
pub mod model;
pub mod repository;

/// Database abstraction layer wrapping SQLx PgPool.
pub struct Database {
    pool: PgPool,
}

impl Database {
    pub async fn connect(url: &str) -> anyhow::Result<Self> {
        let pool = PgPool::connect(url).await?;
        Ok(Self { pool })
    }

    pub fn pool(&self) -> &PgPool {
        &self.pool
    }

    /// Helper to map SQLx errors to CrakenError.
    pub fn map_error(err: sqlx::Error) -> CrakenError {
        match err {
            sqlx::Error::RowNotFound => CrakenError::NotFound("record not found".to_string()),
            _ => CrakenError::Internal(err.to_string()),
        }
    }
}

/// Transaction wrapper for scoped database operations.
pub struct Transaction<'a> {
    pub(crate) tx: sqlx::Transaction<'a, sqlx::Postgres>,
}

impl<'a> Transaction<'a> {
    pub async fn commit(self) -> Result<(), sqlx::Error> {
        self.tx.commit().await
    }

    pub async fn rollback(self) -> Result<(), sqlx::Error> {
        self.tx.rollback().await
    }
}