liuchang/mall-ebtp-cloud
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix(uboot-common): 幂等性修改redis key通过线程存储

Browse Source
This commit is contained in:
chuhang
2021-09-30 09:44:47 +08:00
parent 9724d7ae1d
commit a256bc0ff4
1 changed files with 24 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

31
uboot-common/src/main/java/com/chinaunicom/mall/ebtp/common/idempotent/aspect/IdempotentAspect.java
View File

@ -5,6 +5,7 @@ import com.chinaunicom.mall.ebtp.common.exception.common.CommonExceptionEnum;
import com.chinaunicom.mall.ebtp.common.idempotent.annotation.Idempotent; import com.chinaunicom.mall.ebtp.common.idempotent.annotation.Idempotent;
import com.chinaunicom.mall.ebtp.common.util.HttpContextUtils; import com.chinaunicom.mall.ebtp.common.util.HttpContextUtils;
import com.chinaunicom.mall.ebtp.common.util.JsonUtils; import com.chinaunicom.mall.ebtp.common.util.JsonUtils;
import io.netty.util.concurrent.FastThreadLocal;
import lombok.extern.slf4j.Slf4j; import lombok.extern.slf4j.Slf4j;
import org.aspectj.lang.JoinPoint; import org.aspectj.lang.JoinPoint;
import org.aspectj.lang.annotation.After; import org.aspectj.lang.annotation.After;
@ -20,6 +21,7 @@ import javax.servlet.http.HttpServletRequest;
import java.lang.reflect.Method; import java.lang.reflect.Method;
import java.time.LocalDateTime; import java.time.LocalDateTime;
import java.util.Arrays; import java.util.Arrays;
import java.util.Optional;
import java.util.concurrent.TimeUnit; import java.util.concurrent.TimeUnit;
/** /**
@ -35,6 +37,9 @@ public class IdempotentAspect {
@Autowired @Autowired
private RedisTemplate<String, Object> redisTemplate; private RedisTemplate<String, Object> redisTemplate;
/**redis键值*/
public static final FastThreadLocal<String> REDIS_KEY = new FastThreadLocal<>();
@Pointcut("@annotation(com.chinaunicom.mall.ebtp.common.idempotent.annotation.Idempotent)") @Pointcut("@annotation(com.chinaunicom.mall.ebtp.common.idempotent.annotation.Idempotent)")
public void pointCut() { public void pointCut() {
} }
@ -52,6 +57,10 @@ public class IdempotentAspect {
Idempotent idempotent = method.getAnnotation(Idempotent.class); Idempotent idempotent = method.getAnnotation(Idempotent.class);
String key = getKey(joinPoint, request); String key = getKey(joinPoint, request);
/*清理线程*/
this.removeThreadLocal();
/*启用线程*/
REDIS_KEY.set(key);
long expireTime = idempotent.expireTime(); long expireTime = idempotent.expireTime();
String info = idempotent.info(); String info = idempotent.info();
@ -87,21 +96,23 @@ public class IdempotentAspect {
} }
@After("pointCut()") @After("pointCut()")
public void afterPointCut(JoinPoint joinPoint) { public void afterPointCut() {
HttpServletRequest request = HttpContextUtils.getHttpServletRequest(); String key = REDIS_KEY.get();
String key = getKey(joinPoint, request); Object object = redisTemplate.opsForValue().get(key);
Optional.ofNullable(object).ifPresent(o-> {
Object o = redisTemplate.opsForValue().get(key);
if (o == null) {
return;
}
boolean delKey = (boolean) o; boolean delKey = (boolean) o;
if (delKey) { if (delKey) {
redisTemplate.delete(key); redisTemplate.delete(key);
log.info("[idempotent]:has removed key={}", key); log.info("[idempotent]:has removed key={}", key);
} }
});
/*清理线程*/
this.removeThreadLocal();
}
private void removeThreadLocal() {
REDIS_KEY.remove();
} }
} }