优化aop语法 使用spring自动注入注解 基于注解拦截的aop注解不可能为空

RuoYi

2021-09-18 76db434373791347a54daa4fca88826d86e68699

优化aop语法 使用spring自动注入注解 基于注解拦截的aop注解不可能为空

 src/main/java/com/ruoyi/framework/aspectj/DataScopeAspect.java

@@ -1,12 +1,8 @@
package com.ruoyi.framework.aspectj;



import java.lang.reflect.Method;

import org.aspectj.lang.JoinPoint;

import org.aspectj.lang.Signature;

import org.aspectj.lang.annotation.Aspect;

import org.aspectj.lang.annotation.Before;

import org.aspectj.lang.annotation.Pointcut;

import org.aspectj.lang.reflect.MethodSignature;

import org.springframework.stereotype.Component;

import com.ruoyi.common.utils.SecurityUtils;

import com.ruoyi.common.utils.StringUtils;

@@ -18,7 +14,7 @@


/**

 * 数据过滤处理

 * 
 *

 * @author ruoyi

 */

@Aspect

@@ -55,27 +51,15 @@
     */

    public static final String DATA_SCOPE = "dataScope";



    // 配置织入点

    @Pointcut("@annotation(com.ruoyi.framework.aspectj.lang.annotation.DataScope)")

    public void dataScopePointCut()

    {

    }



    @Before("dataScopePointCut()")

    public void doBefore(JoinPoint point) throws Throwable

    @Before("@annotation(controllerDataScope)")

    public void doBefore(JoinPoint point, DataScope controllerDataScope) throws Throwable

    {

        clearDataScope(point);

        handleDataScope(point);

        handleDataScope(point, controllerDataScope);

    }



    protected void handleDataScope(final JoinPoint joinPoint)

    protected void handleDataScope(final JoinPoint joinPoint, DataScope controllerDataScope)

    {

        // 获得注解

        DataScope controllerDataScope = getAnnotationLog(joinPoint);

        if (controllerDataScope == null)

        {

            return;

        }

        // 获取当前的用户

        LoginUser loginUser = SecurityUtils.getLoginUser();

        if (StringUtils.isNotNull(loginUser))

@@ -92,10 +76,10 @@


    /**

     * 数据范围过滤

     * 
     *

     * @param joinPoint 切点

     * @param user 用户

     * @param alias 别名

     * @param userAlias 别名

     */

    public static void dataScopeFilter(JoinPoint joinPoint, SysUser user, String deptAlias, String userAlias)

    {

@@ -148,22 +132,6 @@
                baseEntity.getParams().put(DATA_SCOPE, " AND (" + sqlString.substring(4) + ")");

            }

        }

    }



    /**

     * 是否存在注解，如果存在就获取

     */

    private DataScope getAnnotationLog(JoinPoint joinPoint)

    {

        Signature signature = joinPoint.getSignature();

        MethodSignature methodSignature = (MethodSignature) signature;

        Method method = methodSignature.getMethod();



        if (method != null)

        {

            return method.getAnnotation(DataScope.class);

        }

        return null;

    }



    /**


 src/main/java/com/ruoyi/framework/aspectj/DataSourceAspect.java

@@ -1,17 +1,12 @@
package com.ruoyi.framework.aspectj;



import java.util.Objects;

import org.aspectj.lang.ProceedingJoinPoint;

import org.aspectj.lang.annotation.Around;

import org.aspectj.lang.annotation.Aspect;

import org.aspectj.lang.annotation.Pointcut;

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;

import com.ruoyi.framework.aspectj.lang.annotation.DataSource;

import com.ruoyi.framework.datasource.DynamicDataSourceContextHolder;



@@ -27,22 +22,11 @@
{

    protected Logger logger = LoggerFactory.getLogger(getClass());



    @Pointcut("@annotation(com.ruoyi.framework.aspectj.lang.annotation.DataSource)"

            + "|| @within(com.ruoyi.framework.aspectj.lang.annotation.DataSource)")

    public void dsPointCut()

    @Around("@annotation(dataSource) || @within(dataSource)")

    public Object around(ProceedingJoinPoint point, DataSource dataSource) throws Throwable

    {



    }



    @Around("dsPointCut()")

    public Object around(ProceedingJoinPoint point) throws Throwable

    {

        DataSource dataSource = getDataSource(point);



        if (StringUtils.isNotNull(dataSource))

        {

            DynamicDataSourceContextHolder.setDataSourceType(dataSource.value().name());

        }

        DynamicDataSourceContextHolder.setDataSourceType(dataSource.value().name());



        try

        {

@@ -55,18 +39,4 @@
        }

    }



    /**

     * 获取需要切换的数据源

     */

    public DataSource getDataSource(ProceedingJoinPoint point)

    {

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

        DataSource dataSource = AnnotationUtils.findAnnotation(signature.getMethod(), DataSource.class);

        if (Objects.nonNull(dataSource))

        {

            return dataSource;

        }



        return AnnotationUtils.findAnnotation(signature.getDeclaringType(), DataSource.class);

    }

}


 src/main/java/com/ruoyi/framework/aspectj/LogAspect.java

@@ -1,18 +1,13 @@
package com.ruoyi.framework.aspectj;



import java.lang.reflect.Method;

import java.util.Collection;

import java.util.Iterator;

import java.util.Map;

import javax.servlet.http.HttpServletRequest;

import javax.servlet.http.HttpServletResponse;

import org.aspectj.lang.JoinPoint;

import org.aspectj.lang.Signature;

import org.aspectj.lang.annotation.AfterReturning;

import org.aspectj.lang.annotation.AfterThrowing;

import org.aspectj.lang.annotation.Aspect;

import org.aspectj.lang.annotation.Pointcut;

import org.aspectj.lang.reflect.MethodSignature;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

import org.springframework.stereotype.Component;

@@ -43,21 +38,15 @@
{

    private static final Logger log = LoggerFactory.getLogger(LogAspect.class);



    // 配置织入点

    @Pointcut("@annotation(com.ruoyi.framework.aspectj.lang.annotation.Log)")

    public void logPointCut()

    {

    }



    /**

     * 处理完请求后执行

     *

     * @param joinPoint 切点

     */

    @AfterReturning(pointcut = "logPointCut()", returning = "jsonResult")

    public void doAfterReturning(JoinPoint joinPoint, Object jsonResult)

    @AfterReturning(pointcut = "@annotation(controllerLog)", returning = "jsonResult")

    public void doAfterReturning(JoinPoint joinPoint, Log controllerLog, Object jsonResult)

    {

        handleLog(joinPoint, null, jsonResult);

        handleLog(joinPoint, controllerLog, null, jsonResult);

    }



    /**

@@ -66,22 +55,16 @@
     * @param joinPoint 切点

     * @param e 异常

     */

    @AfterThrowing(value = "logPointCut()", throwing = "e")

    public void doAfterThrowing(JoinPoint joinPoint, Exception e)

    @AfterThrowing(value = "@annotation(controllerLog)", throwing = "e")

    public void doAfterThrowing(JoinPoint joinPoint, Log controllerLog, Exception e)

    {

        handleLog(joinPoint, e, null);

        handleLog(joinPoint, controllerLog, e, null);

    }



    protected void handleLog(final JoinPoint joinPoint, final Exception e, Object jsonResult)

    protected void handleLog(final JoinPoint joinPoint, Log controllerLog, final Exception e, Object jsonResult)

    {

        try

        {

            // 获得注解

            Log controllerLog = getAnnotationLog(joinPoint);

            if (controllerLog == null)

            {

                return;

            }



            // 获取当前的用户

            LoginUser loginUser = SecurityUtils.getLoginUser();

@@ -173,22 +156,6 @@
    }



    /**

     * 是否存在注解，如果存在就获取

     */

    private Log getAnnotationLog(JoinPoint joinPoint) throws Exception

    {

        Signature signature = joinPoint.getSignature();

        MethodSignature methodSignature = (MethodSignature) signature;

        Method method = methodSignature.getMethod();



        if (method != null)

        {

            return method.getAnnotation(Log.class);

        }

        return null;

    }



    /**

     * 参数拼装

     */

    private String argsArrayToString(Object[] paramsArray)

@@ -196,11 +163,11 @@
        String params = "";

        if (paramsArray != null && paramsArray.length > 0)

        {

            for (int i = 0; i < paramsArray.length; i++)

            for (Object o : paramsArray)

            {

                if (StringUtils.isNotNull(paramsArray[i]) && !isFilterObject(paramsArray[i]))

                if (StringUtils.isNotNull(o) && !isFilterObject(o))

                {

                    Object jsonObj = JSON.toJSON(paramsArray[i]);

                    Object jsonObj = JSON.toJSON(o);

                    params += jsonObj.toString() + " ";

                }

            }

@@ -225,17 +192,17 @@
        else if (Collection.class.isAssignableFrom(clazz))

        {

            Collection collection = (Collection) o;

            for (Iterator iter = collection.iterator(); iter.hasNext();)

            for (Object value : collection)

            {

                return iter.next() instanceof MultipartFile;

                return value instanceof MultipartFile;

            }

        }

        else if (Map.class.isAssignableFrom(clazz))

        {

            Map map = (Map) o;

            for (Iterator iter = map.entrySet().iterator(); iter.hasNext();)

            for (Object value : map.entrySet())

            {

                Map.Entry entry = (Map.Entry) iter.next();

                Map.Entry entry = (Map.Entry) value;

                return entry.getValue() instanceof MultipartFile;

            }

        }


 src/main/java/com/ruoyi/framework/aspectj/RateLimiterAspect.java

@@ -4,10 +4,8 @@
import java.util.Collections;

import java.util.List;

import org.aspectj.lang.JoinPoint;

import org.aspectj.lang.Signature;

import org.aspectj.lang.annotation.Aspect;

import org.aspectj.lang.annotation.Before;

import org.aspectj.lang.annotation.Pointcut;

import org.aspectj.lang.reflect.MethodSignature;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

@@ -49,16 +47,9 @@
        this.limitScript = limitScript;

    }



    // 配置织入点

    @Pointcut("@annotation(com.ruoyi.framework.aspectj.lang.annotation.RateLimiter)")

    public void rateLimiterPointCut()

    @Before("@annotation(rateLimiter)")

    public void doBefore(JoinPoint point, RateLimiter rateLimiter) throws Throwable

    {

    }



    @Before("rateLimiterPointCut()")

    public void doBefore(JoinPoint point) throws Throwable

    {

        RateLimiter rateLimiter = getAnnotationRateLimiter(point);

        String key = rateLimiter.key();

        int time = rateLimiter.time();

        int count = rateLimiter.count();

@@ -82,22 +73,6 @@
        {

            throw new RuntimeException("服务器限流异常，请稍后再试");

        }

    }



    /**

     * 是否存在注解，如果存在就获取

     */

    private RateLimiter getAnnotationRateLimiter(JoinPoint joinPoint)

    {

        Signature signature = joinPoint.getSignature();

        MethodSignature methodSignature = (MethodSignature) signature;

        Method method = methodSignature.getMethod();



        if (method != null)

        {

            return method.getAnnotation(RateLimiter.class);

        }

        return null;

    }



    public String getCombineKey(RateLimiter rateLimiter, JoinPoint point)