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

fix: FlexMapperProxy.getMethodDsKey

Browse Source
This commit is contained in:
Michael Yang 2024-06-03 11:36:16 +08:00
parent 18f7406f79
commit c5af830585
1 changed files with 20 additions and 28 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -25,18 +25,14 @@ import com.mybatisflex.core.mybatis.FlexConfiguration;
import com.mybatisflex.core.row.RowMapper; import com.mybatisflex.core.row.RowMapper;
import com.mybatisflex.core.table.TableInfo; import com.mybatisflex.core.table.TableInfo;
import com.mybatisflex.core.table.TableInfoFactory; import com.mybatisflex.core.table.TableInfoFactory;
import com.mybatisflex.core.util.MapUtil;
import com.mybatisflex.core.util.StringUtil; import com.mybatisflex.core.util.StringUtil;
import org.apache.ibatis.reflection.ExceptionUtil; import org.apache.ibatis.reflection.ExceptionUtil;
import org.apache.ibatis.session.SqlSession; import org.apache.ibatis.session.SqlSession;
import java.lang.reflect.Method; import java.lang.reflect.Method;
import java.util.Map; import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;
public class FlexMapperProxy<T> extends MybatisMapperProxy<T> { public class FlexMapperProxy<T> extends MybatisMapperProxy<T> {
private static final String NULL_KEY = "@NK@";
private static final Map<Method, String> methodDsKeyCache = new ConcurrentHashMap<>();
private final FlexDataSource dataSource; private final FlexDataSource dataSource;
public FlexMapperProxy(SqlSession sqlSession, Class<T> mapperInterface, Map<Method, MapperMethodInvoker> methodCache, public FlexMapperProxy(SqlSession sqlSession, Class<T> mapperInterface, Map<Method, MapperMethodInvoker> methodCache,
@ -106,33 +102,29 @@ public class FlexMapperProxy<T> extends MybatisMapperProxy<T> {
private static String getMethodDsKey(Method method, Object proxy) { private static String getMethodDsKey(Method method, Object proxy) {
String result = MapUtil.computeIfAbsent(methodDsKeyCache, method, m -> { UseDataSource methodAnno = method.getAnnotation(UseDataSource.class);
UseDataSource methodAnno = method.getAnnotation(UseDataSource.class); if (methodAnno != null && StringUtil.isNotBlank(methodAnno.value())) {
if (methodAnno != null && StringUtil.isNotBlank(methodAnno.value())) { return methodAnno.value();
return methodAnno.value(); }
}
Class<?>[] interfaces = proxy.getClass().getInterfaces(); Class<?>[] interfaces = proxy.getClass().getInterfaces();
for (Class<?> anInterface : interfaces) { for (Class<?> anInterface : interfaces) {
UseDataSource classAnno = anInterface.getAnnotation(UseDataSource.class); UseDataSource classAnno = anInterface.getAnnotation(UseDataSource.class);
if (classAnno != null && StringUtil.isNotBlank(classAnno.value())) { if (classAnno != null && StringUtil.isNotBlank(classAnno.value())) {
return classAnno.value(); return classAnno.value();
}
}
if (interfaces[0] != RowMapper.class) {
TableInfo tableInfo = TableInfoFactory.ofMapperClass(interfaces[0]);
if (tableInfo != null) {
String tableDsKey = tableInfo.getDataSource();
if (StringUtil.isNotBlank(tableDsKey)) {
return tableDsKey;
} }
} }
}
if (interfaces[0] != RowMapper.class) { return null;
TableInfo tableInfo = TableInfoFactory.ofMapperClass(interfaces[0]);
if (tableInfo != null) {
String tableDsKey = tableInfo.getDataSource();
if (StringUtil.isNotBlank(tableDsKey)) {
return tableDsKey;
}
}
}
return NULL_KEY;
});
return NULL_KEY.equals(result) ? null : result;
} }
} }