登录日志记录

IDEA/src/main/java/com/example/api/aspect/LogAspect.java

@@ -0,0 +1,86 @@
+package com.example.api.aspect;
+
+import com.example.api.annotation.Log;
+import com.example.api.model.entity.SystemLog;
+import com.example.api.service.SystemLogService;
+import com.example.api.utils.IpUtil;
+import com.example.api.utils.JwtTokenUtil;
+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 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 java.lang.reflect.Method;
+import java.time.LocalDateTime;
+
+/**
+ * LogAspect 类是一个切面（Aspect），用于记录系统日志。
+ */
+@Aspect // 表明这是一个切面
+@Component // 表明这是一个 Spring 组件
+public class LogAspect {
+    @Autowired // 自动注入 SystemLogService
+    private SystemLogService logService;
+
+    /**
+     * 定义切点，匹配所有标记了 @Log 注解的方法。
+     */
+    @Pointcut("@annotation(com.example.api.annotation.Log)")
+    public void pt(){}
+
+    /**
+     * 环绕通知，用于在目标方法执行前后记录日志。
+     * @param point 连接点
+     * @return 方法执行结果
+     * @throws Throwable 可能抛出的异常
+     */
+    @Around("pt()")
+    public Object Around(ProceedingJoinPoint point) throws Throwable {
+        // 记录开始时间
+        long beginTime = System.currentTimeMillis();
+        Object res = null;
+        try {
+            // 执行方法
+            res = point.proceed();
+        } finally {
+            // 计算执行时长
+            long time = System.currentTimeMillis() - beginTime;
+            recordLog(point);
+        }
+        return res;
+    }
+
+    /**
+     * 记录日志的方法。
+     * @param point 连接点
+     */
+    private void recordLog(ProceedingJoinPoint point){
+        // 获取当前请求对象
+        ServletRequestAttributes requestAttributes = (ServletRequestAttributes) RequestContextHolder.getRequestAttributes();
+        HttpServletRequest request = requestAttributes.getRequest();
+        // 获取目标方法签名
+        MethodSignature signature = (MethodSignature) point.getSignature();
+        Method method = signature.getMethod();
+        Log annotation = method.getAnnotation(Log.class);
+        // 封装日志对象
+        SystemLog systemLog = new SystemLog();
+        systemLog.setModule(annotation.module());
+        systemLog.setBusinessType(annotation.type().getName());
+        systemLog.setIp(IpUtil.getIpAddr(request));
+        systemLog.setTime(LocalDateTime.now());
+        // 获取方法的全路径
+        systemLog.setMethod(signature.getDeclaringTypeName()+"."+signature.getName());
+        // 获取token,并解析token来获取当前账号
+        String token = request.getHeader(JwtTokenUtil.TOKEN_HEADER);
+        systemLog.setAccount(JwtTokenUtil.getUsername(token));
+        // 持久化到数据库
+        logService.record(systemLog);
+    }
+}