|
@@ -1,137 +0,0 @@
|
|
|
-package com.ruoyi.framework.aspectj;
|
|
|
-
|
|
|
-
|
|
|
-import com.ruoyi.common.annotation.RedisLock;
|
|
|
-import com.ruoyi.common.constant.Constants;
|
|
|
-import com.ruoyi.common.core.redis.RedisLockManager;
|
|
|
-import lombok.extern.slf4j.Slf4j;
|
|
|
-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.core.annotation.Order;
|
|
|
-import org.springframework.stereotype.Component;
|
|
|
-
|
|
|
-import java.lang.reflect.Method;
|
|
|
-import java.util.ArrayList;
|
|
|
-import java.util.List;
|
|
|
-import java.util.concurrent.TimeUnit;
|
|
|
-
|
|
|
-/**
|
|
|
- * 分布式锁(注解实现版本)
|
|
|
- *
|
|
|
- * @author shenxinquan
|
|
|
- */
|
|
|
-
|
|
|
-@Slf4j
|
|
|
-@Aspect
|
|
|
-@Order(9)
|
|
|
-@Component
|
|
|
-public class RedisLockAspect {
|
|
|
-
|
|
|
- @Autowired
|
|
|
- private RedisLockManager redisLockManager;
|
|
|
-
|
|
|
- @Pointcut("@annotation(com.ruoyi.common.annotation.RedisLock)")
|
|
|
- public void annotationPointcut() {
|
|
|
- }
|
|
|
-
|
|
|
- @Around("annotationPointcut()")
|
|
|
- public Object doAround(ProceedingJoinPoint joinPoint) throws Throwable {
|
|
|
- // 获得当前访问的class
|
|
|
- Class<?> className = joinPoint.getTarget().getClass();
|
|
|
- // 获得访问的方法名
|
|
|
- String methodName = joinPoint.getSignature().getName();
|
|
|
- // 得到方法的参数的类型
|
|
|
- Class<?>[] argClass = ((MethodSignature) joinPoint.getSignature()).getParameterTypes();
|
|
|
- Object[] args = joinPoint.getArgs();
|
|
|
- String key = "";
|
|
|
- // 默认30秒过期时间
|
|
|
- int expireTime = 30;
|
|
|
-
|
|
|
- try {
|
|
|
- // 得到访问的方法对象
|
|
|
- Method method = className.getMethod(methodName, argClass);
|
|
|
- method.setAccessible(true);
|
|
|
- // 判断是否存在@RedisLock注解
|
|
|
- if (method.isAnnotationPresent(RedisLock.class)) {
|
|
|
- RedisLock annotation = method.getAnnotation(RedisLock.class);
|
|
|
- key = getRedisKey(args, annotation.key());
|
|
|
- expireTime = getExpireTime(annotation);
|
|
|
- }
|
|
|
- } catch (Exception e) {
|
|
|
- throw new RuntimeException("redis分布式锁注解参数异常", e);
|
|
|
- }
|
|
|
-
|
|
|
- // 声明锁名称
|
|
|
- key = Constants.REDIS_LOCK_KEY + key;
|
|
|
- Object res;
|
|
|
- try {
|
|
|
- if (redisLockManager.getLock(key, expireTime, TimeUnit.SECONDS)) {
|
|
|
- log.info("lock => key : " + key + " , ThreadName : " + Thread.currentThread().getName());
|
|
|
- try {
|
|
|
- res = joinPoint.proceed();
|
|
|
- return res;
|
|
|
- } catch (Exception e) {
|
|
|
- throw new RuntimeException(e);
|
|
|
- } finally {
|
|
|
- redisLockManager.unLock(key);
|
|
|
- log.info("unlock => key : " + key + " , ThreadName : " + Thread.currentThread().getName());
|
|
|
- }
|
|
|
- } else {
|
|
|
- throw new RuntimeException("redis分布式锁注解参数异常");
|
|
|
- }
|
|
|
- } catch (IllegalMonitorStateException e) {
|
|
|
- log.error("lock timeout => key : " + key + " , ThreadName : " + Thread.currentThread().getName());
|
|
|
- throw new RuntimeException("lock timeout => key : " + key);
|
|
|
- } catch (Exception e) {
|
|
|
- throw new Exception("redis分布式未知异常", e);
|
|
|
- }
|
|
|
- }
|
|
|
-
|
|
|
- private int getExpireTime(RedisLock annotation) {
|
|
|
- return annotation.expireTime();
|
|
|
- }
|
|
|
-
|
|
|
- private String getRedisKey(Object[] args, String primalKey) {
|
|
|
- if (args.length == 0) {
|
|
|
- return primalKey;
|
|
|
- }
|
|
|
- // 获取#p0...集合
|
|
|
- List<String> keyList = getKeyParsList(primalKey);
|
|
|
- for (String keyName : keyList) {
|
|
|
- int keyIndex = Integer.parseInt(keyName.toLowerCase().replace("#p", ""));
|
|
|
- Object parValue = args[keyIndex];
|
|
|
- primalKey = primalKey.replace(keyName, String.valueOf(parValue));
|
|
|
- }
|
|
|
- return primalKey.replace("+", "").replace("'", "");
|
|
|
- }
|
|
|
-
|
|
|
- /**
|
|
|
- * 获取key中#p0中的参数名称
|
|
|
- */
|
|
|
- private static List<String> getKeyParsList(String key) {
|
|
|
- List<String> listPar = new ArrayList<>();
|
|
|
- if (key.contains("#")) {
|
|
|
- int plusIndex = key.substring(key.indexOf("#")).indexOf("+");
|
|
|
- int indexNext = 0;
|
|
|
- String parName;
|
|
|
- int indexPre = key.indexOf("#");
|
|
|
- if (plusIndex > 0) {
|
|
|
- indexNext = key.indexOf("#") + plusIndex;
|
|
|
- parName = key.substring(indexPre, indexNext);
|
|
|
- } else {
|
|
|
- parName = key.substring(indexPre);
|
|
|
- }
|
|
|
- listPar.add(parName.trim());
|
|
|
- key = key.substring(indexNext + 1);
|
|
|
- if (key.contains("#")) {
|
|
|
- listPar.addAll(getKeyParsList(key));
|
|
|
- }
|
|
|
- }
|
|
|
- return listPar;
|
|
|
- }
|
|
|
-
|
|
|
-}
|