设计一个利用事务特性可以阻塞线程的排他锁,并且通过注解和 AOP 来实现

ops/2025/1/12 22:35:37/

设计思路:
利用数据库表记录锁标识:通过唯一标识符(如方法名 + 参数),我们可以在数据库中插入一条记录,表示当前方法正在执行。这条记录需要记录插入时间。
注解:通过注解标识哪些方法需要加锁,我们可以动态生成唯一标识符。
AOP:使用 AOP 切面处理方法调用逻辑,确保在进入目标方法之前判断是否已经有其他线程持有锁。如果有,则阻塞当前线程,直到锁被释放。
事务:利用事务的隔离性,保证在同一时刻,只有一个线程能够执行某个方法,其他线程需要等待。
超时检测:如果在某个时间点锁没有被释放(例如,锁存在时间超过10分钟),则认为发生了死锁,自动清理相关记录。
步骤及代码实现

  1. 数据库表设计
    首先,我们需要一个数据库表来存储锁的状态。表的结构大致如下:
CREATE TABLE method_lock (id BIGINT AUTO_INCREMENT PRIMARY KEY,lock_key VARCHAR(255) NOT NULL, -- 锁的唯一标识(方法名+参数)created_time TIMESTAMP DEFAULT CURRENT_TIMESTAMP, -- 锁的创建时间updated_time TIMESTAMP DEFAULT CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP, -- 最后更新时间status VARCHAR(50) DEFAULT 'LOCKED', -- 锁的状态,LOCKED 表示锁定,RELEASED 表示释放expired BOOLEAN DEFAULT FALSE -- 是否过期,10分钟未释放的锁算死锁
);
  1. 创建锁的注解
    接下来,我们定义一个注解来标识需要加锁的方法。
@Target(ElementType.METHOD)
@Retention(RetentionPolicy.RUNTIME)
public @interface MethodLock {String value(); // 锁的标识(如方法名 + 参数的唯一标识符)
}
  1. 实现 AOP 切面
    在切面中,我们要通过 @Around 来拦截带有 @MethodLock 注解的方法,获取方法的唯一标识,并在数据库中插入或查询锁的状态。
@Aspect
@Component
public class MethodLockAspect {@Autowiredprivate LockRepository lockRepository; // 用于查询和操作数据库的 repository@Around("@annotation(methodLock)") // 拦截带有 @MethodLock 注解的方法public Object around(ProceedingJoinPoint joinPoint, MethodLock methodLock) throws Throwable {String lockKey = methodLock.value() + getMethodSignature(joinPoint); // 生成锁的唯一标识符// 获取当前时间戳long currentTime = System.currentTimeMillis();// 尝试获取锁if (tryAcquireLock(lockKey, currentTime)) {try {// 如果获取到锁,则执行目标方法return joinPoint.proceed();} finally {// 目标方法执行完毕后释放锁releaseLock(lockKey);}} else {// 如果无法获取到锁,则阻塞当前线程或者抛出异常throw new RuntimeException("Unable to acquire lock, try again later.");}}private boolean tryAcquireLock(String lockKey, long currentTime) {// 查询数据库中是否存在该锁记录LockEntity lockEntity = lockRepository.findByLockKey(lockKey);if (lockEntity == null) {// 如果不存在锁记录,则插入新的锁记录LockEntity newLockEntity = new LockEntity();newLockEntity.setLockKey(lockKey);newLockEntity.setCreatedTime(new Timestamp(currentTime));lockRepository.save(newLockEntity);return true;} else {// 如果锁存在,检查锁是否已经过期(超过10分钟)long lockAge = currentTime - lockEntity.getCreatedTime().getTime();if (lockAge > 10 * 60 * 1000) {// 如果锁超时超过10分钟,认为是死锁,清理并重新获取锁lockRepository.deleteByLockKey(lockKey); // 删除死锁记录LockEntity newLockEntity = new LockEntity();newLockEntity.setLockKey(lockKey);newLockEntity.setCreatedTime(new Timestamp(currentTime));lockRepository.save(newLockEntity);return true;}// 如果锁没有过期,则阻塞当前线程return false;}}private void releaseLock(String lockKey) {// 释放锁:删除数据库中的锁记录lockRepository.deleteByLockKey(lockKey);}private String getMethodSignature(ProceedingJoinPoint joinPoint) {// 获取方法签名(例如方法名 + 参数)String methodName = joinPoint.getSignature().getName();Object[] args = joinPoint.getArgs();StringBuilder signature = new StringBuilder(methodName);for (Object arg : args) {signature.append("-").append(arg != null ? arg.toString() : "null");}return signature.toString();}
}
  1. LockRepository 示例
    假设使用的是 Spring Data JPA 来进行数据库操作,我们可以定义一个简单的 Repository 来操作 method_lock 表。
public interface LockRepository extends JpaRepository<LockEntity, Long> {LockEntity findByLockKey(String lockKey);void deleteByLockKey(String lockKey);
}
  1. LockEntity 实体类
    LockEntity 对应数据库中的 method_lock 表:
@Entity
@Table(name = "method_lock")
public class LockEntity {@Id@GeneratedValue(strategy = GenerationType.IDENTITY)private Long id;private String lockKey;@Column(name = "created_time")private Timestamp createdTime;@Column(name = "updated_time")private Timestamp updatedTime;private String status;private Boolean expired;// Getters and Setters
}
  1. 使用示例
    现在,我们可以在需要加锁的方法上使用 @MethodLock 注解,示例如下:
@Service
public class MyService {@MethodLock("uniqueLockKey")@Transactionalpublic void executeTask(String param) {// 执行需要加锁的逻辑System.out.println("Executing task with param: " + param);}
}

关键点总结:
锁的唯一标识:通过方法名和参数生成一个唯一的标识符 lockKey。
数据库表记录锁:通过表记录锁的状态,保证只有一个线程能获取到锁。
AOP 和注解结合:使用 AOP 和注解对方法进行拦截,判断是否已经有其他线程在执行,若没有则获取锁,若有则阻塞或抛出异常。
事务特性:保证同一时刻,只有一个线程能够执行目标方法,其他线程需要等待,。
死锁处理:在锁存在超过 10 分钟时认为是死锁,进行清理。
通过这种方式,可以实现基于事务的排他锁,确保多个线程访问同一资源时进行排队执行

解释:
例如,多个接口调用一个方法,要求多个线程调用该方法时,这些线程排队执行,该方法使用注解,aop切面,通过该注解上的特定字符串和方法名,组成唯一标识,将该标识插入表中,并且记录插入时间,并且该标识是唯一的,当有线程调用该方法时,先组装唯一标识,查询表中是否有该标识的数据,并判断是否时间距离现在是否超过了10分钟,如果超过10分钟就是死锁了,直接删除该标识的所有记录,如果没有查到就插入一条记录,如果该方法没有执行完,就在一个事务里面,该事务没有结束,当有其他方法调用该方法时,就回去查询并插入,由于前一个事务未结束,导致当前唯一标识一致,卡在插入数据时,从而达到阻塞的目的,最后方法结束后将该数据删除。


http://www.ppmy.cn/ops/149570.html

相关文章

PyTorch reshape函数介绍

torch.reshape 是 PyTorch 用于改变张量形状的函数之一。它不会改变张量的数据&#xff0c;而是重新组织其元素以适应新的形状。 reshape 的使用 torch.reshape(input, shape) → Tensorinput&#xff1a;输入张量。shape&#xff1a;新形状&#xff0c;使用整数或 -1 指定各维…

手机的ip地址是根据电话卡归属地定吗

在智能手机普及的今天&#xff0c;IP地址作为我们连接互联网的“门牌号”&#xff0c;其来源和确定方式常常引发用户的好奇。特别是关于手机IP地址是否与电话卡的归属地直接相关&#xff0c;这一话题更是众说纷纭。本文将深入探讨这一问题&#xff0c;为您揭开手机IP地址与电话…

【Unity插件】解决移动端UI安全区问题 - Safe Area Helper

在移动端设计界面时&#xff0c;必须要考虑的就是UI的安全区。 Unity本身也提供了Safearea的API。 但在asset store时已经有人提供了免费的插件&#xff08;Safe Area Helper&#xff09;&#xff0c;我们可以直接使用。 插件链接&#xff1a; https://assetstore.unity.com/p…

Linux 虚拟机与windows主机之间的文件传输--设置共享文件夹方式

Linux 虚拟机与windows主机之间的文件传输 设置共享文件夹方式 在虚拟机中打开终端查看是否已经新建完成&#xff0c;到文件夹中找到它看一下&#xff0c;这个位置就能存储东西啦

【物联网原理与运用】知识点总结(上)

目录 名词解释汇总 第一章 物联网概述 1.1物联网的基本概念及演进 1.2 物联网的内涵 1.3 物联网的特性——泛在性 1.4 物联网的基本特征与属性&#xff08;五大功能域&#xff09; 1.5 物联网的体系结构 1.6 物联网的关键技术 1.7 物联网的应用领域 第二章 感知与识别技术 2.1 …

【C语言】_类型重命名typedef关键字

目录 1. 从复杂类型理解类型重命名的必要性 1.1 示例1 1.2 示例2 2. typedef 用法 2.1 简单类型重命名 2.2 复杂指针类型重命名 2.2.1 数组指针类型 2.2.2 函数指针类型 3. 使用typedef简化复杂类型的定义 4. #define与typedef 1. 从复杂类型理解类型重命名的必要性 …

阿里云ios镜像源

阿里云镜像源&#xff1a;阿里巴巴开源镜像站-OPSX镜像站-阿里云开发者社区 下载centos7

排列高手

这篇主要是求再排位为 {1&#xff0c;3&#xff0c;4&#xff0c;....&#xff0c;n&#xff0c;2}的最优顺序下求mex。 但不知道为什么这样是最优 子数列的个数公式&#xff1a; 对于一个长度为N的数组&#xff0c; #include <bits/stdc.h> using namespace std; lon…