注：本文的多資料源配置及切換的實作方法是，在框架中封裝，具體專案中配置及使用，也適用于多模塊專案
組態檔資料源讀取
通過springboot的Envioment和Binder物件進行讀取，無需手動宣告DataSource的Bean
yml資料源配置格式如下:
spring:
datasource:
master:
type: com.alibaba.druid.pool.DruidDataSource
driverClassName: com.mysql.cj.jdbc.Driver
url: jdbc:mysql://localhost:3306/main?
useUnicode=true&characterEncoding=utf8&zeroDateTimeBehavior=convertToNull&serverTimezone=Asia/Shanghai
username: root
password: 11111
cluster:
- key: db1
type: com.alibaba.druid.pool.DruidDataSource
driverClassName: com.mysql.cj.jdbc.Driver
url: jdbc:mysql://localhost:3306/haopanframetest_db1?
useUnicode=true&characterEncoding=utf8&zeroDateTimeBehavior=convertToNull&serverTimezone=Asia/Shanghai
username: root
password: 11111
- key: db2
type: com.alibaba.druid.pool.DruidDataSource
driverClassName: com.mysql.cj.jdbc.Driver
url: jdbc:mysql://localhost:3306/haopanframetest_db2?
useUnicode=true&characterEncoding=utf8&zeroDateTimeBehavior=convertToNull&serverTimezone=Asia/Shanghai
username: root
password: 11111
master為主資料庫必須配置，cluster下的為從庫，選擇性配置
獲取組態檔資訊代碼如下
@Autowired
private Environment env;
@Autowired
private ApplicationContext applicationContext;
private Binder binder;
binder = Binder.get(env);
List<Map> configs = binder.bind("spring.datasource.cluster", Bindable.listOf(Map.class)).get();
for (int i = 0; i < configs.size(); i++) {
config = configs.get(i);
String key = ConvertOp.convert2String(config.get("key"));
String type = ConvertOp.convert2String(config.get("type"));
String driverClassName = ConvertOp.convert2String(config.get("driverClassName"));
String url = ConvertOp.convert2String(config.get("url"));
String username = ConvertOp.convert2String(config.get("username"));
String password = ConvertOp.convert2String(config.get("password"));
}
動態加入資料源
定義獲取資料源的Service，具體專案中進行實作
public interface ExtraDataSourceService {
List<DataSourceModel> getExtraDataSourc();
}
獲取對應Service的所有實作類進行呼叫
private List<DataSourceModel> getExtraDataSource(){
List<DataSourceModel> dataSourceModelList = new ArrayList<>();
Map<String, ExtraDataSourceService> res =
applicationContext.getBeansOfType(ExtraDataSourceService.class);
for (Map.Entry en :res.entrySet()) {
ExtraDataSourceService service = (ExtraDataSourceService)en.getValue();
dataSourceModelList.addAll(service.getExtraDataSourc());
}
return dataSourceModelList;
}
通過代碼進行資料源注冊
主要是用過繼承類AbstractRoutingDataSource，重寫setTargetDataSources/setDefaultTargetDataSource方法
// 創建資料源
public boolean createDataSource(String key, String driveClass, String url, String username, String password, String databasetype) {
try {
try { // 排除連接不上的錯誤
Class.forName(driveClass);
DriverManager.getConnection(url, username, password);// 相當于連接資料庫
} catch (Exception e) {
return false;
}
@SuppressWarnings("resource")
DruidDataSource druidDataSource = new DruidDataSource();
druidDataSource.setName(key);
druidDataSource.setDriverClassName(driveClass);
druidDataSource.setUrl(url);
druidDataSource.setUsername(username);
druidDataSource.setPassword(password);
druidDataSource.setInitialSize(1); //初始化時建立物理連接的個數，初始化發生在顯示呼叫init方法，或者第一次getConnection時
druidDataSource.setMaxActive(20); //最大連接池數量
druidDataSource.setMaxWait(60000); //獲取連接時最大等待時間，單位毫秒，當鏈接數已經達到了最大鏈接數的時候，應用如果還要獲取鏈接就會出現等待的現象，等待鏈接釋放并回到鏈接池，如果等待的時間過長就應該踢掉這個等待，不然應用很可能出現雪崩現象
druidDataSource.setMinIdle(5); //最小連接池數量
String validationQuery = "select 1 from dual";
druidDataSource.setTestOnBorrow(true); //申請連接時執行validationQuery檢測連接是否有效，這里建議配置為TRUE，防止取到的連接不可用
druidDataSource.setTestWhileIdle(true);//建議配置為true，不影響性能，并且保證安全性，申請連接的時候檢測，如果空閑時間大于timeBetweenEvictionRunsMillis，執行validationQuery檢測連接是否有效，
druidDataSource.setValidationQuery(validationQuery); //用來檢測連接是否有效的sql，要求是一個查詢陳述句，如果validationQuery為null，testOnBorrow、testOnReturn、testWhileIdle都不會起作用，
druidDataSource.setFilters("stat");//屬性型別是字串，通過別名的方式配置擴展插件，常用的插件有：監控統計用的filter:stat日志用的filter:log4j防御sql注入的filter:wall
druidDataSource.setTimeBetweenEvictionRunsMillis(60000); //配置間隔多久才進行一次檢測，檢測需要關閉的空閑連接，單位是毫秒
druidDataSource.setMinEvictableIdleTimeMillis(180000); //配置一個連接在池中最小生存的時間，單位是毫秒，這里配置為3分鐘180000
druidDataSource.setKeepAlive(true); //打開druid.keepAlive之后，當連接池空閑時，池中的minIdle數量以內的連接，空閑時間超過minEvictableIdleTimeMillis，則會執行keepAlive操作，即執行druid.validationQuery指定的查詢SQL，一般為select * from dual，只要minEvictableIdleTimeMillis設定的小于防火墻切斷連接時間，就可以保證當連接空閑時自動做保活檢測，不會被防火墻切斷
druidDataSource.setRemoveAbandoned(true); //是否移除泄露的連接/超過時間限制是否回收，
druidDataSource.setRemoveAbandonedTimeout(3600); //泄露連接的定義時間(要超過最大事務的處理時間)；單位為秒，這里配置為1小時
druidDataSource.setLogAbandoned(true); //移除泄露連接發生是是否記錄日志
druidDataSource.init();
this.dynamicTargetDataSources.put(key, druidDataSource);
setTargetDataSources(this.dynamicTargetDataSources);// 將map賦值給父類的TargetDataSources
super.afterPropertiesSet();// 將TargetDataSources中的連接資訊放入resolvedDataSources管理
log.info(key+"資料源初始化成功");
//log.info(key+"資料源的概況："+druidDataSource.dump());
return true;
} catch (Exception e) {
log.error(e + "");
return false;
}
}
通過切面注解統一切換
定義注解
@Retention(RetentionPolicy.RUNTIME)
@Target({ElementType.METHOD, ElementType.TYPE, ElementType.PARAMETER})
@Documented
public @interface TargetDataSource {
String value() default "master"; //該值即key值
}
定義基于執行緒的切換類
public class DBContextHolder {
private static Logger log = LoggerFactory.getLogger(DBContextHolder.class);
// 對當前執行緒的操作-執行緒安全的
private static final ThreadLocal<String> contextHolder = new ThreadLocal<String>();
// 呼叫此方法，切換資料源
public static void setDataSource(String dataSource) {
contextHolder.set(dataSource);
log.info("已切換到資料源:{}",dataSource);
}
// 獲取資料源
public static String getDataSource() {
return contextHolder.get();
}
// 洗掉資料源
public static void clearDataSource() {
contextHolder.remove();
log.info("已切換到主資料源");
}
}
定義切面
方法的注解優先級高于類注解，一般用于Service的實作類
@Aspect
@Component
@Order(Ordered.HIGHEST_PRECEDENCE)
public class DruidDBAspect {
private static Logger logger = LoggerFactory.getLogger(DruidDBAspect.class);
@Autowired
private DynamicDataSource dynamicDataSource;
/**
* 切面點 指定注解
* */
@Pointcut("@annotation(com.haopan.frame.common.annotation.TargetDataSource) " +
"|| @within(com.haopan.frame.common.annotation.TargetDataSource)")
public void dataSourcePointCut() {
}
/**
* 攔截方法指定為 dataSourcePointCut
* */
@Around("dataSourcePointCut()")
public Object around(ProceedingJoinPoint point) throws Throwable {
MethodSignature signature = (MethodSignature) point.getSignature();
Class targetClass = point.getTarget().getClass();
Method method = signature.getMethod();
TargetDataSource targetDataSource = (TargetDataSource)targetClass.getAnnotation(TargetDataSource.class);
TargetDataSource methodDataSource = method.getAnnotation(TargetDataSource.class);
if(targetDataSource != null || methodDataSource != null){
String value;
if(methodDataSource != null){
value = https://www.cnblogs.com/yanpeng19940119/p/methodDataSource.value();
}else {
value = targetDataSource.value();
}
DBContextHolder.setDataSource(value);
logger.info("DB切換成功，切換至{}",value);
}
try {
return point.proceed();
} finally {
logger.info("清除DB切換");
DBContextHolder.clearDataSource();
}
}
}
分庫切換
開發程序中某個庫的某個表做了拆分操作，相同的某一次資料庫操作可能對應到不同的庫，需要對方法級別進行精確攔截，可以定義一個業務層面的切面，規定每個方法必須第一個引數為dbName，根據具體業務找到對應的庫傳參
@Around("dataSourcePointCut()")
public Object around(ProceedingJoinPoint point) throws Throwable {
MethodSignature signature = (MethodSignature) point.getSignature();
Class targetClass = point.getTarget().getClass();
Method method = signature.getMethod();
ProjectDataSource targetDataSource =
(ProjectDataSource)targetClass.getAnnotation(ProjectDataSource.class);
ProjectDataSource methodDataSource = method.getAnnotation(ProjectDataSource.class);
String valuehttps://www.cnblogs.com/yanpeng19940119/p/= "";
if(targetDataSource != null || methodDataSource != null){
//獲取方法定義引數
DefaultParameterNameDiscoverer discover = new DefaultParameterNameDiscoverer();
String[] parameterNames = discover.getParameterNames(method);
//獲取傳入目標方法的引數
Object[] args = point.getArgs();
for (int i=0;i<parameterNames.length;i++){
String pName = parameterNames[i];
if(pName.toLowerCase().equals("dbname")){
value = https://www.cnblogs.com/yanpeng19940119/p/ConvertOp.convert2String(args[i]);
}
}
if(!StringUtil.isEmpty(value)){
DBContextHolder.setDataSource(value);
logger.info("DB切換成功，切換至{}",value);
}
}
try {
return point.proceed();
} finally {
if(!StringUtil.isEmpty(value)){
logger.info("清除DB切換");
DBContextHolder.clearDataSource();
}
}
}
相關代碼
轉載請註明出處，本文鏈接：https://www.uj5u.com/houduan/92579.html
標籤：Java
上一篇：面試題精選:字串替換
下一篇：Java的重寫