一文教你如何用Redis构建高性能锁

前言

在这里粗略的说一下,zk锁性能比redis低的原因:zk中的角色分为leader,flower,每次写请求只能请求leader,leader会把写请求广播到所有flower,如果flower都成功才会提交给leader,其实这里相当于一个2PC的过程。在加锁的时候是一个写请求,当写请求很多时,zk会有很大的压力,最后导致服务器响应很慢。

正题:

什么情况下需要加锁?

当多个线程、用户同时竞争同一个资源时,需要加锁。比如,下订单减库存,抢票,选课,抢红包等。如果在此处没有锁的控制,会导致很严重的问题,下订单减库存的时候不加锁,会导致商品超卖;抢票的时候不加锁,会导致两个人抢到同一个位置;选课的时候没有锁的控制,导致选课成功的人数大于教室的座位数;抢红包时没有锁的控制,抢到红包的金额大于红包的实际金额。

什么是分布式锁?

学过JAVA多线程的朋友都知道,为了防止多个线程同时执行同一段代码,可以用synchronized关键字或JAVA API中ReentrantLock类来控制。

但是目前几乎任何一个系统都往往部署多台机器的,单机部署的应用很少,synchronized和ReentrantLock发挥不出任何作用,此时就需要一把全局的锁,来代替JAVA中的synchronized和ReentrantLock。

当Thread1线程获取到锁,执行锁中的代码,其他线程或其他机器再次请求该锁,发现锁被Thread1占用,加锁失败。当Thread1释放锁,其他线程则可以获取到锁并执行相应的操作。

我们可以用Jedis中是setnx命令来构建这把锁,首先,我列举一些错误的构建锁的方式:

错误例子1

Long lock= jedis.setnx(key,value); 
if(lock>0){ 
//执行业务逻辑 
} 

通过setnx命令创建一个key、value,如果key不存在,则加锁成功。这样做有什么问题呢?如果执行加锁操作成功,在释放锁的时候,系统宕机,导致这个key永远不会被del掉,也就是说其他线程一直获取不到锁,

导致死锁发生。为了避免这种情况,请看下面的代码

错误例子2

Long lock= jedis.setnx(key,value); 
if(lock>0){ 
 jedis.expire(key,expireTime); 
} 

和上面的例子类似,唯一不同的是这里多了一步设置key过期时间的操作。如果在del的时候系统宕机,等过期时间一到,Redis会删除这个key。

其他线程可以再次获取锁。这样就可以万无一失了吗?这里有一个问题,如果在第一步setnx成功后,突然网络闪断,expire命令执行失败,同样也有死锁的风险。这两步并不具备原子性,不保证全部成功或全部失败。

正确的构建方式

public static boolean getDistributedLock(Jedis jedis, String lockKey, String requestId, int expireTime) { 
 String result = jedis.set(lockKey, requestId, "NX", "EX", expireTime); 
 if (LOCK_SUCCESS.equals(result)) { 
return true; 
 } 
return false; 
} 

参数解释:

key:键

value:值

nx:如果当前key存在,则set失败,否则成功

ex:设置key的过期时间

expireTime:key的过期时间,时间到了,Redis会自动删除key和value。

这个命令,将上面的错误例子2中的两个操作合为一个原子操作,保证了同时成功或同时失败。

解锁方式:

错误例子1:

jedis.del(key); 

执行这个操作的线程,不去判断锁的拥有者就删除锁。

还记的set命令可以设置value吗?在获取锁的操作时,主要是判断key是否存在,那么value有什么用呢??如果在删除锁的时候,不去判断当前锁的拥有者,任何线程都可以释放锁。这个时候,value值就起到作用了。

错误例子2:

if(value==jedis.get(key)){ 
 jedis.del(key); 
} 

我们在加锁的时候,可以将value设置成唯一标识当前线程的一个值,这个值可以是一个UUID,当释放锁的时间,判断value是否和set时的值相同,如果相同,则说明加锁和释放锁是同一个线程,允许释放。否则释放锁失败。

这样就可以绝对安全了吗??答案当然是否定的。这步操作,同样不具备原子性。如果ThreadA在执行value==jedis.get(key)返回true后的瞬间,del命令还没来的及执行,key过期了,而此时ThreadB获取到锁,之后ThreadA执行del命令,把ThreadB的锁释放掉了。

所以要保证两部操作的原子性,我们不得不利用简单的Lua脚本。

学习的过程当中有遇见学习,行业方面的问题,或者说缺乏系统的学习路线和学习视频可以点击https://shimo.im/docs/VqQR6tPrpR3C3tjq/

正确的解锁姿势:

public static boolean releaseDistributedLock(Jedis jedis, String lockKey, String requestId) { 
String script = "if redis.call('get', KEYS[1]) == ARGV[1] then return redis.call('del', KEYS[1]) else return 0 end"; 
 Object result = jedis.eval(script, Collections.singletonList(lockKey), Collections.singletonList(requestId)); 
 if (RELEASE_SUCCESS.equals(result)) { 
 return true; 
 } 
 return false; 
} 

Redis在2.6后内部内嵌Lua脚本解释器,所以我们可以通过简单的Lua脚本来保证上述操作的原子性。代码中的Lua脚本的的意思是:我们把LockKey赋值给KEYS[1],把RequestId赋值给ARGV[1],如果key中的值等于RequestId,返回true否则返回false。这样就保证了释放锁操作时原子的,并且当前客户端只会释放当前客户端的锁。

©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 216,544评论 6 501
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 92,430评论 3 392
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 162,764评论 0 353
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 58,193评论 1 292
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 67,216评论 6 388
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 51,182评论 1 299
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 40,063评论 3 418
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 38,917评论 0 274
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 45,329评论 1 310
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 37,543评论 2 332
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 39,722评论 1 348
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 35,425评论 5 343
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 41,019评论 3 326
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 31,671评论 0 22
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 32,825评论 1 269
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 47,729评论 2 368
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 44,614评论 2 353

推荐阅读更多精彩内容