背景
使用@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来判断锁失效了没。
版权声明:本文为waltonhuang原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。