【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

View File

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