update 优化 redis锁pr 代码

疯狂的狮子li

2021-06-04 dcf125a08ad3f0694815265850df78d62f1444fa

update 优化 redis锁pr 代码

 ruoyi-admin/src/main/java/com/ruoyi/web/controller/system/RedisLockController.java

ÎļþÒÑɾ³ý

 ruoyi-common/src/main/java/com/ruoyi/common/annotation/RedisLock.java

@@ -7,6 +7,8 @@

/**
 * åˆ†å¸ƒå¼é”ï¼ˆæ³¨è§£æ¨¡å¼ï¼Œä¸æŽ¨èä½¿ç”¨ï¼Œæœ€å¥½ç”¨é”çš„工具类）
 *
 * @author shenxinquan
 */

@Target({ElementType.METHOD})
@@ -14,14 +16,12 @@
public @interface RedisLock {

    /**
     * é”è¿‡æœŸæ—¶é—´
     * @return
     * é”è¿‡æœŸæ—¶é—´ é»˜è®¤30秒
     */
    int expireTime() default 30;//30秒
    int expireTime() default 30;

    /**
     * é”key值
     * @return
     */
    String key() default "redisLockKey";
}

 ruoyi-common/src/main/java/com/ruoyi/common/core/redis/RedisLockUtil.java

ÎļþÒÑɾ³ý

 ruoyi-demo/src/main/java/com/ruoyi/demo/controller/RedisLockController.java

¶Ô±ÈÐÂÎļþ
@@ -0,0 +1,35 @@
package com.ruoyi.demo.controller;

import com.ruoyi.common.annotation.RedisLock;
import com.ruoyi.common.core.domain.AjaxResult;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;


/**
 * æµ‹è¯•åˆ†å¸ƒå¼é”çš„样例
 *
 * @author shenxinquan
 */
@RestController
@RequestMapping("/demo/redisLock")
public class RedisLockController {

    /**
     * #p0 æ ‡è¯†å–第一个参数为redis锁的key
     */
    @GetMapping("/getLock")
    @RedisLock(expireTime = 10, key = "#p0")
    public AjaxResult<String> getLock(String key, String value) {
        try {
            // åŒæ—¶è¯·æ±‚排队
//            Thread.sleep(5000);
            // é”è¶…时测试
            Thread.sleep(11000);
        } catch (InterruptedException e) {
            e.printStackTrace();
        }
        return AjaxResult.success("操作成功",value);
    }
}

 ruoyi-framework/src/main/java/com/ruoyi/framework/aspectj/RedisLockAspect.java

ÎļþÃû´Ó ruoyi-common/src/main/java/com/ruoyi/common/core/redis/RedisLockAspect.java ÐÞ¸Ä
@@ -1,15 +1,15 @@
package com.ruoyi.common.core.redis;
package com.ruoyi.framework.aspectj;


import com.ruoyi.common.annotation.RedisLock;
import com.ruoyi.common.utils.file.ImageUtils;
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.slf4j.Logger;
import org.slf4j.LoggerFactory;
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;
@@ -21,15 +21,20 @@

/**
 * åˆ†å¸ƒå¼é”ï¼ˆæ³¨è§£å®žçŽ°ç‰ˆæœ¬ï¼‰
 *
 * @author shenxinquan
 */
@Component

@Slf4j
@Aspect
@Order(9)
@Component
public class RedisLockAspect {
    @Autowired
    private RedisLockUtil redisUtil;

    private static final Logger log = LoggerFactory.getLogger(RedisLockAspect.class);
    @Autowired
    private RedissonClient redissonClient;

    private static final String LOCK_TITLE = "RedisLock_";

    @Pointcut("@annotation(com.ruoyi.common.annotation.RedisLock)")
    public void annotationPointcut() {
@@ -47,6 +52,7 @@
        String key = "";
        // é»˜è®¤30秒过期时间
        int expireTime = 30;

        try {
            // å¾—到访问的方法对象
            Method method = className.getMethod(methodName, argClass);
@@ -60,18 +66,26 @@
        } catch (Exception e) {
            throw new RuntimeException("redis分布式锁注解参数异常", e);
        }
        Object res = new Object();
        if (redisUtil.acquire(key, expireTime, TimeUnit.SECONDS)) {
            try {
                res = joinPoint.proceed();
                return res;
            } catch (Exception e) {
                throw new RuntimeException(e);
            } finally {
                redisUtil.release(key);

        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分布式锁注解参数异常");
            }
        }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);
        }
    }

@@ -95,9 +109,6 @@

    /**
     * èŽ·å–key中#p0中的参数名称
     *
     * @param key
     * @return
     */
    private static List<String> getKeyParsList(String key) {
        List<String> listPar = new ArrayList<>();
@@ -107,7 +118,7 @@
            String parName;
            int indexPre = key.indexOf("#");
            if (plusIndex > 0) {
                indexNext = key.indexOf("#") + key.substring(key.indexOf("#")).indexOf("+");
                indexNext = key.indexOf("#") + plusIndex;
                parName = key.substring(indexPre, indexNext);
            } else {
                parName = key.substring(indexPre);
@@ -121,4 +132,36 @@
        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());
    }

}