架构(十三)动态本地锁

一、引言

        加锁大家都知道,但是目前提供动态锁的基本都是分布式锁,根据订单或者某个收费款项进行加锁。比如这个1订单要收刷卡费用,那就OREDER_1做为key丢到redis进行分布式加锁。这也是当下分布式锁最流行的方式。

        但是对于平台项目或者一些并发程度低的场景,分布式锁就没有必要了,本地锁更加方便。但是本地锁只有synchronized、ReentrantLock之类的方式,想动态的加锁只用他们是实现不了的。

二、实现

        那么如果要实现动态的本地锁怎么做呢?

        先看看redis的分布式锁是怎么做的,他其实就是利用redis的单线程往里面存一个值,如果已经有线程存了,并发的线程就存不进去,只能等。只不过各个redis的客户端还要考虑删除的并发性、锁超时删除、加锁等待这些问题。

1、ConcurrentHashMap

        借鉴这个方案,本地也可以加个存储,为了并发的可读性使用ConcurrentHashMap,这里可以有效的避免其他线程解锁删除缓存

private static final ConcurrentHashMap<String, String> map = 
new ConcurrentHashMap<>();

        加锁就把OREDER_1塞到map里面塞的过程需要防止并发,所以使用synchronized之类的就可以,因为map塞数据可比业务执行的加锁时间短多了

private synchronized static boolean getLock(String key) {// map里面塞一下很快,可以使用synchronizedif (map.containsKey(key)) {// 懒汉模式,再判断一遍,免得两个线程一起通过了外层的判断return false;}map.put(key,key);return true;}

        加锁的方法就是先判断一下有没有已经占了位置的,没有就往map里面占位置

public static boolean tryLock(String key, long timeout, TimeUnit unit) {if (map.containsKey(key)) {return false;}return getLock(key);}

        解锁就是直接删除

public static void unLock(String key) {if (!map.containsKey(key)) {return;}// 释放锁map.remove(key);}

        这是最简单的做法,那么上面的实现有什么问题呢?最大的问题就是删除的时候可能被其他线程给删了,毕竟不会所有人都按照预想的去使用工具,安全是架构应该考虑的。

        还有锁的超时、等待多长时间没有锁就失败两个功能点

2、优化解锁、锁超时

       要优化这个实现就可以结合ReentrantLock,他有判断是否本线程加锁和等待多长时间进行加锁的api,如果自己实现相当于把他里面的线程存储和睡眠唤醒给重复做一遍,没有必要。

        那么怎么用它呢,map的value存储一个lock,相当于一个key一个lock,生成和删除的时候可以使用synchronized,这个和刚刚说的一样,map里面塞一个删一个是很快的,new一个lock和lock.unlock也是很快的,主要的时间都在业务处理和同一场景下不同单号之间的阻塞

public class LockKeyUtil {private static final ConcurrentHashMap<String, ReentrantLock> map = new ConcurrentHashMap<>();/*** 从map里获取锁 如果存在则返回 不存在则创建** @param key key*/private synchronized static ReentrantLock getReentrantLock(String key) {if (map.containsKey(key)) {return map.get(key);}return map.compute(key, (k, lock) -> {lock = new ReentrantLock();return lock;});}/*** * @param key* @param waitTimeout* @param unit* @return*/public static boolean tryLock(String key, long waitTimeout, TimeUnit unit) {ReentrantLock lock = getReentrantLock(key);boolean res;try {res = lock.tryLock(waitTimeout, unit);} catch (InterruptedException e) {unLock(key);res = false;}return res;}/*** 释放锁** @param key key*/public static synchronized void unLock(String key) {if (!map.containsKey(key)) {return;}ReentrantLock lock = map.get(key);// 释放锁if (lock.isHeldByCurrentThread()) {lock.unlock();map.remove(key);}}}

3、优化加锁队列

        上面的实现可以看到还有一个问题,如果在tryLock的时候,多个线程进入了,那么第一个线程解锁的时候把他移除map就有问题了,所以unlock还可以根据加锁队列优化一下

        通过getQueueLength知道有没有在等待加锁的线程,当然了这三行代码不是原子性的,所以是有可能刚刚取完队列还没删除map之前就有线程去加锁了,但是这种情况并发几率可以说是万分之一不到,可以不考虑

public synchronized static void unLock(String key) {if (!map.containsKey(key)) {return;}ReentrantLock lock = map.get(key);// 释放锁if (lock.isHeldByCurrentThread()) {int wait = lock.getQueueLength();lock.unlock();if (wait == 0) {map.remove(key);}}}

4、执行超时自动解锁

        这里还有一个执行超时自动解锁的功能,其实感觉没必要,用的时候一般都是在finally里面去unlock,所以几乎不会有不解锁的情况

String key = LOCK + request.getId();boolean locked = LockKeyUtil.tryLock(key, 3, TimeUnit.SECONDS);if (!locked) {throw new OrderException("LOCK_FAIL");}try {//业务处理} finally {LockKeyUtil.unLock(key);}

       

三、测试

1、相同key测试代码

        这段代码主要是让线程1或者3先拿到锁,那么其中一个和3就一定处于等待状态,如果是3在等,他到最后也抢不到锁

public static void main(String[] args) {String key = "order_1666";Thread thread1 = new Thread(() -> {LocalDateTime now = LocalDateTime.now();DateTimeFormatter formatter = DateTimeFormatter.ofPattern("yyyy-MM-dd HH:mm:ss");String formattedDateTime = now.format(formatter);System.out.println("thread1:" + Thread.currentThread().getName() + " start:" + formattedDateTime);boolean res = LockKeyUtil.tryLock(key, 2, TimeUnit.SECONDS);System.out.println("thread1:" + Thread.currentThread().getName() + " lock res:" + res);try {Thread.sleep(2000);} catch (InterruptedException e) {e.printStackTrace();}LockKeyUtil.unLock(key, true);now = LocalDateTime.now();formattedDateTime = now.format(formatter);System.out.println("thread1:" + Thread.currentThread().getName() + " end:" + formattedDateTime);});Thread thread2 = new Thread(() -> {try {Thread.sleep(1);LocalDateTime now = LocalDateTime.now();DateTimeFormatter formatter = DateTimeFormatter.ofPattern("yyyy-MM-dd HH:mm:ss");String formattedDateTime = now.format(formatter);System.out.println("thread2:" + Thread.currentThread().getName() + " start:" + formattedDateTime);boolean res = LockKeyUtil.tryLock(key, 5, TimeUnit.SECONDS);System.out.println("thread2:" + Thread.currentThread().getName() + " lock res:" + res);Thread.sleep(5000);LockKeyUtil.unLock(key, true);now = LocalDateTime.now();formattedDateTime = now.format(formatter);System.out.println("thread2:" + Thread.currentThread().getName() + " end:" + formattedDateTime);} catch (InterruptedException e) {e.printStackTrace();}});Thread thread3 = new Thread(() -> {LocalDateTime now = LocalDateTime.now();DateTimeFormatter formatter = DateTimeFormatter.ofPattern("yyyy-MM-dd HH:mm:ss");String formattedDateTime = now.format(formatter);System.out.println("thread3:" + Thread.currentThread().getName() + " start:" + formattedDateTime);boolean res = LockKeyUtil.tryLock(key, 1, TimeUnit.SECONDS);System.out.println("thread3:" + Thread.currentThread().getName() + " lock res:" + res);try {Thread.sleep(1000);} catch (InterruptedException e) {e.printStackTrace();}LockKeyUtil.unLock(key, true);now = LocalDateTime.now();formattedDateTime = now.format(formatter);System.out.println("thread3:" + Thread.currentThread().getName() + " end:" + formattedDateTime);});thread1.start();thread2.start();thread3.start();}

 

2、相同key测试结果

        结果和预期相符,线程1先抢到了,执行完之后线程2 抢到,线程3在过程中就失败了

        整个执行链路也给打出来了

thread1:Thread-0 start:2024-02-06 13:59:12
thread3:Thread-2 start:2024-02-06 13:59:12
thread2:Thread-1 start:2024-02-06 13:59:12
thread:Thread-0 tryLock key:order_1666 start
thread:Thread-2 tryLock key:order_1666 start
thread:Thread-1 tryLock key:order_1666 start
thread:Thread-0 getReentrantLock key:order_1666 start
thread:Thread-0 getReentrantLock key:order_1666 new ReentrantLock()
thread:Thread-1 getReentrantLock key:order_1666 start
thread:Thread-0 tryLock key:order_1666res:true
thread1:Thread-0 lock res:true
thread:Thread-1 getReentrantLock key:order_1666 containsKey
thread:Thread-2 getReentrantLock key:order_1666 start
thread:Thread-2 getReentrantLock key:order_1666 containsKey
thread:Thread-2 tryLock key:order_1666res:false
thread3:Thread-2 lock res:false
thread:Thread-0 unLock key:order_1666 unlock success
thread:Thread-2 unLock key:order_1666 not isHeldByCurrentThread
thread:Thread-1 tryLock key:order_1666res:true
thread2:Thread-1 lock res:true
thread1:Thread-0 end:2024-02-06 13:59:14
thread3:Thread-2 end:2024-02-06 13:59:14
thread:Thread-1 unLock key:order_1666 unlock success
thread:Thread-1 unLock key:order_1666 map remove
thread2:Thread-1 end:2024-02-06 13:59:19

3、不同key测试

        就是多加一个key去加锁解锁,看能不能同时加同时解        

public static void main(String[] args) {String key = "order_1666";Thread thread1 = new Thread(() -> {LocalDateTime now = LocalDateTime.now();DateTimeFormatter formatter = DateTimeFormatter.ofPattern("yyyy-MM-dd HH:mm:ss");String formattedDateTime = now.format(formatter);System.out.println("thread1:" + Thread.currentThread().getName() + " start:" + formattedDateTime);boolean res = LockKeyUtil.tryLock(key, 2, TimeUnit.SECONDS);System.out.println("thread1:" + Thread.currentThread().getName() + " lock res:" + res);try {Thread.sleep(2000);} catch (InterruptedException e) {e.printStackTrace();}LockKeyUtil.unLock(key, true);now = LocalDateTime.now();formattedDateTime = now.format(formatter);System.out.println("thread1:" + Thread.currentThread().getName() + " end:" + formattedDateTime);});Thread thread2 = new Thread(() -> {try {Thread.sleep(1);LocalDateTime now = LocalDateTime.now();DateTimeFormatter formatter = DateTimeFormatter.ofPattern("yyyy-MM-dd HH:mm:ss");String formattedDateTime = now.format(formatter);System.out.println("thread2:" + Thread.currentThread().getName() + " start:" + formattedDateTime);boolean res = LockKeyUtil.tryLock(key, 5, TimeUnit.SECONDS);System.out.println("thread2:" + Thread.currentThread().getName() + " lock res:" + res);Thread.sleep(5000);LockKeyUtil.unLock(key, true);now = LocalDateTime.now();formattedDateTime = now.format(formatter);System.out.println("thread2:" + Thread.currentThread().getName() + " end:" + formattedDateTime);} catch (InterruptedException e) {e.printStackTrace();}});String key3 = "order_1999";Thread thread3 = new Thread(() -> {LocalDateTime now = LocalDateTime.now();DateTimeFormatter formatter = DateTimeFormatter.ofPattern("yyyy-MM-dd HH:mm:ss");String formattedDateTime = now.format(formatter);System.out.println("thread3:" + Thread.currentThread().getName() + " start:" + formattedDateTime);boolean res = LockKeyUtil.tryLock(key3, 1, TimeUnit.SECONDS);System.out.println("thread3:" + Thread.currentThread().getName() + " lock res:" + res);try {Thread.sleep(1000);} catch (InterruptedException e) {e.printStackTrace();}LockKeyUtil.unLock(key3, true);now = LocalDateTime.now();formattedDateTime = now.format(formatter);System.out.println("thread3:" + Thread.currentThread().getName() + " end:" + formattedDateTime);});thread1.start();thread2.start();thread3.start();}

        结果是不同的key可以同时加锁的,这就实现了锁的动态性

thread2:Thread-1 start:2024-02-06 14:06:28
thread1:Thread-0 start:2024-02-06 14:06:28
thread3:Thread-2 start:2024-02-06 14:06:28
thread3:Thread-2 lock res:true
thread2:Thread-1 lock res:true
thread3:Thread-2 end:2024-02-06 14:06:29
thread1:Thread-0 lock res:false
thread1:Thread-0 end:2024-02-06 14:06:32
thread2:Thread-1 end:2024-02-06 14:06:33

四、总结        

        最后的实现就是这样了

f6d008a35cc144f89dec2df89714664d.png

        看起来很简单的功能,要考虑的东西其实很多,这种实现也不是唯一的,有兴趣可以跟作者讨论其他的实现方案

 

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处:http://www.hqwc.cn/news/466798.html

如若内容造成侵权/违法违规/事实不符,请联系编程知识网进行投诉反馈email:809451989@qq.com,一经查实,立即删除!

相关文章

【深度学习】基于多层感知机的手写数字识别

案例2&#xff1a;构建自己的多层感知机: MNIST手写数字识别 相关知识点: numpy科学计算包&#xff0c;如向量化操作&#xff0c;广播机制等 1 任务目标 1.1 数据集简介 ​ MNIST手写数字识别数据集是图像分类领域最常用的数据集之一&#xff0c;它包含60,000张训练图片&am…

fatal error: rtiostream_utils.h: No such file or directory, rtiostream.h

fatal error: rtiostream_utils.h: No such file or directory 我的设置&#xff1a;

C语言rand随机数知识解析和猜数字小游戏

rand随机数 rand C语言中提供了一个可以随机生成一个随机数的函数&#xff1a;rand&#xff08;&#xff09; 函数原型&#xff1a; int rand(void);rand函数返回的值的区间是&#xff1a;0~RAND_MAX(32767)之间。大部分编译器都是32767。 #include<stdlib.h> int ma…

linux系统下vscode portable版本的python环境搭建003:venv

这里写自定义目录标题 python安装方案一. 使用源码安装&#xff08;有[构建工具](https://blog.csdn.net/ResumeProject/article/details/136095629)的情况下&#xff09;方案二.使用系统包管理器 虚拟环境安装TESTCG 本文目的&#xff1a;希望在获得一个新的系统之后&#xff…

vue 向某个网址 传递数据

1. 需求 现在有一个网站需要 配置上另一个网站的东西 类似这样的东西吧 就是我需要再一个网站上 右边或者其他地方 放另一个页面的地址 这个地址需要给我传递东西 或我这个网站给其他的网站传递token了 id等 2.解决 window.parent.postMessage({ token: loginRes.token, id:…

【Java程序设计】【C00249】基于Springboot的私人健身与教练预约管理系统(有论文)

基于Springboot的私人健身与教练预约管理系统&#xff08;有论文&#xff09; 项目简介项目获取开发环境项目技术运行截图 项目简介 这是一个基于Springboot的私人健身与教练预约管理系统 本系统分为系统功能模块、管理员功能模块、教练功能模块以及用户功能模块。 系统功能模…

2.11日学习打卡----初学RocketMQ(二)

2.11日学习打卡 一. RocketMQ整合springboot 首先配置pom.xml文件 <dependency><groupId>org.projectlombok</groupId><artifactId>lombok</artifactId><scope>annotationProcessor</scope></dependency><dependency>…

数据库基础学习笔记

一.基础概念 数据库、数据库管理系统、SQL 主流数据库&#xff1a; mysql的安装&#xff1a;略 mysql图形化界面的安装&#xff1a;略 二.数据模型 1). 关系型数据库&#xff08;RDBMS&#xff09; 概念&#xff1a;建立在关系模型基础上&#xff0c;由多张相互连接的二维表…

爬爬爬——今天是浏览器窗口切换和给所选人打钩(自动化)

学习爬虫路还很长&#xff0c;第一阶段花了好多天了&#xff0c;还在底层&#xff0c;虽然不是我专业要学习的语言&#xff0c;和必备的知识&#xff0c;但是我感觉还挺有意思的。加油&#xff0c;这两天把建模和ai也不学了&#xff0c;唉过年了懒了&#xff01; 加油坚持就是…

例38:使用Frame(分组框)

建立一个EXE工程&#xff0c;在窗体上放两个Frame框。分别放两组单选按钮表示性别和收入&#xff0c;注意每组单选按钮的组名要一样。在按钮中输入代码&#xff1a; Sub Form1_Command1_BN_Clicked(hWndForm As hWnd, hWndControl As hWnd)If Frame1.Visible ThenFrame1.Visib…

幻兽帕鲁游戏配置面板在哪里?腾讯云轻量应用服务器的一键参数配置面板

如何找到游戏配置面板&#xff1f; 目前&#xff0c;满足以下条件的Lighthouse&#xff0c;可以在实例详情页-应用管理页看到幻兽帕鲁的游戏配置面板。 如果你使用了一键/极简部署的方式开服&#xff0c;那么需要保存游戏存档后&#xff0c;将服务器重装系统&#xff0c;否则将…