利用 redis 实现分布式锁
分四个阶段,徐徐渐进,后续阶段将解决前序阶段出现的问题
第一阶段:
@GetMapping("testLock")
public void testLock(){
//1获取锁,setne
Boolean lock = redisTemplate.opsForValue().setIfAbsent("lock", "111");
//2获取锁成功、查询num的值
if(lock){
Object value = redisTemplate.opsForValue().get("num");
//2.1判断num为空return
if(StringUtils.isEmpty(value)){
return;
}
//2.2有值就转成成int
int num = Integer.parseInt(value+"");
//2.3把redis的num加1
redisTemplate.opsForValue().set("num", ++num);
//2.4释放锁,del
redisTemplate.delete("lock");
}else{
//3获取锁失败、每隔0.1秒再获取
try {
Thread.sleep(100);
testLock();
} catch (InterruptedException e) {
e.printStackTrace();
}
}
}
存在问题:
setnx 某个业务服务程序占好了锁,但在执行业务过程中,程序挂掉。没有执行解锁操作,造成后续服务程序无法加锁。
解决方案:
为锁加上自动过期时间
第二阶段:
加锁的过程中定义过期时间
存在问题:
某个业务服务程序占好了锁并设置了过期时间,但执行业务的时间过长,锁已经自动解除了。该业务在执行完后依然去解锁,则可能会解除掉其他业务所加的锁。
解决方案:
给锁加上 UUID ,这样即使执行业务时间过长,在其结束后,就无法解除其他业务所加的锁。
第三阶段:
给锁的设置 UUID
存在问题:
在业务将要解除锁时,先获取锁,与自身的 UUID 进行比较,再将解除锁的命令发到 redis 上去执行。但在比较和发送的过程中,自身 UUID 所对应的锁已过期,且别人也已经新加上锁,解锁命令到达执行,则会将其他人的锁解除。
解决方案:
使用 LUA 脚本,将 UUID 比较 和 解除锁 的步骤 原子化 。
第四阶段:
使用 LUA 脚本,将 UUID 比较 和 解除锁 的步骤 原子化
@GetMapping("testLockLua")
public void testLockLua() {
//1 声明一个uuid ,将做为一个value 放入我们的key所对应的值中
String uuid = UUID.randomUUID().toString();
//2 定义一个锁:lua 脚本可以使用同一把锁,来实现删除!
String skuId = "25"; // 访问skuId 为25号的商品 100008348542
String locKey = "lock:" + skuId; // 锁住的是每个商品的数据
// 3 获取锁
Boolean lock = redisTemplate.opsForValue().setIfAbsent(locKey, uuid, 3, TimeUnit.SECONDS);
// 如果true
if (lock) {
// 执行的业务逻辑开始
// 获取缓存中的num 数据
Object value = redisTemplate.opsForValue().get("num");
// 如果是空直接返回
if (StringUtils.isEmpty(value)) {
return;
}
// 不是空 如果说在这出现了异常! 那么delete 就删除失败! 也就是说锁永远存在!
int num = Integer.parseInt(value + "");
// 使num 每次+1 放入缓存
redisTemplate.opsForValue().set("num", String.valueOf(++num));
/*使用lua脚本来锁*/
// 定义lua 脚本
String script = "if redis.call('get', KEYS[1]) == ARGV[1] then return redis.call('del', KEYS[1]) else return 0 end";
// 使用redis执行lua执行
DefaultRedisScript<Long> redisScript = new DefaultRedisScript<>();
redisScript.setScriptText(script);
// 设置一下返回值类型 为Long
// 因为删除判断的时候,返回的0,给其封装为数据类型。如果不封装那么默认返回String 类型,
// 那么返回字符串与0 会有发生错误。
redisScript.setResultType(Long.class);
// 第一个要是script 脚本 ,第二个需要判断的key,第三个就是key所对应的值。
redisTemplate.execute(redisScript, Arrays.asList(locKey), uuid);
} else {
// 其他线程等待
try {
// 睡眠
Thread.sleep(1000);
// 睡醒了之后,调用方法。
testLockLua();
} catch (InterruptedException e) {
e.printStackTrace();
}
}
}