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

feat: 缓存方法对应的数据源。

Browse Source
This commit is contained in:
Suomm 2023-06-25 20:40:31 +08:00
parent 6c74c9bdb1
commit f05ab3e23e
1 changed files with 37 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

55
mybatis-flex-spring/src/main/java/com/mybatisflex/spring/datasource/DataSourceInterceptor.java
View File

@ -22,6 +22,12 @@ import com.mybatisflex.core.datasource.DataSourceKey;
import com.mybatisflex.core.util.StringUtil; import com.mybatisflex.core.util.StringUtil;
import org.aopalliance.intercept.MethodInterceptor; import org.aopalliance.intercept.MethodInterceptor;
import org.aopalliance.intercept.MethodInvocation; import org.aopalliance.intercept.MethodInvocation;
import org.springframework.core.MethodClassKey;
import java.lang.reflect.Method;
import java.util.Map;
import java.util.Objects;
import java.util.concurrent.ConcurrentHashMap;
/** /**
* 多数据源切换拦截器。 * 多数据源切换拦截器。
@ -39,7 +45,7 @@ public class DataSourceInterceptor implements MethodInterceptor {
return invocation.proceed(); return invocation.proceed();
} }
dsKey = determineDataSourceKey(invocation); dsKey = findDataSourceKey(invocation.getMethod(), invocation.getThis());
if (StringUtil.isBlank(dsKey)) { if (StringUtil.isBlank(dsKey)) {
return invocation.proceed(); return invocation.proceed();
} }
@ -52,35 +58,48 @@ public class DataSourceInterceptor implements MethodInterceptor {
} }
} }
private String determineDataSourceKey(MethodInvocation invocation) { /**
* 缓存方法对应的数据源。
*/
private final Map<Object, String> dsCache = new ConcurrentHashMap<>();
private String findDataSourceKey(Method method, Object targetObject) {
Object cacheKey = new MethodClassKey(method, targetObject.getClass());
String dsKey = this.dsCache.get(cacheKey);
if (dsKey == null) {
dsKey = determineDataSourceKey(method, targetObject);
if (dsKey == null) {
dsKey = "";
}
this.dsCache.put(cacheKey, dsKey);
}
return dsKey;
}
private String determineDataSourceKey(Method method, Object targetObject) {
// 方法上定义有 UseDataSource 注解 // 方法上定义有 UseDataSource 注解
UseDataSource annotation = invocation.getMethod().getAnnotation(UseDataSource.class); UseDataSource annotation = method.getAnnotation(UseDataSource.class);
if (annotation != null) { if (annotation != null) {
return annotation.value(); return annotation.value();
} }
Object target = invocation.getThis(); // 类上定义有 UseDataSource 注解
Class<?> targetClass = targetObject.getClass();
annotation = targetClass.getAnnotation(UseDataSource.class);
if (annotation != null) {
return annotation.value();
}
if (target != null) { // 接口上定义有 UseDataSource 注解
// 类上定义有 UseDataSource 注解 Class<?>[] interfaces = targetClass.getInterfaces();
Class<?> targetClass = target.getClass(); for (Class<?> anInterface : interfaces) {
annotation = targetClass.getAnnotation(UseDataSource.class); annotation = anInterface.getAnnotation(UseDataSource.class);
if (annotation != null) { if (annotation != null) {
return annotation.value(); return annotation.value();
} }
// 接口上定义有 UseDataSource 注解
Class<?>[] interfaces = targetClass.getInterfaces();
for (Class<?> anInterface : interfaces) {
annotation = anInterface.getAnnotation(UseDataSource.class);
if (annotation != null) {
return annotation.value();
}
}
} }
return null; return null;
} }