细粒度锁的实现

2024-09-08 01:08
文章标签 实现 细粒度

本文主要是介绍细粒度锁的实现,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!

 最近在工作上碰见了一些高并发的场景需要加锁来保证业务逻辑的正确性,并且要求加锁后性能不能受到太大的影响。初步的想法是通过数据的时间戳,id等关键字来加锁,从而保证不同类型数据处理的并发性。而java自身api提供的锁粒度太大,很难同时满足这些需求,于是自己动手写了几个简单的扩展...

    1. 分段锁

        借鉴concurrentHashMap的分段思想,先生成一定数量的锁,具体使用的时候再根据key来返回对应的lock。这是几个实现里最简单,性能最高,也是最终被采用的锁策略,代码如下:

/*** 分段锁,系统提供一定数量的原始锁,根据传入对象的哈希值获取对应的锁并加锁* 注意:要锁的对象的哈希值如果发生改变,有可能导致锁无法成功释放!!!*/
public class SegmentLock<T> {private Integer segments = 16;//默认分段数量private final HashMap<Integer, ReentrantLock> lockMap = new HashMap<>();public SegmentLock() {init(null, false);}public SegmentLock(Integer counts, boolean fair) {init(counts, fair);}private void init(Integer counts, boolean fair) {if (counts != null) {segments = counts;}for (int i = 0; i < segments; i++) {lockMap.put(i, new ReentrantLock(fair));}}public void lock(T key) {ReentrantLock lock = lockMap.get((key.hashCode()>>>1) % segments);lock.lock();}public void unlock(T key) {ReentrantLock lock = lockMap.get((key.hashCode()>>>1) % segments);lock.unlock();}
}

    2. 哈希锁

        上述分段锁的基础上发展起来的第二种锁策略,目的是实现真正意义上的细粒度锁。每个哈希值不同的对象都能获得自己独立的锁。在测试中,在被锁住的代码执行速度飞快的情况下,效率比分段锁慢 30% 左右。如果有长耗时操作,感觉表现应该会更好。代码如下:

public class HashLock<T> {private boolean isFair = false;private final SegmentLock<T> segmentLock = new SegmentLock<>();//分段锁private final ConcurrentHashMap<T, LockInfo> lockMap = new ConcurrentHashMap<>();public HashLock() {}public HashLock(boolean fair) {isFair = fair;}public void lock(T key) {LockInfo lockInfo;segmentLock.lock(key);try {lockInfo = lockMap.get(key);if (lockInfo == null) {lockInfo = new LockInfo(isFair);lockMap.put(key, lockInfo);} else {lockInfo.count.incrementAndGet();}} finally {segmentLock.unlock(key);}lockInfo.lock.lock();}public void unlock(T key) {LockInfo lockInfo = lockMap.get(key);if (lockInfo.count.get() == 1) {segmentLock.lock(key);try {if (lockInfo.count.get() == 1) {lockMap.remove(key);}} finally {segmentLock.unlock(key);}}lockInfo.count.decrementAndGet();lockInfo.unlock();}private static class LockInfo {public ReentrantLock lock;public AtomicInteger count = new AtomicInteger(1);private LockInfo(boolean fair) {this.lock = new ReentrantLock(fair);}public void lock() {this.lock.lock();}public void unlock() {this.lock.unlock();}}
}

 

    3. 弱引用锁

        哈希锁因为引入的分段锁来保证锁创建和销毁的同步,总感觉有点瑕疵,所以写了第三个锁来寻求更好的性能和更细粒度的锁。这个锁的思想是借助java的弱引用来创建锁,把锁的销毁交给jvm的垃圾回收,来避免额外的消耗。

        有点遗憾的是因为使用了ConcurrentHashMap作为锁的容器,所以没能真正意义上的摆脱分段锁。这个锁的性能比 HashLock 快10% 左右。锁代码:

/*** 弱引用锁,为每个独立的哈希值提供独立的锁功能*/
public class WeakHashLock<T> {private ConcurrentHashMap<T, WeakLockRef<T, ReentrantLock>> lockMap = new ConcurrentHashMap<>();private ReferenceQueue<ReentrantLock> queue = new ReferenceQueue<>();public ReentrantLock get(T key) {if (lockMap.size() > 1000) {clearEmptyRef();}WeakReference<ReentrantLock> lockRef = lockMap.get(key);ReentrantLock lock = (lockRef == null ? null : lockRef.get());while (lock == null) {lockMap.putIfAbsent(key, new WeakLockRef<>(new ReentrantLock(), queue, key));lockRef = lockMap.get(key);lock = (lockRef == null ? null : lockRef.get());if (lock != null) {return lock;}clearEmptyRef();}return lock;}@SuppressWarnings("unchecked")private void clearEmptyRef() {Reference<? extends ReentrantLock> ref;while ((ref = queue.poll()) != null) {WeakLockRef<T, ? extends ReentrantLock> weakLockRef = (WeakLockRef<T, ? extends ReentrantLock>) ref;lockMap.remove(weakLockRef.key);}}private static final class WeakLockRef<T, K> extends WeakReference<K> {final T key;private WeakLockRef(K referent, ReferenceQueue<? super K> q, T key) {super(referent, q);this.key = key;}}
}

 

后记

    最开始想借助 locksupport 和 AQS 来实现细粒度锁,写着写着发现正在实现的东西和java 原生的锁区别不大,于是放弃改为对java自带锁的封装,浪费了不少时间。

    实际上在实现了这些细粒度锁之后,又有了新的想法,比如可以通过分段思想将数据提交给专门的线程来处理,可以减少大量线程的阻塞时间,留待日后探索...

这篇关于细粒度锁的实现的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!



http://www.chinasem.cn/article/1146687

相关文章

Spring Security自定义身份认证的实现方法

《SpringSecurity自定义身份认证的实现方法》:本文主要介绍SpringSecurity自定义身份认证的实现方法,下面对SpringSecurity的这三种自定义身份认证进行详细讲解,... 目录1.内存身份认证(1)创建配置类(2)验证内存身份认证2.JDBC身份认证(1)数据准备 (2)配置依

利用python实现对excel文件进行加密

《利用python实现对excel文件进行加密》由于文件内容的私密性,需要对Excel文件进行加密,保护文件以免给第三方看到,本文将以Python语言为例,和大家讲讲如何对Excel文件进行加密,感兴... 目录前言方法一:使用pywin32库(仅限Windows)方法二:使用msoffcrypto-too

C#使用StackExchange.Redis实现分布式锁的两种方式介绍

《C#使用StackExchange.Redis实现分布式锁的两种方式介绍》分布式锁在集群的架构中发挥着重要的作用,:本文主要介绍C#使用StackExchange.Redis实现分布式锁的... 目录自定义分布式锁获取锁释放锁自动续期StackExchange.Redis分布式锁获取锁释放锁自动续期分布式

springboot使用Scheduling实现动态增删启停定时任务教程

《springboot使用Scheduling实现动态增删启停定时任务教程》:本文主要介绍springboot使用Scheduling实现动态增删启停定时任务教程,具有很好的参考价值,希望对大家有... 目录1、配置定时任务需要的线程池2、创建ScheduledFuture的包装类3、注册定时任务,增加、删

SpringBoot整合mybatisPlus实现批量插入并获取ID详解

《SpringBoot整合mybatisPlus实现批量插入并获取ID详解》这篇文章主要为大家详细介绍了SpringBoot如何整合mybatisPlus实现批量插入并获取ID,文中的示例代码讲解详细... 目录【1】saveBATch(一万条数据总耗时:2478ms)【2】集合方式foreach(一万条数

使用Python实现矢量路径的压缩、解压与可视化

《使用Python实现矢量路径的压缩、解压与可视化》在图形设计和Web开发中,矢量路径数据的高效存储与传输至关重要,本文将通过一个Python示例,展示如何将复杂的矢量路径命令序列压缩为JSON格式,... 目录引言核心功能概述1. 路径命令解析2. 路径数据压缩3. 路径数据解压4. 可视化代码实现详解1

PyQt6/PySide6中QTableView类的实现

《PyQt6/PySide6中QTableView类的实现》本文主要介绍了PyQt6/PySide6中QTableView类的实现,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学... 目录1. 基本概念2. 创建 QTableView 实例3. QTableView 的常用属性和方法

PyQt6/PySide6中QTreeView类的实现

《PyQt6/PySide6中QTreeView类的实现》QTreeView是PyQt6或PySide6库中用于显示分层数据的控件,本文主要介绍了PyQt6/PySide6中QTreeView类的实现... 目录1. 基本概念2. 创建 QTreeView 实例3. QTreeView 的常用属性和方法属性

Android使用ImageView.ScaleType实现图片的缩放与裁剪功能

《Android使用ImageView.ScaleType实现图片的缩放与裁剪功能》ImageView是最常用的控件之一,它用于展示各种类型的图片,为了能够根据需求调整图片的显示效果,Android提... 目录什么是 ImageView.ScaleType?FIT_XYFIT_STARTFIT_CENTE

pandas中位数填充空值的实现示例

《pandas中位数填充空值的实现示例》中位数填充是一种简单而有效的方法,用于填充数据集中缺失的值,本文就来介绍一下pandas中位数填充空值的实现,具有一定的参考价值,感兴趣的可以了解一下... 目录什么是中位数填充?为什么选择中位数填充?示例数据结果分析完整代码总结在数据分析和机器学习过程中,处理缺失数