Initial project

src/main/java/com/imyeyu/spring/annotation/AOPLogInterceptor.java

@@ -0,0 +1,110 @@
+package com.imyeyu.spring.annotation;
+
+import com.imyeyu.java.TimiJava;
+import com.imyeyu.spring.TimiSpring;
+import com.imyeyu.spring.bean.PageResult;
+import com.imyeyu.spring.entity.IDEntity;
+import com.imyeyu.utils.Text;
+import org.aspectj.lang.JoinPoint;
+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.annotation.Before;
+import org.aspectj.lang.annotation.Pointcut;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+import org.springframework.stereotype.Component;
+
+/**
+ * AOP 切面日志
+ *
+ * @author 夜雨
+ * @version 2021-08-17 16:26
+ */
+@Aspect
+@Component
+public class AOPLogInterceptor {
+
+	public static final String REQUEST_ID = "TIMI_SPRING_REQUEST_ID";
+
+	private static final Logger log = LoggerFactory.getLogger(AOPLogInterceptor.class);
+
+	/** 注入注解 */
+	@Pointcut("@annotation(annotation.com.imyeyu.spring.AOPLog)")
+	public void logPointCut() {
+	}
+
+	/**
+	 * 执行前
+	 *
+	 * @param joinPoint 切入点
+	 */
+	@Before("logPointCut()")
+	public void doBefore(JoinPoint joinPoint) {
+		String uuid = Text.tempUUID();
+		TimiSpring.setSessionAttr(REQUEST_ID, uuid);
+		log.info("ID: {} Request  -> IP: {}, URI: {}", uuid, TimiSpring.getRequestIP(), TimiSpring.getRequest().getRequestURI());
+	}
+
+	/**
+	 * 执行后
+	 *
+	 * @param response 返回内容
+	 * @throws Throwable 异常
+	 */
+	@AfterReturning(returning = "response", pointcut = "logPointCut()")
+	public void doAfterReturning(Object response) throws Throwable {
+		String msg = "ID: {} Response <- Return.";
+		if (response instanceof IDEntity<?> entity) {
+			// 返回实体
+			msg += entity.getClass().getSimpleName() + "." + entity.getId();
+		} else if (response instanceof PageResult<?> pageResult) {
+			// 返回数组
+			if (pageResult.getList().isEmpty()) {
+				msg += "PageResult<?> Empty";
+			} else {
+				if (pageResult.getList().get(0) == null) {
+					msg += "PageResult<?>." + pageResult.getList().size();
+				} else {
+					msg += "PageResult<" + pageResult.getList().get(0).getClass().getSimpleName() + ">[" + pageResult.getList().size() + "]";
+				}
+			}
+			// 返回数据页
+		} else if (response instanceof String string) {
+			// 返回字符串
+			if (string.length() < 64) {
+				msg += string;
+			} else {
+				msg += string.substring(0, 64) + "..";
+			}
+			msg = msg.replaceAll("[\\r\\n]+", "");
+		} else if (response instanceof Boolean bool) {
+			// 返回布尔值
+			msg += bool;
+		} else if (response instanceof Number number) {
+			// 返回数字
+			msg += response.getClass().getSimpleName() + ".[" + number.doubleValue() + "]";
+		} else {
+			// 其他对象
+			if (TimiJava.isNotEmpty(response)) {
+				msg += response.getClass().getSimpleName();
+			} else {
+				msg += "NULL";
+			}
+		}
+		log.info(msg, TimiSpring.getSessionAttr(REQUEST_ID));
+	}
+
+	/**
+	 * 环绕
+	 *
+	 * @param pjp 切入点
+	 * @return 执行返回
+	 * @throws Throwable 异常
+	 */
+	@Around("logPointCut()")
+	public Object doAround(ProceedingJoinPoint pjp) throws Throwable {
+		return pjp.proceed();
+	}
+}