fix 类上数据源类型获取不到问题,修改优先级,先根据方法,再根据类

master
杜龙少 4 years ago committed by Limy
parent 8aa4c266ec
commit d7abb6e7ba

@ -10,6 +10,8 @@ import com.ruoyi.common.enums.DataSourceType;
/** /**
* *
*
*
* *
* @author ruoyi * @author ruoyi
*/ */

@ -1,6 +1,7 @@
package com.ruoyi.framework.aspectj; package com.ruoyi.framework.aspectj;
import java.lang.reflect.Method; import java.lang.reflect.Method;
import java.util.Objects;
import org.aspectj.lang.ProceedingJoinPoint; import org.aspectj.lang.ProceedingJoinPoint;
import org.aspectj.lang.annotation.Around; import org.aspectj.lang.annotation.Around;
import org.aspectj.lang.annotation.Aspect; import org.aspectj.lang.annotation.Aspect;
@ -8,6 +9,7 @@ import org.aspectj.lang.annotation.Pointcut;
import org.aspectj.lang.reflect.MethodSignature; import org.aspectj.lang.reflect.MethodSignature;
import org.slf4j.Logger; import org.slf4j.Logger;
import org.slf4j.LoggerFactory; import org.slf4j.LoggerFactory;
import org.springframework.core.annotation.AnnotationUtils;
import org.springframework.core.annotation.Order; import org.springframework.core.annotation.Order;
import org.springframework.stereotype.Component; import org.springframework.stereotype.Component;
import com.ruoyi.common.annotation.DataSource; import com.ruoyi.common.annotation.DataSource;
@ -16,7 +18,7 @@ import com.ruoyi.common.utils.StringUtils;
/** /**
* *
* *
* @author ruoyi * @author ruoyi
*/ */
@Aspect @Aspect
@ -60,17 +62,14 @@ public class DataSourceAspect
public DataSource getDataSource(ProceedingJoinPoint point) public DataSource getDataSource(ProceedingJoinPoint point)
{ {
MethodSignature signature = (MethodSignature) point.getSignature(); MethodSignature signature = (MethodSignature) point.getSignature();
Class<? extends Object> targetClass = point.getTarget().getClass(); // point.getTarget().getClass(); 这个获取的是动态代理的class上面获取不到自定义注解
DataSource targetDataSource = targetClass.getAnnotation(DataSource.class); // 先获取方法上的获取不到啊再从class上找
if (StringUtils.isNotNull(targetDataSource)) DataSource dataSource = AnnotationUtils.findAnnotation(signature.getMethod(), DataSource.class);
if (Objects.nonNull(dataSource))
{ {
return targetDataSource;
}
else
{
Method method = signature.getMethod();
DataSource dataSource = method.getAnnotation(DataSource.class);
return dataSource; return dataSource;
} }
return AnnotationUtils.findAnnotation(signature.getDeclaringType(), DataSource.class);
} }
} }

Loading…
Cancel
Save