logAspect

package oa.aop;

import com.fasterxml.jackson.core.JsonProcessingException;
import com.fasterxml.jackson.databind.ObjectMapper;
import oa.log.biz.RecordBiz;
import oa.log.po.Record;
import oa.system.ctrl.LoginCtrl;
import oa.utils.JacksonJson;
import org.apache.commons.lang.StringUtils;
import org.aspectj.lang.JoinPoint;
import org.aspectj.lang.Signature;
import org.aspectj.lang.annotation.AfterReturning;
import org.aspectj.lang.annotation.Aspect;
import org.aspectj.lang.annotation.Before;
import org.aspectj.lang.reflect.MethodSignature;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.http.HttpMethod;
import org.springframework.stereotype.Component;
import org.springframework.validation.Errors;
import org.springframework.web.context.request.RequestAttributes;
import org.springframework.web.context.request.RequestContextHolder;
import org.springframework.web.context.request.ServletRequestAttributes;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpSession;
import java.io.Serializable;
import java.util.Enumeration;
import java.util.HashMap;
import java.util.Map;
import java.util.UUID;

@Aspect
@Component
public class LogAspect {

    private static final Logger log = LoggerFactory.getLogger(LogAspect.class);
    private static ThreadLocal<Long> startTime = new ThreadLocal<Long>();
    private static ThreadLocal<String> key = new ThreadLocal<String>();
    private static ObjectMapper objectMapper = new ObjectMapper();

    private RecordBiz logRecordBiz;

    @Autowired
    public void setLogRecordBiz(RecordBiz logRecordBiz) {
        this.logRecordBiz = logRecordBiz;
    }

    /**
     * 请求方法前打印内容
     *
     * @param joinPoint
     */
    @Before("execution(public * oa..*Ctrl.save*(..))"
            + " || execution(public * oa..*Ctrl.update*(..))"
            + " || execution(public * oa..*Ctrl.delete*(..))"
            + " || execution(public * oa..*Ctrl.delBy*(..))"
            + " || execution(public * oa.common.BaseCtrl+.save*(..))"
            + " || execution(public * oa.common.BaseCtrl+.update*(..))"
            + " || execution(public * oa.common.BaseCtrl+.delBy*(..))"
            + " || execution(public * oa.common.BaseCtrl+.delete*(..))")
    public void doBefore(JoinPoint joinPoint) {
        // 请求开始时间
        startTime.set(System.currentTimeMillis());
        RequestAttributes ra = RequestContextHolder.getRequestAttributes();
        ServletRequestAttributes sra = (ServletRequestAttributes) ra;
        HttpServletRequest request = sra.getRequest();
        // 如果有session则返回session如果没有则返回null(避免创建过多的session浪费内存)
        HttpSession session = request.getSession(false);
        // 获取请求头
        Enumeration<String> enumeration = request.getHeaderNames();
        StringBuffer headers = new StringBuffer();
        while (enumeration.hasMoreElements()) {
            String name = enumeration.nextElement();
            String value = request.getHeader(name);
            headers.append(name + ":" + value).append(",");
        }
        String uri = UUID.randomUUID() + "|" + request.getRequestURI();

        String method = request.getMethod();
        StringBuffer params = new StringBuffer();
        if (HttpMethod.GET.toString().equals(method)) {// get请求
            String queryString = request.getQueryString();
            if (StringUtils.isNotBlank(queryString)) {
                // params.append(URLEncodedUtils.encode(queryString, "UTF-8"));
            }
        } else {//其他请求
            Object[] paramsArray = joinPoint.getArgs();
            if (paramsArray != null && paramsArray.length > 0) {
                for (int i = 0; i < paramsArray.length; i++) {
                    if (paramsArray[i] instanceof Serializable) {
                        params.append(paramsArray[i].toString()).append(",");
                    } else {
                        //使用json系列化 反射等等方法 反系列化会影响请求性能建议重写tostring方法实现系列化接口
                        try {
                            String param = objectMapper.writeValueAsString(paramsArray[i]);
                            if (StringUtils.isNotBlank(param))
                                params.append(param).append(",");
                        } catch (JsonProcessingException e) {
                            log.error("doBefore obj to json exception obj={},msg={}", paramsArray[i], e);
                        }
                    }
                }
            }
        }
        key.set(uri);
        log.info("request params>>>>>>uri={},method={},params={},headers={}", uri, method, params, headers);
    }

    @AfterReturning("execution(public * oa..*Ctrl.save*(..))"
            + " || execution(public * oa..*Ctrl.update*(..))"
            + " || execution(public * oa..*Ctrl.delete*(..))"
            + " || execution(public * oa..*Ctrl.delBy*(..))"
            + " || execution(public * oa.common.BaseCtrl+.save*(..))"
            + " || execution(public * oa.common.BaseCtrl+.update*(..))"
            + " || execution(public * oa.common.BaseCtrl+.delBy*(..))"
            + " || execution(public * oa.common.BaseCtrl+.delete*(..))")
    public void log(JoinPoint point) {

        long costTime = System.currentTimeMillis() - startTime.get();
        System.out.println(costTime);

        Object object = point.getTarget();
        //全类名
        String className = object.getClass().getName();
        //Controller的类名去掉"Ctrl"后缀作为日志的模块code
        String moduleCode = object.getClass().getSimpleName();
        moduleCode = moduleCode.replace("Ctrl", "");

        Signature signature = point.getSignature();
        //method的名称作为日志的操作code
        String methodName = signature.getName();

        //获取参数名称与参数值，并封装成Map
        MethodSignature methodSignature = (MethodSignature) signature;
        String[] paramNames = methodSignature.getParameterNames();
        Object[] args = point.getArgs();
        Map<String, Object> map = new HashMap<>();
        for (int i = 0; i < paramNames.length; i++) {
            Object obj = args[i];
            if (obj == null) {    //排除参数值为null的情况
                map.put(paramNames[i], null);
            } else {
                Class<?> cls = obj.getClass();
                if (!Errors.class.isAssignableFrom(cls))    //排除Spring的入参验证结果Errors
                    map.put(paramNames[i], args[i]);
            }
        }
        String paramsJson = JacksonJson.beanToJson(map);
        Record logRecord = new Record(LoginCtrl.getLoginUserInfo().getId(), moduleCode, methodName, className, methodName, paramsJson, costTime);
        logRecordBiz.save(logRecord);
    }
}