huiyu-db-mapper-core 0.1.1

use std::sync::Arc;
use crate::base::db_type::DbType;
use lazy_static::lazy_static;
use dashmap::DashMap;
use tracing::warn;
use tokio::task_local;
use crate::base::error::DatabaseError;

task_local! {
    pub static DB_NAME_REGISTRY: Arc<String>;
}

// 静态默认值，只分配一次
lazy_static! {
    static ref DB_TYPE_REGISTRY: DashMap<String, DbType> = DashMap::new();
    static ref DEFAULT_NAME: Arc<String> = Arc::new("default".to_string());
}

pub fn get_datasource_name() -> Arc<String> {
    if let Some(name) = DB_NAME_REGISTRY.try_get().ok() {
        return name.clone()
    }
    DEFAULT_NAME.clone()
}

pub(crate) fn set_datasource_type(name: String, data_type: DbType) {
    warn!("set_datasource_type: {} {}", name, data_type);
    DB_TYPE_REGISTRY.insert(name, data_type);
}

pub fn get_datasource_type_by_name(name: &str) -> Option<DbType> {
    let data_type = DB_TYPE_REGISTRY.get(name);
    if let Some(data_type) = data_type {
        return Some(data_type.value().clone());
    }
    None
}

pub fn get_datasource_type() -> Result<DbType,DatabaseError> {
    let db_type_opt = get_datasource_type_by_name(&get_datasource_name());
    if db_type_opt.is_none() {
        return Err(DatabaseError::NotFoundError("can't find db type!!!".to_string()));
    }
    Ok(db_type_opt.unwrap())
}