gitee
/
roses
mirror of https://gitee.com/stylefeng/roses
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

【7.1.5】更新CustomDatabaseIdProvider

pull/22/head
fengshuonan 2021-10-14 17:51:03 +08:00
parent f0da0a9dc1
commit f3f5cfba0b
1 changed files with 18 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

32
kernel-d-db/db-sdk-mp/src/main/java/cn/stylefeng/roses/kernel/db/mp/dbid/CustomDatabaseIdProvider.java
View File

@ -48,22 +48,26 @@ public class CustomDatabaseIdProvider implements DatabaseIdProvider {
String url = "";
// 常规获取url的方式
try {
Class<?> XAClass = Class.forName("io.seata.rm.datasource.xa.DataSourceProxyXA");
// xa的dataSource
if (XAClass.isInstance(dataSource)) {
Method xaMethod = XAClass.getMethod("getResourceId");
Object xaResult = xaMethod.invoke(dataSource);
url = xaResult.toString();
} else {
// 其他的dataSource类型还走原有的逻辑
url = dataSource.getConnection().getMetaData().getURL();
}
url = dataSource.getConnection().getMetaData().getURL();
} catch (Exception e) {
log.error("CustomDatabaseIdProvider无法判断当前数据源类型", e);
return DbTypeEnum.MYSQL.getCode();
// 兼容对seata数据源的判断
try {
Class<?> XAClass = Class.forName("io.seata.rm.datasource.xa.DataSourceProxyXA");
// xa的dataSource
if (XAClass.isInstance(dataSource)) {
Method xaMethod = XAClass.getMethod("getResourceId");
Object xaResult = xaMethod.invoke(dataSource);
url = xaResult.toString();
}
} catch (Exception e2) {
log.warn("CustomDatabaseIdProvider无法判断当前数据源类型，默认选择Mysql类型");
return DbTypeEnum.MYSQL.getCode();
}
}
if (url.contains(DbTypeEnum.ORACLE.getCode())) {