royalty 0.1.1

Ecosystem of libraries and tools designed to bring both performance and productivity to backend development, AI systems, data engineering, and distributed systems in Rust
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
51
52
53
54

//! Database connection module.
//!
//! Provides [`connect_db`] for establishing a connection pool to PostgreSQL
//! using SeaORM with configurable settings.
//!
//! ## Connection Pool Settings
//!
//! Configured via environment variables (see [`crate::config::AppConfig`]):
//!
//! - `DB_MAX_CONNECTIONS`: Maximum connections (default: 100)
//! - `DB_MIN_CONNECTIONS`: Minimum connections (default: 5)
//! - `DB_CONNECT_TIMEOUT`: Connection timeout in seconds (default: 8)
//! - `DB_IDLE_TIMEOUT`: Idle timeout in seconds (default: 600)
//! - `DB_MAX_LIFETIME`: Max connection lifetime in seconds (default: 1800)
//!
//! ## Example
//!
//! ```rust,ignore
//! use royalty::{connect_db, AppConfig};
//!
//! let config = AppConfig::from_env()?;
//! let db = connect_db(&config).await?;
//! ```

use crate::config::AppConfig;
use sea_orm::{ConnectOptions, Database, DatabaseConnection};
use std::time::Duration;

/// Establishes a database connection with a configured connection pool.
///
/// # Arguments
///
/// * `config` - Application configuration containing database settings
///
/// # Returns
///
/// A [`DatabaseConnection`] ready for use with SeaORM queries.
pub async fn connect_db(config: &AppConfig) -> anyhow::Result<DatabaseConnection> {
    let mut opt = ConnectOptions::new(config.database_url.clone());
    opt.max_connections(config.db_max_connections)
        .min_connections(config.db_min_connections)
        .connect_timeout(Duration::from_secs(config.db_connect_timeout))
        .acquire_timeout(Duration::from_secs(config.db_connect_timeout))
        .idle_timeout(Duration::from_secs(config.db_idle_timeout))
        .max_lifetime(Duration::from_secs(config.db_max_lifetime))
        .sqlx_logging(true)
        .sqlx_logging_level(tracing::log::LevelFilter::Info);

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

    tracing::info!("Database connection established");

    Ok(db)
}