操作日志新增消耗时间属性

RuoYi

2023-02-16 ee2ecbc0b4fe1900f0c2f71f34fc5dfe8b910d74

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

@@ -1,26 +1,23 @@
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.aspectj.lang.annotation.Before;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

import org.springframework.core.NamedThreadLocal;

import org.springframework.stereotype.Component;

import org.springframework.validation.BindingResult;

import org.springframework.web.multipart.MultipartFile;

import org.springframework.web.servlet.HandlerMapping;

import com.alibaba.fastjson.JSON;

import com.alibaba.fastjson2.JSON;

import com.ruoyi.common.enums.HttpMethod;

import com.ruoyi.common.filter.PropertyPreExcludeFilter;

import com.ruoyi.common.utils.SecurityUtils;

import com.ruoyi.common.utils.ServletUtils;

import com.ruoyi.common.utils.StringUtils;

@@ -43,10 +40,19 @@
{

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



    // 配置织入点

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

    public void logPointCut()

    /** 排除敏感属性字段 */

    public static final String[] EXCLUDE_PROPERTIES = { "password", "oldPassword", "newPassword", "confirmPassword" };



    /** 计算操作消耗时间 */

    private static final ThreadLocal<Long> TIME_THREADLOCAL = new NamedThreadLocal<Long>("Cost Time");



    /**

     * 处理请求前执行

     */

    @Before(value = "@annotation(controllerLog)")

    public void boBefore(JoinPoint joinPoint, Log controllerLog)

    {

        TIME_THREADLOCAL.set(System.currentTimeMillis());

    }



    /**

@@ -54,10 +60,10 @@
     *

     * @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,23 +72,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();



@@ -92,7 +91,7 @@
            // 请求的地址

            String ip = IpUtils.getIpAddr(ServletUtils.getRequest());

            operLog.setOperIp(ip);

            operLog.setOperUrl(ServletUtils.getRequest().getRequestURI());

            operLog.setOperUrl(StringUtils.substring(ServletUtils.getRequest().getRequestURI(), 0, 255));

            if (loginUser != null)

            {

                operLog.setOperName(loginUser.getUsername());

@@ -111,15 +110,20 @@
            operLog.setRequestMethod(ServletUtils.getRequest().getMethod());

            // 处理设置注解上的参数

            getControllerMethodDescription(joinPoint, controllerLog, operLog, jsonResult);

            // 设置消耗时间

            operLog.setCostTime(System.currentTimeMillis() - TIME_THREADLOCAL.get());

            // 保存数据库

            AsyncManager.me().execute(AsyncFactory.recordOper(operLog));

        }

        catch (Exception exp)

        {

            // 记录本地异常日志

            log.error("==前置通知异常==");

            log.error("异常信息:{}", exp.getMessage());

            exp.printStackTrace();

        }

        finally

        {

            TIME_THREADLOCAL.remove();

        }

    }



@@ -167,25 +171,9 @@
        }

        else

        {

            Map<?, ?> paramsMap = (Map<?, ?>) ServletUtils.getRequest().getAttribute(HandlerMapping.URI_TEMPLATE_VARIABLES_ATTRIBUTE);

            operLog.setOperParam(StringUtils.substring(paramsMap.toString(), 0, 2000));

            Map<?, ?> paramsMap = ServletUtils.getParamMap(ServletUtils.getRequest());

            operLog.setOperParam(StringUtils.substring(JSON.toJSONString(paramsMap, excludePropertyPreFilter()), 0, 2000));

        }

    }



    /**

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

     */

    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;

    }



    /**

@@ -196,16 +184,30 @@
        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]);

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

                    try

                    {

                        String jsonObj = JSON.toJSONString(o, excludePropertyPreFilter());

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

                    }

                    catch (Exception e)

                    {

                    }

                }

            }

        }

        return params.trim();

    }



    /**

     * 忽略敏感属性

     */

    public PropertyPreExcludeFilter excludePropertyPreFilter()

    {

        return new PropertyPreExcludeFilter().addExcludes(EXCLUDE_PROPERTIES);

    }



    /**

@@ -225,17 +227,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;

            }

        }