使用自定义注解实现redisson分布式锁
作者：SillinessPlus
这篇文章主要介绍了使用自定义注解实现redisson分布式锁，具有很好的参考价值，具有很好的参考价值，希望对大家有所帮助。如有错误或未考虑完全的地方，望不吝赐教
自定义注解实现redisson分布式锁
自定义注解
package com.example.demo.annotation;
import java.lang.annotation.*;
/**
* desc: 自定义 redisson 分布式锁注解
*
* @author: 邢阳
* @mail: xydeveloper@126.com
* @create 2021-05-28 16:50
*/
@Target({ElementType.METHOD})
@Retention(RetentionPolicy.RUNTIME)
@Documented
@Inherited
public @interface Lock {
/**
* 锁的key spel 表达式
*
* @return
*/
String key();
/**
* 持锁时间
*
* @return
*/
long keepMills() default 20;
/**
* 没有获取到锁时，等待时间
*
* @return
*/
long maxSleepMills() default 30;
}
aop解析注解
package com.example.demo.utils;
import com.example.demo.annotation.Lock;
import org.aspectj.lang.ProceedingJoinPoint;
import org.aspectj.lang.annotation.Around;
import org.aspectj.lang.annotation.Aspect;
import org.aspectj.lang.reflect.MethodSignature;
import org.redisson.api.RLock;
import org.redisson.api.RedissonClient;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.core.DefaultParameterNameDiscoverer;
import org.springframework.expression.EvaluationContext;
import org.springframework.expression.spel.standard.SpelExpressionParser;
import org.springframework.expression.spel.support.StandardEvaluationContext;
import org.springframework.stereotype.Component;
import java.util.Objects;
import java.util.concurrent.TimeUnit;
/**
* desc: 解析 自定义 redisson 分布式锁注解
*
* @author: 邢阳
* @mail: xydeveloper@126.com
* @create 2021-05-28 16:50
*/
@Aspect
@Component
public class LockAspect {
@Autowired
private RedissonClient redissonClient;
/**
* 用于SpEL表达式解析.
*/
private final SpelExpressionParser spelExpressionParser = new SpelExpressionParser();
/**
* 用于获取方法参数定义名字.
*/
private final DefaultParameterNameDiscoverer defaultParameterNameDiscoverer = new DefaultParameterNameDiscoverer();
@Around("@annotation(com.example.demo.annotation.Lock)")
public Object around(ProceedingJoinPoint proceedingJoinPoint) throws Throwable {
Object object = null;
RLock lock = null;
try {
// 获取注解实体信息
Lock lockEntity = (((MethodSignature) proceedingJoinPoint.getSignature()).getMethod())
.getAnnotation(Lock.class);
// 根据名字获取锁实例
lock = redissonClient.getLock(getKeyBySpeL(lockEntity.key(), proceedingJoinPoint));
if (Objects.nonNull(lock)) {
if (lock.tryLock(lockEntity.maxSleepMills(), lockEntity.keepMills(), TimeUnit.SECONDS)) {
object = proceedingJoinPoint.proceed();
} else {
throw new RuntimeException();
}
}
} finally {
if (Objects.nonNull(lock) && lock.isHeldByCurrentThread()) {
lock.unlock();
}
}
return object;
}
/**
* 获取缓存的key
*
* key 定义在注解上，支持SPEL表达式
*
* @return
*/
public String getKeyBySpeL(String spel, ProceedingJoinPoint proceedingJoinPoint) {
MethodSignature methodSignature = (MethodSignature) proceedingJoinPoint.getSignature();
String[] paramNames = defaultParameterNameDiscoverer.getParameterNames(methodSignature.getMethod());
EvaluationContext context = new StandardEvaluationContext();
Object[] args = proceedingJoinPoint.getArgs();
for (int i = 0; i < args.length; i++) {
context.setVariable(paramNames[i], args[i]);
}
return String.valueOf(spelExpressionParser.parseExpression(spel).getValue(context));
}
}
service中使用注解加锁使用
/**
* desc: 锁
*
* @author: 邢阳
* @mail: xydeveloper@126.com
* @create 2021-05-28 17:58
*/
@Service
public class LockService {
@Lock(key = "#user.id", keepMills = 10, maxSleepMills = 15)
public String lock(User user) {
System.out.println("持锁");
return "";
}
}
redisson分布式锁应用
分布式架构一定会用到分布式锁。目前公司使用的基于redis的redisson分布式锁。
应用场景
1.订单修改操作，首先要获取该订单的分布式锁，能取到才能去操作。lockey可以是订单的主键id。
2.库存操作，也要按照客户+仓库+sku维护锁定该库存，进行操作。
代码：
Redisson管理类
public class RedissonManager {
private static RedissonClient redisson;
static {
Config config = new Config();
config.useSentinelServers()
.addSentinelAddress("redis://127.0.0.1:26379","redis://127.0.0.1:7301", "redis://127.0.0.1:7302")
.setMasterName("mymaster")
.setReadMode(ReadMode.SLAVE)
.setTimeout(10000).setDatabase(0).setPassword("123***");
redisson = Redisson.create(config);
}
/**
* 获取Redisson的实例对象
* @return
*/
public static RedissonClient getRedisson(){ return redisson;}
}分布式锁
import org.redisson.api.RLock;
import org.redisson.api.RedissonClient;
import java.util.concurrent.TimeUnit;
public class DistributedLock {
private static RedissonClient redissonClient = RedissonManager.getRedisson();
public static boolean tryLock(String lockKey, TimeUnit unit, int waitTime, int leaseTime) {
RLock lock = redissonClient.getLock(lockKey);
try {
return lock.tryLock(waitTime, leaseTime, unit);
} catch (InterruptedException e) {
return false;
}
}
public static void unlock(String lockKey) {
RLock lock = redissonClient.getLock(lockKey);
lock.unlock();
}
}测试类
public class RedissonTest {
public static void main(String[] args) throws Exception{
Thread.sleep(2000L);
for (int i = 0; i < 3; i++) {
new Thread(() -> {
try {
//tryLock，第三个参数是等待时间，5秒内获取不到锁，则直接返回。 第四个参数 30是30秒后强制释放
boolean hasLock = DistributedLock.tryLock("lockKey", TimeUnit.SECONDS,5,30);
//获得分布式锁
if(hasLock){
System.out.println("idea1: " + Thread.currentThread().getName() + "获得了锁");
/**
* 由于在DistributedLock.tryLock设置的等待时间是5s，
* 所以这里如果休眠的小于5秒，这第二个线程能获取到锁，
* 如果设置的大于5秒，则剩下的线程都不能获取锁。可以分别试试2s,和8s的情况
*/
Thread.sleep(10000L);
DistributedLock.unlock("lockKey");
} else {
System.out.println("idea1: " + Thread.currentThread().getName() + "无法获取锁");
}
} catch (Exception e) {
e.printStackTrace();
}
}) .start();
}
}
}我们再打开一个idea，可以把代码复制一份。同事启动两个RedissonTest ，模拟了并发操作。
测试结果：
idea2: Thread-1获得了锁
idea2: Thread-0无法获取锁
idea2: Thread-2无法获取锁
idea1: Thread-2无法获取锁
idea1: Thread-0无法获取锁
idea1: Thread-1无法获取锁
从测试结果发现，最后是只能有一个idea的一个线程能获取到锁。
以上为个人经验，希望能给大家一个参考，也希望大家多多支持脚本之家。