Redis分布式锁解决多进程/多线程下单个进程/单个线程运行

1.导语

在业务开发中像订单写入,一般需要单线程来保证订单写入数据库,防止数据多次被插入。

最近,有两台容器,当程序运行时,会发送多份通知,那么需要保证同一时刻只有一个进程(一台容器)来运行,此时用分布式锁解决该问题。

业界也有许多解决这种方案,这里以redis分布式锁来解决。

简单来说就是采用golang redis库实现下面方案即可。

2.redis的分布式锁实现

2.1 setnx+expire

setnx key value,将key设置为value,当键不存在时,才能成功,若键存在,什么也不做,成功返回1,失败返回0。

SETNX实际上就是SET IF NOT Exists的缩写。

 

setnx key val expire key seconds 

但是, 大型对撞机上述两个操作不具有原子性,如果执行完第一条指令应用异常或者重启了,锁将无法过期。

2.2 lua脚本

既然是原子性无法保证,那就采用执行lua脚本的原子性,将上述两个操作封装到lua脚本中便可以实现。

 

if redis.call('setnx',KEYS[1],ARGV[1]) == 1 then    redis.call('expire',KEYS[1],产品中心ARGV[2]) else    return 0 end; 

2.3 携带TTL的set

从 Redis 2.6.12 版本开始, SET 命令的行为可以通过一系列参数来修改。

 

SET key value [EX seconds]  [PX milliseconds] [NX|XX] 

将字符串值 value 关联到 key 。

如果 key 已经持有其他值, SET 就覆写旧值,无视类型。

对于某个原本带有生存时间(TTL)的键来说, 当 SET 命令成功在这个键上执行时, 这个键原有的 TTL 将被清除。

EX second :设置键的过期时间为 second 秒。SET key value EX second 效果等同于 SETEX key second value 。 PX millisecond :设置键的过期时间为 millisecond 毫秒。SET key value PX millisecond 效果等同于 PSETEX key millisecond value 。 NX :只在键不存在时,才对键进行设置操作。SET key value NX 效果等同于 SETNX key value 。 XX :只在键已经存在时,才对键进行设置操作。

直接使用可能存在如下问题:

超时解锁导致并发

例如:如果线程 A 成功获取锁并设置过期时间 30 秒,但线程 A 执行时间超过了 30 秒,锁过期自动释放,此时线程 B 获取到了锁,线程 A 和线程 B 并发执行。

A、B 两个线程发生并发显然是不被允许的,一般有两种方式解决该问题:

解决方案:1)确保代码在过期时间之前释放。2)为获取锁的线程增加守护线程,为将要过期但未释放的锁增加有效时间。

锁被别的线程误删除。

例如:如果线程 A 成功获取到了锁,并且设置了过期时间 30 秒,但线程 A 执行时间超过了 30 秒,锁过期自动释放,此时线程 B 获取到了锁;随后 A 执行完成,线程 A 使用 DEL 命令来释放锁,但此时线程 B 加的锁还没有执行完成,线程 A 实际释放的线程 B 加的锁。

解决方案是:通过在 value 中设置当前线程加锁的标识,在删除之前验证 key 对应的 value 判断锁是否是当前线程持有。可生成一个 UUID 标识当前线程,使用 lua 脚本做验证标识和解锁操作。

学习文章:

https://xiaomi-info.github.io/2019/12/17/redis-distributed-lock/

https://zhuanlan.zhihu.com/p/115848078