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

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

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

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

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

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

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

代码语言：javascript复制

setnx key val
expire key seconds

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

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

代码语言：javascript复制

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

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

代码语言：javascript复制

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

0 人点赞