分布式锁
本文整理自黑马程序员相关资料
问题的引入
在平时单服务的情况下,我们使用互斥锁可以保证同一时刻只有一个线程执行自己的业务。原理是,在JVM内部维护了一个锁监视器,锁监视器保证了同一时刻只有一个线程获取到锁。但是如果开启了多个服务,就会有多个JVM,从而有多个不同的锁监视器,每个锁监视器监视自己JVM内部的线程,因此一个JVM内部的线程获取到锁,并不影响其他JVM内部的线程获取锁。从而导致并发安全问题。因此,我们需要独立于JVM之外的锁监视器对所有的线程统一管理。
概念
满足分布式系统或集群模式下多进程可见并且互斥的锁。
常见分布式锁的实现比较
MySQL | Redis | Zookeeper | |
---|---|---|---|
互斥 | 利用Mysql本身的互斥锁机制 | 利用setnx这样的互斥命令 | 利用节点的唯一性和有序性实现互斥 |
高可用 | 好 | 好 | 好 |
高性能 | 一般 | 好 | 一般 |
安全性 | 断开连接,自动释放锁 | 利用锁超时时间,到期释放 | 临时节点,断开连接自动释放 |
基于Redis的分布式锁
最基本的分布式锁
获取锁:
利用Redis的SETNX保证互斥的特性,同时设置锁过期时间,避免服务宕机不能执行释放锁的操作而导致死锁。
释放锁:
删除对应的键即可
流程图如下所示:
保证释放锁的线程是持有锁的线程本身
前面提到的最基本的分布式锁存在着一些问题。如果获取锁的线程1阻塞,在该线程阻塞期间,锁超时释放了,这时线程2就可以获取到锁,接着执行自己的业务。线程1在完成自己的业务后释放锁。这时线程3也获得了锁执行自己的业务,这样就造成了线程2和线程3都获取到了锁,从而造成了线程安全问题。如下图所示
为了解决未持有锁的线程释放锁这个问题,在锁中存入线程标识,在释放锁之前先判断锁标识是否是本身线程。如果标识是自己,则释放锁。其流程图如下所示
保证释放锁的原子性
由于前面加入了判断,判断与释放是两步。有可能在判断时持有锁的线程1阻塞,直到超时释放锁,线程2拿到了锁,线程1被唤醒并执行释放锁,导致线程3也拿到了锁。造成了两个线程同时持有锁的线程安全问题。如下所示
为了解决这个问题,使用Lua脚本,在一个脚本中编写多条Redis命令,确保多条命令执行时的原子性。
释放锁的业务流程如下所示
-- 这里的 KEYS[1] 就是锁的key,这里的ARGV[1] 就是当前线程标示 -- 获取锁中的标示,判断是否与当前线程标示一致 if (redis.call('GET', KEYS[1]) == ARGV[1]) then -- 一致,则删除锁 return redis.call('DEL', KEYS[1]) end -- 不一致,则直接返回 return 0
到目前为止,一个基于Redis的基本的分布式锁就完成了。但还是存在着以下问题
- 不可重入:同一线城无法多次获取统一把锁
- 不可重试:获取锁只尝试一次就返回,没有重试机制
- 超时释放问题:锁超时释放虽然可以避免死锁,但是如果业务执行耗时较长,也会导致锁释放,存在安全隐患
- 主从一致性问题:如果Redis提供了主从集群,主从同步存在延迟,当主节点宕机时,从节点没有同步主节点中的锁数据。其他线程就会拿到锁
Redisson分布式锁简单介绍
Redisson可重入锁原理
获取锁的Lua脚本
local key = KEYS[1]; -- 锁的key local threadId = ARGV[1]; -- 线程唯一标识 local releaseTime = ARGV[2]; -- 锁的自动释放时间 -- 判断是否存在 if(redis.call('exists', key) == 0) then -- 不存在, 获取锁 redis.call('hset', key, threadId, '1'); -- 设置有效期 redis.call('expire', key, releaseTime); return 1; -- 返回结果 end; -- 锁已经存在,判断threadId是否是自己 if(redis.call('hexists', key, threadId) == 1) then -- 不存在, 获取锁,重入次数+1 redis.call('hincrby', key, threadId, '1'); -- 设置有效期 redis.call('expire', key, releaseTime); return 1; -- 返回结果 end; return 0; -- 代码走到这里,说明获取锁的不是自己,获取锁失败
释放锁的Lua脚本
local key = KEYS[1]; -- 锁的key local threadId = ARGV[1]; -- 线程唯一标识 local releaseTime = ARGV[2]; -- 锁的自动释放时间 -- 判断当前锁是否还是被自己持有 if (redis.call('HEXISTS', key, threadId) == 0) then return nil; -- 如果已经不是自己,则直接返回 end; -- 是自己的锁,则重入次数-1 local count = redis.call('HINCRBY', key, threadId, -1); -- 判断是否重入次数是否已经为0 if (count > 0) then -- 大于0说明不能释放锁,重置有效期然后返回 redis.call('EXPIRE', key, releaseTime); return nil; else -- 等于0说明可以释放锁,直接删除 redis.call('DEL', key); return nil; end;
Redisson分布式锁原理
- 可重入:利用hash结构记录线程id和重入次数
- 可重试:利用信号量和PubSub功能实现等待、唤醒,获取锁失败的重试机制
- 超时续约:利用watchDog,每隔一段时间(releaseTime/3),重置超时时间。