兰宝车间质量管理系统
blame | 历史 | 补丁 | 提交 | 提交对比 | 忽略空白
疯狂的狮子li
2020-05-20 98941acd2aadf435058412b4abe61fb6437433e8 
 ruoyi/src/main/java/com/ruoyi/framework/aspectj/DataSourceAspect.java


@@ -1,6 +1,6 @@


package com.ruoyi.framework.aspectj;




import java.lang.reflect.Method;


import java.util.Objects;


import org.aspectj.lang.ProceedingJoinPoint;


import org.aspectj.lang.annotation.Around;


import org.aspectj.lang.annotation.Aspect;


@@ -8,6 +8,7 @@


import org.aspectj.lang.reflect.MethodSignature;


import org.slf4j.Logger;


import org.slf4j.LoggerFactory;


import org.springframework.core.annotation.AnnotationUtils;


import org.springframework.core.annotation.Order;


import org.springframework.stereotype.Component;


import com.ruoyi.common.utils.StringUtils;


@@ -60,17 +61,12 @@


    public DataSource getDataSource(ProceedingJoinPoint point)


    {


        MethodSignature signature = (MethodSignature) point.getSignature();


        Class<? extends Object> targetClass = point.getTarget().getClass();


        DataSource targetDataSource = targetClass.getAnnotation(DataSource.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 AnnotationUtils.findAnnotation(signature.getDeclaringType(), DataSource.class);


    }


}