soph_database/support/
database.rs

1use crate::{config, Database, DatabaseResult};
2use sea_orm::{ConnectOptions, ConnectionTrait, DatabaseConnection, DbBackend};
3use soph_config::support::config;
4use std::time::Duration;
5
6impl Database {
7    pub async fn new() -> DatabaseResult<Self> {
8        let config = config().parse::<config::Database>()?;
9        let mut opt = ConnectOptions::new(config.url.to_string());
10
11        opt.max_connections(config.max_connections)
12            .min_connections(config.min_connections)
13            .connect_timeout(Duration::from_millis(config.connect_timeout))
14            .idle_timeout(Duration::from_millis(config.idle_timeout))
15            .acquire_timeout(Duration::from_millis(config.acquire_timeout))
16            .max_lifetime(Duration::from_millis(config.max_lifetime))
17            .sqlx_logging(config.sqlx_logging)
18            .sqlx_logging_level(config.sqlx_logging_level.clone().into())
19            .sqlx_slow_statements_logging_settings(
20                config.sqlx_slow_statements_logging_level.clone().into(),
21                Duration::from_millis(config.sqlx_slow_statements_logging_threshold),
22            )
23            .test_before_acquire(config.test_before_acquire);
24
25        let connection = sea_orm::Database::connect(opt).await?;
26
27        Ok(Self { connection })
28    }
29
30    pub fn driver(&self) -> &str {
31        match self.get_database_backend() {
32            DbBackend::MySql => "mysql",
33            DbBackend::Postgres => "postgres",
34            DbBackend::Sqlite => "sqlite",
35        }
36    }
37
38    pub fn connection(&self) -> &DatabaseConnection {
39        &self.connection
40    }
41}
42
43impl std::ops::Deref for Database {
44    type Target = DatabaseConnection;
45
46    fn deref(&self) -> &Self::Target {
47        &self.connection
48    }
49}
soph_database/support/database.rs

soph_database/support/
database.rs
