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

use sqlx::{postgres::PgPoolOptions, Pool, Postgres};

use crate::errors::sdk::SDKError;

#[derive(Clone)]
pub struct SDKEngine {
    pub pool: Box<Pool<Postgres>>,
}

pub async fn new_postgres_engine(
    database_url: &str,
    with_migration: bool,
) -> Result<SDKEngine, SDKError> {
    let pool = PgPoolOptions::new()
        .max_connections(3)
        .connect(database_url)
        .await?;

    if with_migration {
        sqlx::migrate!().run(&pool).await?;
    }

    Ok(SDKEngine {
        pool: Box::new(pool),
    })
}