package com.lf.server.aspect;
|
|
import com.lf.server.annotation.SysLog;
|
import com.lf.server.entity.sys.OperateEntity;
|
import com.lf.server.entity.sys.UserEntity;
|
import com.lf.server.helper.WebHelper;
|
import com.lf.server.service.sys.OperateService;
|
import com.lf.server.service.sys.TokenService;
|
import io.swagger.annotations.Api;
|
import org.apache.commons.logging.Log;
|
import org.apache.commons.logging.LogFactory;
|
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.beans.factory.annotation.Autowired;
|
import org.springframework.stereotype.Component;
|
|
import javax.servlet.http.HttpServletRequest;
|
import java.lang.reflect.Method;
|
|
/**
|
* 日志切面类
|
* @author WWW
|
*/
|
@Aspect
|
@Component
|
public class LogAspect {
|
@Autowired
|
private TokenService tokenService;
|
|
@Autowired
|
private OperateService operateService;
|
|
private final static String SELECT = "select";
|
|
private final static String INSERT = "insert";
|
|
private final static String UPDATE = "update";
|
|
private final static String DELETE = "delete";
|
|
private final static String UPLOAD = "upload";
|
|
private final static String DOWNLOAD = "download";
|
|
private final static String COUNT = "count";
|
|
private final static String SPLIT = "\\\\";
|
|
private static final Log log = LogFactory.getLog(LogAspect.class);
|
|
@Pointcut("@annotation(com.lf.server.annotation.SysLog)")
|
public void logPointCut() {
|
//
|
}
|
|
/**
|
* 环绕通知 @Around，也可以使用 @Before (前置通知) 或 @After (后置通知)
|
*/
|
@Around("logPointCut()")
|
public Object around(ProceedingJoinPoint point) throws Throwable {
|
long beginTime = System.currentTimeMillis();
|
|
// 执行方法
|
Object result = point.proceed();
|
|
// 执行时长(毫秒)
|
long time = System.currentTimeMillis() - beginTime;
|
|
// 保存日志
|
saveLog(point, time);
|
|
return result;
|
}
|
|
/**
|
* 保存日志
|
*/
|
private void saveLog(ProceedingJoinPoint joinPoint, long time) {
|
try {
|
MethodSignature signature = (MethodSignature) joinPoint.getSignature();
|
Class<?> clazz = joinPoint.getTarget().getClass();
|
Method method = signature.getMethod();
|
|
HttpServletRequest req = WebHelper.getRequest();
|
String url = req.getServletPath();
|
|
OperateEntity oe = new OperateEntity();
|
oe.setIp(WebHelper.getIpAddress(req));
|
oe.setUrl(url);
|
oe.setExec(time);
|
|
// 设置模块
|
Api api = clazz.getAnnotation(Api.class);
|
if (api != null) {
|
String[] strs = api.tags()[0].split(SPLIT);
|
oe.setModular1(strs[0]);
|
oe.setModular2(strs[1]);
|
}
|
|
// 设置类名\方法名、类别
|
oe.setClazz(clazz.getName() + "." + method.getName());
|
oe.setType(getType(method.getName()));
|
|
// 设置备注
|
SysLog sysLog = method.getAnnotation(SysLog.class);
|
if (sysLog != null) {
|
oe.setBak(sysLog.value());
|
}
|
|
UserEntity ue = tokenService.getCurrentUser(req);
|
if (ue != null) {
|
oe.setUserid(ue.getId());
|
}
|
|
operateService.insertOperate(oe);
|
} catch (Exception ex) {
|
log.error(ex.getMessage(), ex);
|
}
|
}
|
|
/**
|
* 获取操作类型
|
*/
|
private static int getType(String methodName) {
|
if (methodName.contains(SELECT)) {
|
return 1;
|
}
|
if (methodName.contains(INSERT)) {
|
return 2;
|
}
|
if (methodName.contains(UPDATE)) {
|
return 3;
|
}
|
if (methodName.contains(DELETE)) {
|
return 4;
|
}
|
if (methodName.contains(UPLOAD)) {
|
return 5;
|
}
|
if (methodName.contains(DOWNLOAD)) {
|
return 6;
|
}
|
// methodName.indexOf(COUNT) > -1
|
if (methodName.contains(COUNT)) {
|
return 7;
|
}
|
|
return 0;
|
}
|
}
|
