packagecom.smartdata.pms.log;importcn.hutool.core.util.StrUtil;importcn.hutool.core.util.URLUtil;importcn.hutool.json.JSONUtil;importio.swagger.annotations.ApiOperation;importlombok.extern.slf4j.Slf4j;importorg.aspectj.lang.JoinPoint;importorg.aspectj.lang.ProceedingJoinPoint;importorg.aspectj.lang.Signature;import org.aspectj.lang.annotation.*;importorg.aspectj.lang.reflect.MethodSignature;importorg.springframework.core.annotation.Order;importorg.springframework.stereotype.Component;importorg.springframework.util.StringUtils;importorg.springframework.web.bind.annotation.RequestBody;importorg.springframework.web.bind.annotation.RequestParam;importorg.springframework.web.context.request.RequestContextHolder;importorg.springframework.web.context.request.ServletRequestAttributes;importjavax.servlet.http.HttpServletRequest;importjava.lang.reflect.Method;importjava.lang.reflect.Parameter;importjava.util.ArrayList;importjava.util.HashMap;importjava.util.List;importjava.util.Map;/*** @ProjectName: smartdata
* @Package: com.smartdata.pms.log
* @ClassName: WebLogAspect
* @Author: heluwei
* @Description: 统一日志处理切面
* @Aspect:用于定义切面
* @Before:通知方法会在目标方法调用之前执行
* @After:通知方法会在目标方法返回或抛出异常后执行
* @AfterReturning:通知方法会在目标方法返回后执行
* @AfterThrowing:通知方法会在目标方法抛出异常后执行
* @Around:通知方法会将目标方法封装起来
* @Pointcut:定义切点表达式
* @Date: 2020/3/25 11:10
* @Version: 1.0*/@Aspect//用于定义切面
@Component //注入到Spring容器中
@Order(1) //定义Spring IOC容器中Bean的执行顺序的优先级 值越小越先执行
@Slf4jpublic classWebLogAspect {
@Pointcut("execution(public * com.smartdata.pms.controller.*.*(..))")public voidwebLog() {
}
@Before("webLog()")public void doBefore(JoinPoint joinPoint) throwsThrowable {
}
@AfterReturning(value= "webLog()", returning = "ret")public void doAfterReturning(Object ret) throwsThrowable {
}
@Around("webLog()")public Object doAround(ProceedingJoinPoint joinPoint) throwsThrowable {long startTime =System.currentTimeMillis();//获取当前请求对象
ServletRequestAttributes attributes =(ServletRequestAttributes) RequestContextHolder.getRequestAttributes();
HttpServletRequest request=attributes.getRequest();//记录请求信息
WebLog webLog = newWebLog();
Object result=joinPoint.proceed();
Signature signature=joinPoint.getSignature();
MethodSignature methodSignature=(MethodSignature) signature;
Method method=methodSignature.getMethod();if (method.isAnnotationPresent(ApiOperation.class)) {
ApiOperation apiOperation= method.getAnnotation(ApiOperation.class);
webLog.setDescription(apiOperation.value());
}long endTime =System.currentTimeMillis();
String urlStr=request.getRequestURL().toString();
webLog.setBasePath(StrUtil.removeSuffix(urlStr, URLUtil.url(urlStr).getPath()));
webLog.setIp(request.getRemoteUser());
webLog.setMethod(request.getMethod());
webLog.setParameter(getParameter(method, joinPoint.getArgs()));
webLog.setResult(result);
webLog.setSpendTime((int) (endTime -startTime));
webLog.setStartTime(startTime);
webLog.setUri(request.getRequestURI());
webLog.setUrl(request.getRequestURL().toString());
log.info("{}", JSONUtil.parse(webLog));returnresult;
}/*** 根据方法和传入的参数获取请求参数*/
privateObject getParameter(Method method, Object[] args) {
List argList = new ArrayList<>();
Parameter[] parameters=method.getParameters();for (int i = 0; i < parameters.length; i++) {//将RequestBody注解修饰的参数作为请求参数
RequestBody requestBody = parameters[i].getAnnotation(RequestBody.class);if (requestBody != null) {
argList.add(args[i]);
}//将RequestParam注解修饰的参数作为请求参数
RequestParam requestParam = parameters[i].getAnnotation(RequestParam.class);if (requestParam != null) {
Map map = new HashMap<>();
String key=parameters[i].getName();if (!StringUtils.isEmpty(requestParam.value())) {
key=requestParam.value();
}
map.put(key, args[i]);
argList.add(map);
}
}if (argList.size() == 0) {return null;
}else if (argList.size() == 1) {return argList.get(0);
}else{returnargList;
}
}
}