admin/mybatis-flex
1
0
Fork 0
mirror of https://gitee.com/mybatis-flex/mybatis-flex.git synced 2025-12-06 16:48:24 +08:00
Code Issues Packages Projects Releases Wiki Activity

refactor: optimize FlexMapperProxy.java

Browse Source
This commit is contained in:
Michael Yang 2024-06-02 12:19:04 +08:00
parent 6f6011f406
commit 476e33db42
1 changed files with 8 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
mybatis-flex-core/src/main/java/com/mybatisflex/core/mybatis/binding/FlexMapperProxy.java
View File

@ -60,10 +60,6 @@ public class FlexMapperProxy<T> extends MybatisMapperProxy<T> {
//最终使用的数据源
String finalDsKey = userDsKey;
//由用户指定的数据类型
DbType userDbType = DialectFactory.getHintDbType();
DbType finalDbType = userDbType;
try {
if (StringUtil.isBlank(finalDsKey)) {
finalDsKey = getMethodDsKey(method, proxy);
@ -73,34 +69,29 @@ public class FlexMapperProxy<T> extends MybatisMapperProxy<T> {
finalDsKey = DataSourceKey.getShardingDsKey(finalDsKey, proxy, method, args);
if (StringUtil.isNotBlank(finalDsKey) && !finalDsKey.equals(userDsKey)) {
DataSourceKey.use(finalDsKey);
needClearDsKey = true;
DataSourceKey.use(finalDsKey);
}
if (finalDbType == null) {
DbType hintDbType = DialectFactory.getHintDbType();
if (hintDbType == null) {
if (finalDsKey != null && dataSource != null) {
//使用最终分片获取数据源类型
finalDbType = dataSource.getDbType(finalDsKey);
hintDbType = dataSource.getDbType(finalDsKey);
}
if (finalDbType == null) {
finalDbType = FlexGlobalConfig.getDefaultConfig().getDbType();
if (hintDbType == null) {
hintDbType = FlexGlobalConfig.getDefaultConfig().getDbType();
}
needClearDbType = true;
DialectFactory.setHintDbType(finalDbType);
DialectFactory.setHintDbType(hintDbType);
}
return cachedInvoker(method).invoke(proxy, method, args, sqlSession);
} catch (Throwable e) {
throw ExceptionUtil.unwrapThrowable(e);
} finally {
if (needClearDbType) {
if (userDbType != null) {
DialectFactory.setHintDbType(userDbType);
} else {
DialectFactory.clearHintDbType();
}
DialectFactory.clearHintDbType();
}
if (needClearDsKey) {
if (userDsKey != null) {