基于Redis实现分布式锁(超详细)

本文将从基于Redis实现分布式锁的几种坑的解决一步一步的进入到使用lua脚本+Redis

首先:实现分布式锁 可以⽤ Redis、Zookeeper、Mysql数据库这⼏种 , 性能最好的是Redis且是最容易理解,
同时 设计分布式锁应该考虑的东⻄:
排他性
在分布式应⽤集群中,同⼀个⽅法在同⼀时间只能被⼀台机器上的⼀个线程执⾏
容错性
分布式锁⼀定能得到释放,⽐如客户端奔溃或者⽹络中断

满⾜可重⼊、⾼性能、⾼可⽤

还需要注意分布式锁的开销、锁粒度

根据以上几点,那一般来说,大家容易想到的写法就是
1.加锁 SETNX key value
setnx 的含义就是 SET if Not Exists,有两个参数setnx(key, value),该⽅法是原⼦性操作,如果 key 不存在,则设置当前 key 成功,返回 1;如果当前 key 已经存在,则设置当前 key 失败,返回 0。
2.解锁 del (key)
得到锁的线程执⾏完任务,需要释放锁,以便其他线程可以进⼊,调⽤ del(key)
3配置锁超时 expire (key,30s)
客户端奔溃或者⽹络中断,资源将会永远被锁住,即死锁,因此需要给key配置过期时间,以保证即使没有被显式释放,这把锁也要在⼀定时间后⾃动释放.
我们根据上述思想,我们以领取优惠卷服务为例,不难得出一个伪代码:

methodA(){
 String key = "coupon_66"
 if(setnx(key,1== 1{
 expire(key,30,TimeUnit.MILLISECONDS)
 try {
 //做对应的业务逻辑
 //查询⽤户是否已经领券
 //如果没有则扣减库存
 //新增领劵记录
 } finally {
 del(key)
 }
 }else{
 //睡眠100毫秒,然后⾃旋调⽤本⽅法
 methodA()
 }
}

但是如果这样写,是存在很大问题的,大家可以先自己想想都会存在哪些问题。

问题引出
首先:多个命令之间不是原⼦性操作,如setnx和expire之间,如果setnx成功,但是expire失败,且宕机了,则这个资源就是死锁。
解决
使⽤原⼦命令:设置和配置过期时间 setnx / setex
如: set key 1 ex 30 nx
例如java⾥⾯
redisTemplate.opsForValue().setIfAbsent(“seckill_1”,1,30,TimeUnit.MILLISECONDS)
其次
业务超时,存在其他线程勿删,key 30秒过期,假如线程A执⾏很慢超过30秒,则key就被释放了,其他线程B就得到了锁,这个时候线程A执⾏完成,⽽B还没执⾏完成,结果就是线程A删除了线程B加的锁
例如下图:
在这里插入图片描述这时,大家又想了,这么简单,可以直接在 del 释放锁之前做⼀个判断,验证当前的锁是不是⾃⼰加的锁, 那 value 存当前线程的标识或者uuid就行了啊,那这时伪码就变成了这样

String key = "coupon_66"
String value =
Thread.currentThread().getId()
if(setnx(key,value) == 1{
 expire(key,30,TimeUnit.MILLISECONDS)
 try {
 //做对应的业务逻辑
 } finally {
 //删除锁,判断是否是当前线程加的
 if(get(key).equals(value)){
 //还存在时间间隔
 del(key)
 }
 }
}else{
 
 //睡眠100毫秒,然后⾃旋调⽤本⽅法
}

那这时,问题就又又又来了,假设当线程A获取到正常值时,返回带代码中判断期间锁过期了,线程B刚好重新设置了新值,线程A那边有判断value是⾃⼰的标识,然后调⽤del⽅法,结果就是删除了新设置的线程B的值

不难发现上面的问题核⼼都是判断和删除命令 不是原⼦性操作导致

那我们如何解决呢?

这时就需要分布式锁lua脚本+redis

采⽤ lua脚本+redis, 由于判断和删除是lua脚本执⾏,所以要么全成功,要么全失败,这就保证了原子性

//获取lock的值和传递的值⼀样,调⽤删除操作返回1,否
则返回0
其中lua脚本如下
String script = “if redis.call(‘get’,KEYS[1])== ARGV[1] then return
redis.call(‘del’,KEYS[1]) else return 0 end”;
//Arrays.asList(lockKey)是key列表,uuid是参数
Integer result = redisTemplate.execute(new DefaultRedisScript<>(script, Integer.class),Arrays.asList(lockKey), uuid);

现在完整伪代码如下

String uuid = CommonUtil.generateUUID();
String lockKey = "lock:coupon:"+couponId;
Boolean nativeLock=redisTemplate.opsForValue().setIfAbsent(lockKey,uuid,Duration.ofSeconds(30));
 if(nativeLock){
 //加锁成功
 try {
 //执⾏业务 TODO
 }finally {
 String script = "if
redis.call('get',KEYS[1]) == ARGV[1] then return redis.call('del',KEYS[1]) else return 0 end";
 Integer result =redisTemplate.execute(newDefaultRedisScript<>(script, Integer.class),Arrays.asList(lockKey), uuid);
 }
 }else {
 //加锁失败,睡眠100毫秒,⾃旋重试
 try {
TimeUnit.MILLISECONDS.sleep(100L);
 } catch (InterruptedException e)
{ }
 return addCoupon( couponId,couponCategory);
 }

完成。

  • 6
    点赞
  • 21
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论
基于 Redis 实现分布式锁可以利用 Redis 的原子性操作和时特性来实现。下面是一个基本的实现思路: 1. 获取锁:使用 Redis 的 SETNX 命令,如果指定的锁 key 不存在,则设置该 key 的值为当前时间戳加上锁的时时间,并返回成功;否则,返回失败。 2. 释放锁:使用 Redis 的 EVAL 命令,通过 Lua 脚本来实现原子性的删除锁。脚本的内容是先判断锁是否存在且时,如果是则删除锁并返回成功;否则,返回失败。 下面是一个简单的 Python 代码示例: ```python import redis import time class RedisLock: def __init__(self, redis_client, lock_key, expire_time): self.redis = redis_client self.lock_key = lock_key self.expire_time = expire_time def acquire(self): while True: timestamp = int(time.time() * 1000) + self.expire_time acquired = self.redis.set(self.lock_key, timestamp, nx=True, px=self.expire_time) if acquired: return True time.sleep(0.001) def release(self): lua_script = """ if redis.call("exists", KEYS[1]) == 1 then local current_value = tonumber(redis.call("get", KEYS[1])) if current_value and current_value <= tonumber(ARGV[1]) then return redis.call("del", KEYS[1]) end end return 0 """ self.redis.eval(lua_script, 1, self.lock_key, int(time.time() * 1000) + self.expire_time) # 使用示例 redis_client = redis.Redis(host='localhost', port=6379, db=0) lock = RedisLock(redis_client, 'my_lock', 1000) # 锁的时时间为 1000 毫秒 if lock.acquire(): try: # 执行需要加锁的代码 pass finally: lock.release() ``` 需要注意的是,以上代码仅是一个简单的实现示例,实际使用中还需要考虑异常处理、锁的可重入性、锁的可拥有时间等问题。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

工藤学编程

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值