soph_database/support/
database.rs

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
use crate::{config, Database, DatabaseResult};
use sea_orm::{ConnectOptions, ConnectionTrait, DatabaseConnection, DbBackend};
use soph_config::support::config;
use std::time::Duration;

impl Database {
    pub async fn new() -> DatabaseResult<Self> {
        let config = config().parse::<config::Database>()?;
        let mut opt = ConnectOptions::new(config.url.to_string());

        opt.max_connections(config.max_connections)
            .min_connections(config.min_connections)
            .connect_timeout(Duration::from_millis(config.connect_timeout))
            .idle_timeout(Duration::from_millis(config.idle_timeout))
            .acquire_timeout(Duration::from_millis(config.acquire_timeout))
            .max_lifetime(Duration::from_millis(config.max_lifetime))
            .sqlx_logging(config.sqlx_logging)
            .sqlx_logging_level(config.sqlx_logging_level.clone().into())
            .sqlx_slow_statements_logging_settings(
                config.sqlx_slow_statements_logging_level.clone().into(),
                Duration::from_millis(config.sqlx_slow_statements_logging_threshold),
            )
            .test_before_acquire(config.test_before_acquire);

        let connection = sea_orm::Database::connect(opt).await?;

        Ok(Self { connection })
    }

    pub fn driver(&self) -> &str {
        match self.get_database_backend() {
            DbBackend::MySql => "mysql",
            DbBackend::Postgres => "postgres",
            DbBackend::Sqlite => "sqlite",
        }
    }

    pub fn connection(&self) -> &DatabaseConnection {
        &self.connection
    }
}

impl std::ops::Deref for Database {
    type Target = DatabaseConnection;

    fn deref(&self) -> &Self::Target {
        &self.connection
    }
}