疯狂的狮子li
2021-06-16 f9967cca4ca696860986ccd3497bc5687b3544ba
ruoyi-framework/src/main/java/com/ruoyi/framework/aspectj/RedisLockAspect.java
@@ -3,14 +3,13 @@
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.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;
@@ -33,7 +32,7 @@
public class RedisLockAspect {
   @Autowired
   private RedissonClient redissonClient;
   private RedisLockManager redisLockManager;
   @Pointcut("@annotation(com.ruoyi.common.annotation.RedisLock)")
   public void annotationPointcut() {
@@ -70,14 +69,16 @@
      key = Constants.REDIS_LOCK_KEY + key;
      Object res;
      try {
         if (acquire(key, expireTime, TimeUnit.SECONDS)) {
         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 {
               release(key);
               redisLockManager.unLock(key);
               log.info("unlock => key : " + key + " , ThreadName : " + Thread.currentThread().getName());
            }
         } else {
            throw new RuntimeException("redis分布式锁注解参数异常");
@@ -131,34 +132,6 @@
         }
      }
      return listPar;
   }
   /**
    * 加锁(RLock)带超时时间的
    */
   private boolean acquire(String key, long expire, TimeUnit expireUnit) {
      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) {
      //获取所对象
      RLock mylock = redissonClient.getLock(lockName);
      //释放锁(解锁)
      mylock.unlock();
      log.info("unlock => key : " + lockName + " , ThreadName : " + Thread.currentThread().getName());
   }
}