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());
+	}
+
+}