怎么在MySQL数据库中实现一个分布式锁

49次阅读
没有评论

共计 4070 个字符,预计需要花费 11 分钟才能阅读完成。

自动写代码机器人,免费开通

这篇文章将为大家详细讲解有关怎么在 MySQL 数据库中实现一个分布式锁,文章内容质量较高,因此丸趣 TV 小编分享给大家做个参考,希望大家阅读完这篇文章后对相关知识有一定的了解。

建表

CREATE TABLE `globallocktable` ( `id` int(11) NOT NULL AUTO_INCREMENT,
 `lockKey` varchar(60) NOT NULL COMMENT  锁名称 ,
 `createTime` datetime NOT NULL COMMENT  创建时间 ,
 PRIMARY KEY (`id`),
 UNIQUE KEY `lockKey` (`lockKey`)
) ENGINE=InnoDB DEFAULT CHARSET=utf8 ROW_FORMAT=DYNAMIC COMMENT= 全局锁 

让别人使用的组件

@Component
public class GlobalLockComponent {
 @Resource
 GlobalLockTableDAO globalLockDAO;
 /**
 *  尝试获得锁,成功为 true,失败为 false
 */
 public boolean tryLock(String key) { return GlobalLockUtil.tryLock(this.globalLockDAO, key);
 }
 /**
 *  如果已经有其他程序占用该锁,并且超过 timeoutMs(毫秒)时间,就强制清除这个锁占用
 *  即根据 key 先删除记录,再添加记录
 */
 public boolean tryLockWithClear(String key, Long timeoutMs) { return GlobalLockUtil.tryLockWithClear(this.globalLockDAO, key, timeoutMs);
 }
 /**
 *  释放锁,根据 key 删除记录
 */
 public void releasLock(String key) { GlobalLockUtil.releasLock(this.globalLockDAO, key);
 }
}

锁对象定义如下

public class GlobalLockTable {
 private Integer id;
 private String lockKey;
 private Date createTime;
 //  省略 get 和 set 方法
GlobalLockTableDAO 定义如下
public interface GlobalLockTableDAO { int deleteByPrimaryKey(Integer id);
 int deleteByLockKey(String lockKey);
 GlobalLockTable selectByLockKey(String key);
 int insertSelectiveWithTest(GlobalLockTable record);
}

具体加锁和解锁逻辑

public class GlobalLockUtil { private static Logger logger = LoggerFactory.getLogger(GlobalLockUtil.class);
 private static GlobalLockTable tryLockInternal(GlobalLockTableDAO lockDAO, String key) { GlobalLockTable insert = new GlobalLockTable();
 insert.setCreateTime(new Date());
 insert.setLockKey(key);
 //  注意的地方 1
 int count = lockDAO.insertSelectiveWithTest(insert);
 if (count == 0) { GlobalLockTable ready = lockDAO.selectByLockKey(key);
 logger.warn(can not lock the key: {}, {}, {} , insert.getLockKey(), ready.getCreateTime(),
 ready.getId());
 return ready;
 }
 logger.info(yes got the lock by key: {} , insert.getId(), insert.getLockKey());
 return null;
 }
 /**  超时清除锁占用,并重新加锁  **/
 public static boolean tryLockWithClear(GlobalLockTableDAO lockDAO, String key, Long timeoutMs) { GlobalLockTable lock = tryLockInternal(lockDAO, key);
 if (lock == null) return true;
 if (System.currentTimeMillis() - lock.getCreateTime().getTime()  = timeoutMs) { logger.warn( sorry, can not get the key. : {}, {}, {} , key, lock.getId(), lock.getCreateTime());
 return false;
 }
 logger.warn(the key already timeout wthin : {}, {}, will clear , key, timeoutMs);
 //  注意的地方 2
 int count = lockDAO.deleteByPrimaryKey(lock.getId());
 if (count == 0) { logger.warn( sorry, the key already preemptived by others: {}, {} , lock.getId(), lock.getLockKey());
 return false;
 }
 lock = tryLockInternal(lockDAO, key);
 return lock != null ? false : true;
 }
 /**  加锁  **/
 public static boolean tryLock(GlobalLockTableDAO lockDAO, String key) { return tryLockInternal(lockDAO, key) == null ? true : false;
 }
 /**  解锁  **/
 public static void releasLock(GlobalLockTableDAO lockDAO, String key) { lockDAO.deleteByLockKey(key);
 }
}

这个工具类有 2 个特别有意思的地方,先看注意的地方 2(上面代码中标识了)

1. 为了避免锁长时间不释放,用 Redis 实现的话可以设置锁超时时间,超时自动释放(后面会写用 Redis 实现分布式锁)用 MySQL 实现的话可以先删除后添加。可以看到删除的时候使用 id 删的,不是用 name 删的。为啥呢?先自己想一下

因为如果是通过 name 删的话,有可能别人删了这个锁后,又通过 name 加了锁,还没到超时时间,结果你却根据 name 删除了。通过 id 删的话,当返回的 id= 0 时,说明别人已经重新加锁了,你需要重新获取。

2.GlobalLockTable 对象 dao 层的其他方法都见名知意,来看一个这个方法。即代码中的注意点 1
可以看到每次尝试加锁的时候,并不是先 select,而是直接 insertSelectiveWithTest,这样就少了一个查询时间,提高了效率

insertSelectiveWithTest 的作用是当 lockKey 存在时不进行插入操作,返回 0。当 lockKey 不存在时进行插入操作,返回 1

insert id= insertSelectiveWithTest  useGeneratedKeys= true  keyProperty= id  parameterType= com.javashitang.middleware.lock.mysql.pojo.GlobalLockTable 
 insert into `globallocktable` (`id`,
 `lockKey`, `createTime` )
 select #{id,jdbcType=INTEGER}, #{lockKey,jdbcType=VARCHAR}, #{createTime,jdbcType=TIMESTAMP}
 from dual where not exists
 (select 1 from globallocktable where lockKey = #{lockKey,jdbcType=VARCHAR})
 /insert

使用

当我们想使用时,就只写业务逻辑就行了,非常方便

if (!globalLockComponent.tryLock(name)) {
 //  没有获取到锁返回
 return;
try { //  这里写业务逻辑} catch (Exception e) {
} finally { globalLockComponent.releasLock(name)

关于怎么在 MySQL 数据库中实现一个分布式锁就分享到这里了,希望以上内容可以对大家有一定的帮助,可以学到更多知识。如果觉得文章不错,可以把它分享出去让更多的人看到。

向 AI 问一下细节

正文完
 
丸趣
版权声明:本站原创文章,由 丸趣 2023-12-04发表,共计4070字。
转载说明:除特殊说明外本站除技术相关以外文章皆由网络搜集发布,转载请注明出处。
评论(没有评论)