com.anwen.mongo.cache.global.DataSourceNameCache Maven / Gradle / Ivy
package com.anwen.mongo.cache.global;
import com.anwen.mongo.constant.DataSourceConstant;
import com.anwen.mongo.domain.MongoPlusDsException;
import com.anwen.mongo.model.BaseProperty;
import com.anwen.mongo.toolkit.StringUtils;
import java.util.HashMap;
import java.util.Map;
/**
* MongoClient缓存
*
* @author JiaChaoYang
**/
public class DataSourceNameCache {
/**
* 只存入当前最新的数据源名称即可
* 每次进行操作，都会经过{@link com.anwen.mongo.execute.ExecutorFactory#getExecute()}，通过这里存入的最新数据源名称去获取不同的ClientSession
* @date 2024/4/5 0:06
*/
private static final ThreadLocal dataSource = new InheritableThreadLocal<>();
private static final Map basePropertyMap = new HashMap<>();
public static void setBaseProperty(String ds,BaseProperty baseProperty){
basePropertyMap.put(ds,baseProperty);
}
public static BaseProperty getBaseProperty(String ds){
return basePropertyMap.get(ds);
}
public static void setDataSource(String ds){
dataSource.set(ds);
}
public static String getDatabase(){
String currentDataSource = getDataSource();
BaseProperty baseProperty = basePropertyMap.get(currentDataSource);
if (baseProperty == null){
throw new MongoPlusDsException("The " + currentDataSource+" data source does not exist");
}
return baseProperty.getDatabase();
}
public static String getDataSource(){
String ds = dataSource.get();
if (StringUtils.isBlank(ds)){
ds = DataSourceConstant.DEFAULT_DATASOURCE;
}
return ds;
}
public static void clear(){
dataSource.remove();
}
}