From ad6386a618eb81c401333c6541ec798932991bf6 Mon Sep 17 00:00:00 2001 From: sxq <812980466@qq.com> Date: Fri, 4 Jun 2021 14:46:35 +0800 Subject: [PATCH] =?UTF-8?q?redis=E5=88=86=E5=B8=83=E5=BC=8F=E9=94=81?= =?UTF-8?q?=EF=BC=88=E5=9F=BA=E4=BA=8E=E6=B3=A8=E8=A7=A3=E5=BD=A2=E5=8A=BF?= =?UTF-8?q?=EF=BC=89=E3=80=82?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../system/RedisLockController.java | 44 +++++++ .../ruoyi/common/annotation/RedisLock.java | 27 ++++ .../common/core/redis/RedisLockAspect.java | 124 ++++++++++++++++++ .../common/core/redis/RedisLockUtil.java | 64 +++++++++ 4 files changed, 259 insertions(+) create mode 100644 ruoyi-admin/src/main/java/com/ruoyi/web/controller/system/RedisLockController.java create mode 100644 ruoyi-common/src/main/java/com/ruoyi/common/annotation/RedisLock.java create mode 100644 ruoyi-common/src/main/java/com/ruoyi/common/core/redis/RedisLockAspect.java create mode 100644 ruoyi-common/src/main/java/com/ruoyi/common/core/redis/RedisLockUtil.java diff --git a/ruoyi-admin/src/main/java/com/ruoyi/web/controller/system/RedisLockController.java b/ruoyi-admin/src/main/java/com/ruoyi/web/controller/system/RedisLockController.java new file mode 100644 index 00000000..28102d07 --- /dev/null +++ b/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); + } +} diff --git a/ruoyi-common/src/main/java/com/ruoyi/common/annotation/RedisLock.java b/ruoyi-common/src/main/java/com/ruoyi/common/annotation/RedisLock.java new file mode 100644 index 00000000..e84f5c83 --- /dev/null +++ b/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"; +} diff --git a/ruoyi-common/src/main/java/com/ruoyi/common/core/redis/RedisLockAspect.java b/ruoyi-common/src/main/java/com/ruoyi/common/core/redis/RedisLockAspect.java new file mode 100644 index 00000000..c249dfe9 --- /dev/null +++ b/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 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 getKeyParsList(String key) { + List 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; + } + +} diff --git a/ruoyi-common/src/main/java/com/ruoyi/common/core/redis/RedisLockUtil.java b/ruoyi-common/src/main/java/com/ruoyi/common/core/redis/RedisLockUtil.java new file mode 100644 index 00000000..6dc6591c --- /dev/null +++ b/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()); + } + +}