webserver-rs 0.5.2

use crate::config::Database as DataBaseInfo;
use sea_orm::{ConnectOptions, Database, DatabaseConnection, DbErr};
use std::{collections::HashMap, sync::OnceLock};

static DBCONN_POOL: OnceLock<HashMap<String, DatabaseConnection>> = OnceLock::new();

async fn init_db_pool(db_info: Vec<DataBaseInfo>) -> Result<(), DbErr> {
    let mut map = HashMap::new();
    for db in db_info {
        let opt = ConnectOptions::new(db.protocol)
            .sqlx_logging(false)
            .to_owned();
        map.insert(db.name, Database::connect(opt).await?);
    }
    _ = DBCONN_POOL.set(map);
    Ok(())
}

#[allow(dead_code)]
#[cfg_attr(
    docsrs,
    doc(cfg(any(feature = "mysql", feature = "sqlite", feature = "postgres")))
)]
#[cfg(any(feature = "mysql", feature = "sqlite", feature = "postgres"))]
/// Get database connection by indexing `name` of `[[database]]` set in `config.toml`
pub fn get(key: &str) -> Result<&'static DatabaseConnection, std::io::Error> {
    DBCONN_POOL
        .get()
        .ok_or(std::io::Error::new(
            std::io::ErrorKind::NotFound,
            "Database Connection Pool wasn't initialized when the program startup",
        ))?
        .get(key)
        .ok_or(std::io::Error::new(
            std::io::ErrorKind::NotFound,
            format!("`{key}` is not found in Database Connection Pool"),
        ))
}

#[allow(dead_code)]
pub(crate) async fn init_db_if_enable(db_info: Vec<DataBaseInfo>) -> Result<(), DbErr> {
    if cfg!(feature = "mysql") || cfg!(feature = "sqlite") || cfg!(feature = "postgres") {
        init_db_pool(db_info).await
    } else {
        Ok(())
    }
}