lenso-platform-core 0.1.20

Core runtime primitives for the Lenso backend 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
48
49
50

use crate::config::DatabaseConfig;
use crate::error::{AppError, AppResult, ErrorCode};
use sqlx::postgres::PgPoolOptions;
use sqlx::{PgPool, Postgres, Transaction};

pub type DbPool = PgPool;
pub type DbTransaction<'a> = Transaction<'a, Postgres>;

pub async fn connect_pool(config: &DatabaseConfig) -> AppResult<DbPool> {
    PgPoolOptions::new()
        .max_connections(config.max_connections)
        .connect(&config.url)
        .await
        .map_err(|source| {
            AppError::new(ErrorCode::ExternalDependency, "Database connection failed")
                .with_source(source)
                .retryable()
        })
}

pub async fn ping(pool: &DbPool) -> AppResult<()> {
    sqlx::query("select 1")
        .execute(pool)
        .await
        .map(|_| ())
        .map_err(|source| {
            AppError::new(
                ErrorCode::ExternalDependency,
                "Database health check failed",
            )
            .with_source(source)
            .retryable()
        })
}

pub async fn begin_transaction(pool: &DbPool) -> AppResult<DbTransaction<'_>> {
    pool.begin().await.map_err(map_transaction_error)
}

pub async fn commit_transaction(transaction: DbTransaction<'_>) -> AppResult<()> {
    transaction.commit().await.map_err(map_transaction_error)
}

pub async fn rollback_transaction(transaction: DbTransaction<'_>) -> AppResult<()> {
    transaction.rollback().await.map_err(map_transaction_error)
}

fn map_transaction_error(source: sqlx::Error) -> AppError {
    AppError::new(ErrorCode::Internal, "Database transaction failed").with_source(source)
}