mirror of
https://github.com/dromara/RuoYi-Vue-Plus.git
synced 2025-11-04 08:13:44 +08:00
redis分布式锁（基于注解形势）。
This commit is contained in:
sxq
疯狂的狮子li
@@ -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);
|
||||
}
|
||||
}
|
||||
@@ -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";
|
||||
}
|
||||
@@ -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;
|
||||
}
|
||||
|
||||
}
|
||||
@@ -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());
|
||||
}
|
||||
|
||||
}
|
||||
Reference in New Issue
Block a user