aspect

package com.uuzu.newbasys.aspect;

import java.io.BufferedReader;
import java.lang.reflect.Method;

import javax.servlet.http.HttpServletRequest;

import com.uuzu.newbasys.annotation.LogAPIAccessAnnotation;
import com.uuzu.newbasys.mapper.ba.log.LogAPIAccessMapper;
import com.uuzu.newbasys.model.log.LogAPIAccessModel;

import org.aspectj.lang.ProceedingJoinPoint;
import org.aspectj.lang.annotation.AfterReturning;
import org.aspectj.lang.annotation.Around;
import org.aspectj.lang.annotation.Aspect;
import org.aspectj.lang.reflect.MethodSignature;
import org.json.JSONObject;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.security.core.context.SecurityContextHolder;
import org.springframework.security.core.userdetails.UserDetails;
import org.springframework.stereotype.Component;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

@Aspect
@Component
public class LogAPIAccessAspect {

    private static final Logger logger = LoggerFactory.getLogger(LogAPIAccessAspect.class);

    @Autowired
    private LogAPIAccessMapper mapper;

    @Autowired
    private HttpServletRequest request;

    ThreadLocal<LogAPIAccessModel> logModelTh = new ThreadLocal<>();

    @Around("@annotation(com.uuzu.newbasys.annotation.LogAPIAccessAnnotation)")
    public Object logAPIAccess(ProceedingJoinPoint joinPoint) throws Throwable {

        LogAPIAccessModel logModel = new LogAPIAccessModel();
        logModelTh.set(logModel);

        // username
        String userName = "anonymous";

        if (SecurityContextHolder.getContext().getAuthentication() != null) {
            UserDetails userDetails = (UserDetails) SecurityContextHolder.getContext().getAuthentication()
                    .getPrincipal();
            userName = userDetails.getUsername();
        }

        // 记录下请求内容
        String url = request.getRequestURL().toString();
        String httpMethod = request.getMethod();
        String ip = request.getRemoteAddr();

        MethodSignature methodSignature = (MethodSignature) joinPoint.getSignature();
        Method method = methodSignature.getMethod();
        // 类名
        String className = methodSignature.getDeclaringTypeName();
        // 方法名
        String methodName = methodSignature.getName();
        // 路径参数
        JSONObject params = new JSONObject(request.getParameterMap());
        // requestBody
        StringBuilder sb = new StringBuilder();
        BufferedReader reader = request.getReader();
        try {
            String line;
            while ((line = reader.readLine()) != null) {
                sb.append(line).append('\n');
            }
        } finally {
            reader.close();
        }
        String requestBody = null;
        if (sb.toString().length() > 0) {
            requestBody = sb.toString();
        }

        long start = System.currentTimeMillis();

        Object proceed = joinPoint.proceed();

        // 执行时间
        long executionTime = System.currentTimeMillis() - start;

        // 注解上的描述信息
        LogAPIAccessAnnotation logAnnotation = method.getAnnotation(LogAPIAccessAnnotation.class);
        String logAnnotationValue = "-";
        if (logAnnotation != null) {
            logAnnotationValue = logAnnotation.value();
        }
        String[] logAnnotationValueArray = logAnnotationValue.split("-");

        logModel.setUserName(userName);
        logModel.setProject(logAnnotationValueArray[0]);
        logModel.setPage(logAnnotationValueArray[1]);
        logModel.setDetail(logAnnotationValueArray[2]);
        logModel.setHttpMethod(httpMethod);
        logModel.setUrl(url);
        logModel.setExecutionTime(executionTime);
        logModel.setParams(params.toString());
        logModel.setRequestBody(requestBody);
        logModel.setClassName(className);
        logModel.setMethodName(methodName);
        logModel.setIp(ip);

        return proceed;
    }

    @AfterReturning(pointcut = "@annotation(com.uuzu.newbasys.annotation.LogAPIAccessAnnotation)", returning = "response")
    public void doAfterReturning(Object response) throws Throwable {

        // 处理完请求，返回内容
        LogAPIAccessModel logModel = logModelTh.get();
        logModel.setResponse(response.toString());

        try {
            mapper.save(logModel);
        } catch (Exception e) {
            logger.error("接口统计报错！");
            logger.error(e.getMessage());
        } finally {
            logModelTh.remove();
        }
    }
}

package com.uuzu.newbasys.annotation;

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

@Target(ElementType.METHOD)
@Retention(RetentionPolicy.RUNTIME)
public @interface LogAPIAccessAnnotation {
    // value 格式为 page-detail
    // 实时运营商-搜索
    String value() default "";
}

@PostMapping("/logPostMethodTest")
@LogAPIAccessAnnotation("yoobi-test-log")
public Map<String, Object> logPostMethodTest(@RequestBody Map<String, Object> requestBody,@RequestParam("name") String name,
@RequestParam("age") int age) {

    return requestBody;
}