修改数据源类型优先级，先根据方法，再根据类

RuoYi

2020-05-19 41799c10f0c72bfc91e5b79f3e4df6e585cc5715

修改数据源类型优先级，先根据方法，再根据类

 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);

    }

}


 ruoyi/src/main/java/com/ruoyi/framework/aspectj/lang/annotation/DataSource.java

@@ -10,7 +10,9 @@


/**

 * 自定义多数据源切换注解

 * 
 *

 * 优先级：先方法，后类，如果方法覆盖了类上的数据源类型，以方法的为准，否则以类上的为准

 *

 * @author ruoyi

 */

@Target({ ElementType.METHOD, ElementType.TYPE })