robotech 0.5.1

use crate::db::DbSettings;
use log::info;
use sea_orm::{ConnectOptions, Database, DatabaseConnection};
use sqlx::AnyPool;
use std::sync::OnceLock;

/// 数据库连接
pub static DB_CONN: OnceLock<DatabaseConnection> = OnceLock::new();

/// # 初始化数据库
///
/// 该函数接收数据库配置信息，建立数据库连接，并将连接存储到全局静态变量 `DB_CONN` 中。
/// 连接建立后，可以通过 `DB_CONN` 全局访问数据库连接。
///
/// # 参数
///
/// * `db_settings` - 数据库配置信息，包含连接数据库所需的信息
///
/// # Panics
///
/// * 如果数据库连接失败，程序将 panic
/// * 如果无法设置全局数据库连接，程序将 panic
pub async fn init_db(db_settings: DbSettings) {
    info!("init database...");

    // 获取数据库配置
    let mut opt = ConnectOptions::new(db_settings.url);

    // 设置sql日志按什么级别输出
    opt.sqlx_logging_level(log::LevelFilter::Trace);

    // 连接数据库
    let connection = Database::connect(opt)
        .await
        .expect("Failed to connect to the database");
    // 设置数据库连接到全局变量中
    DB_CONN
        .set(connection.clone())
        .expect("Unable to set database connector");
}

/// # 数据库迁移
///
/// 该函数负责执行数据库迁移操作。它使用提供的数据库配置信息连接到数据库，
/// 然后运行所有待处理的迁移脚本。
///
/// ## 参数
///
/// * `db_settings` - 数据库配置信息，包含连接数据库所需的URL等信息
///
/// ## 返回值
///
/// * `Ok(())` - 迁移成功完成
/// * `Err(sqlx::Error)` - 迁移过程中发生错误
///
/// ## 示例
///
/// ```rust
/// use crate::db::{DbSettings, migrate};
///
/// #[tokio::main]
/// async fn main() {
///     let settings = DbSettings {
///         url: "sqlite://data.db".to_string(),
///     };
///
///     if let Err(e) = migrate(settings).await {
///         eprintln!("数据库迁移失败: {}", e);
///     }
/// }
/// ```
pub async fn migrate(db_settings: DbSettings) -> Result<(), sqlx::Error> {
    info!("migrating database...");
    let db_url = db_settings.url;
    let pool = AnyPool::connect(db_url.as_str()).await?;
    Ok(sqlx::migrate!().run(&pool).await?)
}