redis分布式锁（基于注解形势）。

sxq

2021-06-04 ad6386a618eb81c401333c6541ec798932991bf6

redis分布式锁（基于注解形势）。

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

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

import com.ruoyi.common.annotation.RedisLock;
import com.ruoyi.common.core.domain.AjaxResult;
import com.ruoyi.common.core.domain.entity.SysUser;
import com.ruoyi.common.core.domain.model.LoginBody;
import com.ruoyi.common.core.domain.model.LoginUser;
import com.ruoyi.common.utils.ServletUtils;
import com.ruoyi.framework.web.service.TokenService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.RequestBody;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;


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

    @Autowired
    private TokenService tokenService;

    /**
     * #p0 æ ‡è¯†å–第一个参数为redis锁的key
     * @param loginBody
     * @return
     */
    @GetMapping("/getLock")
    @RedisLock(expireTime=10,key = "#p0")
    public AjaxResult getInfo(@RequestBody LoginBody loginBody){
        LoginUser loginUser = tokenService.getLoginUser(ServletUtils.getRequest());
        SysUser user = loginUser.getUser();
        try {
            Thread.sleep(5000);
        } catch (InterruptedException e) {
            e.printStackTrace();
        }
        return AjaxResult.success(user);
    }
}

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

¶Ô±ÈÐÂÎļþ
@@ -0,0 +1,27 @@
package com.ruoyi.common.annotation;

import java.lang.annotation.ElementType;
import java.lang.annotation.Retention;
import java.lang.annotation.RetentionPolicy;
import java.lang.annotation.Target;

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

@Target({ElementType.METHOD})
@Retention(RetentionPolicy.RUNTIME)
public @interface RedisLock {

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

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

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

¶Ô±ÈÐÂÎļþ
@@ -0,0 +1,124 @@
package com.ruoyi.common.core.redis;


import com.ruoyi.common.annotation.RedisLock;
import com.ruoyi.common.utils.file.ImageUtils;
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.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;

/**
 * åˆ†å¸ƒå¼é”ï¼ˆæ³¨è§£å®žçŽ°ç‰ˆæœ¬ï¼‰
 */
@Component
@Aspect
@Order(9)
public class RedisLockAspect {
    @Autowired
    private RedisLockUtil redisUtil;

    private static final Logger log = LoggerFactory.getLogger(RedisLockAspect.class);

    @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 = 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);
            }
        }else {
            throw new RuntimeException("redis分布式锁注解参数异常");
        }
    }

    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中的参数名称
     *
     * @param key
     * @return
     */
    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("#") + key.substring(key.indexOf("#")).indexOf("+");
                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;
    }

}

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

¶Ô±ÈÐÂÎļþ
@@ -0,0 +1,64 @@
package com.ruoyi.common.core.redis;


import org.redisson.api.RLock;
import org.redisson.api.RedissonClient;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;

import java.util.concurrent.TimeUnit;

@Component
public class RedisLockUtil {

    @Autowired
    private RedissonClient redissonClient;

    private static final String LOCK_TITLE = "redisLock_";

    private static final Logger log = LoggerFactory.getLogger(RedisLockUtil.class);

/*    public boolean getLock(String key){
        key = LOCK_TITLE + key;
        RLock mylock = redissonClient.getLock(key);
        System.err.println("======lock======" + Thread.currentThread().getName());
        return true;
    }*/

    /**
     * åŠ é” ï¼ˆRLock）带超时时间的
     * @param key
     * @param expire
     * @param expireUnit
     * @return
     */
    public boolean acquire(String key, long expire, TimeUnit expireUnit) {
        //声明key对象
        key = LOCK_TITLE + key;
        //获取锁对象
        RLock mylock = redissonClient.getLock(key);
        //加锁，并且设置锁过期时间，防止死锁的产生
        try {
            mylock.tryLock(expire,expire,expireUnit);
        } catch (InterruptedException e) {
             e.getMessage();
             return false;
        }
        System.err.println("======lock======" + Thread.currentThread().getName());
        //加锁成功
        return true;
    }
    //锁的释放
    public void release(String lockName) {
        //必须是和加锁时的同一个key
        String key = LOCK_TITLE + lockName;
        //获取所对象
        RLock mylock = redissonClient.getLock(key);
        //释放锁（解锁）
        mylock.unlock();
        System.err.println("======unlock======" + Thread.currentThread().getName());
    }

}