shedlock分布式锁原理(JDBC)

/ 转载 / 没有评论 / 491浏览

背景

使用@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来判断锁失效了没。