log切面

package com.desv.core.config.log;

import com.alibaba.fastjson.JSON;
import com.desv.core.bean.dto.LogDTO;
import com.desv.core.bean.entity.log.BusinessLog;
import com.desv.core.utils.HttpUtils;
import com.desv.core.utils.UserUtils;
import com.desv.core.utils.log.LogManager;
import com.desv.core.utils.log.LogTaskFactory;
import lombok.extern.slf4j.Slf4j;
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.springframework.stereotype.Component;
import org.springframework.validation.BindingResult;
import org.springframework.web.context.request.RequestContextHolder;
import org.springframework.web.context.request.ServletRequestAttributes;

import java.lang.reflect.Method;
import java.util.Arrays;

/**
 * @desc:
 * @author: Chen Chen
 * @created: 2021-04-08 19:45:13
 */
@Aspect
@Slf4j
@Component
public class LogAspect {
    
    /**
     * @author: Chen Chen
     * @created: 2021/4/8 19:47
     * @desc: 表示匹配带有自定义注解的方法
     */
    @Pointcut("@annotation(com.desv.core.config.log.Log)")
    public void pointcut() {
    }
    
    @Around("pointcut()")
    public Object around(ProceedingJoinPoint point) {
        Object result = null;
        long beginTime = System.currentTimeMillis();
        long endTime = System.currentTimeMillis();
        try {
            log.info("LogAspect: 我在目标方法之前执行！");
            result = point.proceed();
            endTime = System.currentTimeMillis();
            log.info("LogAspect: 我在目标方法之后执行！");
            insertSuccessLog(point, endTime - beginTime);
        } catch (Throwable e) {
            endTime = System.currentTimeMillis();
            insertErrorLog(point, endTime - beginTime);
        }
        return result;
    }
    
    private void insertLog(ProceedingJoinPoint point, long time) {
        MethodSignature signature = (MethodSignature) point.getSignature();
        Method method = signature.getMethod();
        BusinessLog businessLog = new BusinessLog();
        Log userAction = method.getAnnotation(Log.class);
        // 请求的类名
        String className = point.getTarget().getClass().getName();
        // 请求的方法名
        String methodName = signature.getMethod().getName();
//        String methodName = signature.getName();
        
        // 请求的方法参数值
//        String args = Arrays.toString(point.getArgs());
        Object[] args1 = point.getArgs();
        String args = JSON.toJSONString(Arrays.stream(point.getArgs()).filter(o -> !(o instanceof BindingResult)));
        LogDTO dto = new LogDTO();
        dto.setLogName(userAction.name());
        dto.setMsg(userAction.msg());
        dto.setActionName(userAction.action());
        dto.setParams(args);
        dto.setAccessUid(UserUtils.getUid());
        dto.setAccessUserName(UserUtils.getUserName());
        dto.setClassName(className);
        dto.setMethodName(methodName);
        dto.setIp(HttpUtils.getRealIp());
        dto.setUri(HttpUtils.getUri());
        dto.setModuleName(HttpUtils.getContextPath());
        dto.setCostTime(time);
        log.info("log : {} , cost :{}", JSON.toJSONString(dto), time);
        //将RequestAttributes对象设置为子线程共享
        ServletRequestAttributes sra = (ServletRequestAttributes) RequestContextHolder.getRequestAttributes();
        RequestContextHolder.setRequestAttributes(sra, true);
        LogManager.me().executeLog(LogTaskFactory.businessLog(dto));
    }
    
    private void insertSuccessLog(ProceedingJoinPoint point, long time) {
        insertLog(point, time);
    }
    
    private void insertErrorLog(ProceedingJoinPoint point, long time) {
        insertLog(point, time);
    }
    
    private void handle(ProceedingJoinPoint point) throws Exception {
//        //获取拦截的方法名
//        Signature sig = point.getSignature();
//        MethodSignature msig = null;
//        if (!(sig instanceof MethodSignature)) {
//            throw new IllegalArgumentException("该注解只能用于方法");
//        }
//        msig = (MethodSignature) sig;
//        Object target = point.getTarget();
//        Method currentMethod = target.getClass().getMethod(msig.getName(), msig.getParameterTypes());
//        String methodName = currentMethod.getName();
//
//        //获取用户id，admin和api模块获取idUser方式不同
//        String idUser = null;
//        HttpServletRequest request = HttpUtil.getRequest();
//        String token = request.getHeader("Authorization");
//        if(StringUtil.isNotEmpty(token)) {
//            idUser = JwtUtil.getUserId(token);
//        }
//        if(idUser==null) {
//            return ;
//        }
//
//        //获取拦截方法的参数
//        String className = point.getTarget().getClass().getName();
//        Object[] params = point.getArgs();
//
//        //获取操作名称
//        BussinessLog annotation = currentMethod.getAnnotation(BussinessLog.class);
//        String bussinessName = annotation.value();
//        String key = annotation.key();
//        Class dictClass = annotation.dict();
//
//        StringBuilder sb = new StringBuilder();
//        for (Object param : params) {
//            sb.append(param);
//            sb.append(" & ");
//        }
//        //如果涉及到修改,比对变化
//        String msg="";
//        if (bussinessName.contains("修改") || bussinessName.contains("编辑")) {
//            Object obj1 = LogObjectHolder.me().get();
//            Map<String, String> obj2 = null;
//            if(request.getMethod().equals("POST") && !ArrayUtils.isEmpty(params)){
//                obj2 = BeanUtil.beanToStringMap(params[0]);
//            } else{
//                obj2 = HttpUtil.getRequestParameters();
//            }
//            try {
//                msg = BeanUtil.contrastObj(dictClass, key, obj1, obj2);
//            }catch (Exception e){
//                log.info("error : {}",e.getMessage());
//            }
//        } else {
//            Map<String, String> parameters = HttpUtil.getRequestParameters();
//            AbstractDictMap dictMap = (AbstractDictMap) dictClass.newInstance();
//            msg = BeanUtil.parseMutiKey(dictMap,key,parameters);
//        }
//
//        LogManager.me().executeLog(LogTaskFactory.bussinessLog(idUser, bussinessName, className, methodName, msg));
    }
}
 

 

 

package com.desv.core.config.log;

import java.lang.annotation.ElementType;
import java.lang.annotation.Retention;
import java.lang.annotation.RetentionPolicy;
import java.lang.annotation.Target;

/**
 * @desc:
 * @author: Chen Chen
 * @created: 2021-04-08 18:31:50
 */
@Target(ElementType.METHOD)
@Retention(RetentionPolicy.RUNTIME)
public @interface Log {
    String name() default "";
    
    String action() default "";
    
    String msg() default "";
}