From 152781b05131e48bf6e94d71cc72dd54af52a3fb Mon Sep 17 00:00:00 2001
From: houzhongjian <houzhongyi@126.com>
Date: 星期四, 10 四月 2025 14:13:29 +0800
Subject: [PATCH] 恢复iailab-framework
---
iailab-framework/iailab-common-protection/src/main/java/com/iailab/framework/idempotent/core/aop/IdempotentAspect.java | 68 ++++++++++++++++++++++++++++++++++
1 files changed, 68 insertions(+), 0 deletions(-)
diff --git a/iailab-framework/iailab-common-protection/src/main/java/com/iailab/framework/idempotent/core/aop/IdempotentAspect.java b/iailab-framework/iailab-common-protection/src/main/java/com/iailab/framework/idempotent/core/aop/IdempotentAspect.java
new file mode 100644
index 0000000..603f270
--- /dev/null
+++ b/iailab-framework/iailab-common-protection/src/main/java/com/iailab/framework/idempotent/core/aop/IdempotentAspect.java
@@ -0,0 +1,68 @@
+package com.iailab.framework.idempotent.core.aop;
+
+import com.iailab.framework.common.exception.ServiceException;
+import com.iailab.framework.common.exception.enums.GlobalErrorCodeConstants;
+import com.iailab.framework.common.util.collection.CollectionUtils;
+import com.iailab.framework.idempotent.core.annotation.Idempotent;
+import com.iailab.framework.idempotent.core.keyresolver.IdempotentKeyResolver;
+import com.iailab.framework.idempotent.core.redis.IdempotentRedisDAO;
+import lombok.extern.slf4j.Slf4j;
+import org.aspectj.lang.ProceedingJoinPoint;
+import org.aspectj.lang.annotation.Around;
+import org.aspectj.lang.annotation.Aspect;
+import org.springframework.util.Assert;
+
+import java.util.List;
+import java.util.Map;
+
+/**
+ * 拦截声明了 {@link Idempotent} 注解的方法，实现幂等操作
+ *
+ * @author iailab
+ */
+@Aspect
+@Slf4j
+public class IdempotentAspect {
+
+ /**
+ * IdempotentKeyResolver 集合
+ */
+ private final Map<Class<? extends IdempotentKeyResolver>, IdempotentKeyResolver> keyResolvers;
+
+ private final IdempotentRedisDAO idempotentRedisDAO;
+
+ public IdempotentAspect(List<IdempotentKeyResolver> keyResolvers, IdempotentRedisDAO idempotentRedisDAO) {
+ this.keyResolvers = CollectionUtils.convertMap(keyResolvers, IdempotentKeyResolver::getClass);
+ this.idempotentRedisDAO = idempotentRedisDAO;
+ }
+
+ @Around(value = "@annotation(idempotent)")
+ public Object aroundPointCut(ProceedingJoinPoint joinPoint, Idempotent idempotent) throws Throwable {
+ // 获得 IdempotentKeyResolver
+ IdempotentKeyResolver keyResolver = keyResolvers.get(idempotent.keyResolver());
+ Assert.notNull(keyResolver, "找不到对应的 IdempotentKeyResolver");
+ // 解析 Key
+ String key = keyResolver.resolver(joinPoint, idempotent);
+
+ // 1. 锁定 Key
+ boolean success = idempotentRedisDAO.setIfAbsent(key, idempotent.timeout(), idempotent.timeUnit());
+ // 锁定失败，抛出异常
+ if (!success) {
+ log.info("[aroundPointCut][方法({}) 参数({}) 存在重复请求]", joinPoint.getSignature().toString(), joinPoint.getArgs());
+ throw new ServiceException(GlobalErrorCodeConstants.REPEATED_REQUESTS.getCode(), idempotent.message());
+ }
+
+ // 2. 执行逻辑
+ try {
+ return joinPoint.proceed();
+ } catch (Throwable throwable) {
+ // 3. 异常时，删除 Key
+ // 参考美团 GTIS 思路：https://tech.meituan.com/2016/09/29/distributed-system-mutually-exclusive-idempotence-cerberus-gtis.html
+ if (idempotent.deleteKeyWhenException()) {
+ idempotentRedisDAO.delete(key);
+ }
+ throw throwable;
+ }
+ }
+
+}
--
Gitblit v1.9.3