SchedulerLock 分布式锁 原理

背景

使用@SchedulerLock可以实现比较轻量级的简单的定时任务,也可以实现分布式锁,那具体的原理是啥?

一探究竟

不熟悉的同学可以先看一下用法:
https://www.baeldung.com/shedlock-spring

大概就是定义一个数据表
在这里插入图片描述
注意这里的name是primary key,是不可重复的。利用了primary key的这个特性,实现了锁的抢占。细节下面会介绍

然后加注解实现:
在这里插入图片描述

看源码

直接从注解出发,看源码
在注解所在的包内发现一个DefaultLockManager

一看有executeWithTask,应该就是这个代码了
在这里插入图片描述

一直往里看
在这里插入图片描述

doLock
在这里插入图片描述

在这里插入图片描述

insertRecord,看jdbc的实现
在这里插入图片描述

获取锁逻辑

其实很简单,就是一个sql语句

INSERT INTO tableName (name, lock_until, locked_at, locked_by) VALUES(锁名字, 当前时间+最多锁多久, 当前时间, 主机名)

如果插入成功了,就说明抢到了锁。否则认为抢不到。

如果记录已经存在了,就用更新记录来抢锁
在这里插入图片描述

UPDATE  tableName  SET lock_until = 当前时间+最多锁多久, locked_at = 当前时间, locked_by = 主机名 WHERE name = 锁名字 AND lock_until <= 当前时间

找到同一个name的锁,如果发现lock_util小于等于当前时间(锁已经无效),就可以去update;多个机器同时update,只有一个机器可以update成功(修改行数>0),实现了锁的抢占。

释放锁逻辑

在这里插入图片描述
注:lockTime = 最小锁时间 和 当前时间的较大值

UPDATE tableName SET lock_until = lockTime WHERE name = 锁名字

假设最小锁时间30s,当前时间只过了15s,那么就会最小锁时间;否则用当前时间。
这个就是为了保证最少也要锁 最小锁时间 (30s)

总结

SchedulerLock利用mysql实现了分布式锁。
抢锁:
通过插入同一个name(primary key),或者更新同一个name来抢
释放锁:
通过设置lock_until来实现释放,再次抢锁的时候需要通过lock_util来判断锁失效了没。

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章