|
|
@@ -0,0 +1,167 @@
|
|
|
+package com.ruoyi.framework.aspectj;
|
|
|
+
|
|
|
+
|
|
|
+import com.ruoyi.common.annotation.RedisLock;
|
|
|
+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.redisson.api.RLock;
|
|
|
+import org.redisson.api.RedissonClient;
|
|
|
+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 RedissonClient redissonClient;
|
|
|
+
|
|
|
+ private static final String LOCK_TITLE = "RedisLock_";
|
|
|
+
|
|
|
+ @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);
|
|
|
+ }
|
|
|
+
|
|
|
+ Object res;
|
|
|
+ try {
|
|
|
+ if (acquire(key, expireTime, TimeUnit.SECONDS)) {
|
|
|
+ try {
|
|
|
+ res = joinPoint.proceed();
|
|
|
+ return res;
|
|
|
+ } catch (Exception e) {
|
|
|
+ throw new RuntimeException(e);
|
|
|
+ } finally {
|
|
|
+ release(key);
|
|
|
+ }
|
|
|
+ } 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;
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 加锁（RLock）带超时时间的
|
|
|
+ */
|
|
|
+ private boolean acquire(String key, long expire, TimeUnit expireUnit) {
|
|
|
+ //声明key对象
|
|
|
+ key = LOCK_TITLE + key;
|
|
|
+ try {
|
|
|
+ //获取锁对象
|
|
|
+ RLock mylock = redissonClient.getLock(key);
|
|
|
+ //加锁，并且设置锁过期时间，防止死锁的产生
|
|
|
+ mylock.tryLock(expire, expire, expireUnit);
|
|
|
+ } catch (InterruptedException e) {
|
|
|
+ return false;
|
|
|
+ }
|
|
|
+ log.info("lock => key : " + key + " , ThreadName : " + Thread.currentThread().getName());
|
|
|
+ //加锁成功
|
|
|
+ return true;
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 锁的释放
|
|
|
+ */
|
|
|
+ private void release(String lockName) {
|
|
|
+ //必须是和加锁时的同一个key
|
|
|
+ String key = LOCK_TITLE + lockName;
|
|
|
+ //获取所对象
|
|
|
+ RLock mylock = redissonClient.getLock(key);
|
|
|
+ //释放锁（解锁）
|
|
|
+ mylock.unlock();
|
|
|
+ log.info("unlock => key : " + key + " , ThreadName : " + Thread.currentThread().getName());
|
|
|
+ }
|
|
|
+
|
|
|
+}
|
疯狂的狮子Li