完成审计相关功能的开发。

This commit is contained in:
chenxudong 2025-02-14 10:39:15 +08:00
parent 4294808f98
commit 02fa119bcb
2 changed files with 9 additions and 3 deletions

View File

@ -1,6 +1,7 @@
package com.electromagnetic.industry.software.manage.aop; package com.electromagnetic.industry.software.manage.aop;
import cn.hutool.json.JSONUtil; import cn.hutool.json.JSONUtil;
import com.electromagnetic.industry.software.common.pojo.UserLoginInfo;
import com.electromagnetic.industry.software.common.resp.ElectromagneticResult; import com.electromagnetic.industry.software.common.resp.ElectromagneticResult;
import com.electromagnetic.industry.software.common.util.ElectromagneticResultUtil; import com.electromagnetic.industry.software.common.util.ElectromagneticResultUtil;
import com.electromagnetic.industry.software.common.util.UserThreadLocal; import com.electromagnetic.industry.software.common.util.UserThreadLocal;
@ -18,6 +19,7 @@ import javax.servlet.ServletRequest;
import javax.servlet.ServletResponse; import javax.servlet.ServletResponse;
import java.util.HashMap; import java.util.HashMap;
import java.util.Map; import java.util.Map;
import java.util.Objects;
@Aspect @Aspect
@Component @Component
@ -67,6 +69,12 @@ public class ServiceAspect {
StopWatch stopwatch = new StopWatch(); StopWatch stopwatch = new StopWatch();
stopwatch.start("接口:" + methodInfo); stopwatch.start("接口:" + methodInfo);
Object rvt = jp.proceed(); Object rvt = jp.proceed();
if (Objects.isNull(UserThreadLocal.getUser())) {
UserLoginInfo userLoginInfo = new UserLoginInfo();
UserThreadLocal.set(userLoginInfo);
}
UserThreadLocal.setReqArgs(paramInfo); UserThreadLocal.setReqArgs(paramInfo);
if (rvt instanceof ResponseEntity) { if (rvt instanceof ResponseEntity) {
UserThreadLocal.setRes(ElectromagneticResultUtil.success("")); UserThreadLocal.setRes(ElectromagneticResultUtil.success(""));

View File

@ -124,11 +124,9 @@ public class LoginInterceptor implements HandlerInterceptor {
userAccessLog.setAccessSuccess(false); userAccessLog.setAccessSuccess(false);
userAccessLog.setFailureReason(result.getErrorMessage()); userAccessLog.setFailureReason(result.getErrorMessage());
} }
} else { } else { // 返回为ResponseEntity且状态为失败
userAccessLog.setAccessSuccess(false); userAccessLog.setAccessSuccess(false);
} }
// TODO 如果是下载的文件那么返回的是ResponseEntity而不是ElectromagneticResult
userAccessLogMapper.insert(userAccessLog); userAccessLogMapper.insert(userAccessLog);
UserThreadLocal.remove(); UserThreadLocal.remove();
} }