springboot中设计基于Redisson的分布式锁注解

ops/2024/11/22 23:31:56/

如何使用AOP设计一个分布式锁注解?

1、在pom.xml中配置依赖

        <dependency><groupId>org.springframework</groupId><artifactId>spring-aspects</artifactId><version>5.3.26</version></dependency><dependency><groupId>org.springframework</groupId><artifactId>spring-aop</artifactId></dependency>

2、逻辑代码

创建一下多个文件夹,并复制粘贴进代码

2.1、RedissonConfig.java

需要配置一下Redisson

java">import org.redisson.Redisson;
import org.redisson.api.RedissonClient;
import org.redisson.config.Config;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;/*** Redisson 配置**/
@Configuration
public class RedissonConfig {@Beanpublic RedissonClient redissonClient() {Config config = new Config();config.useSingleServer().setAddress("redis://192.168.57.111:6379") // Redis地址.setPassword(null) // 如果没有密码,设置为null.setDatabase(0); // 使用的Redis数据库索引RedissonClient redissonClient = Redisson.create(config);// 测试连接try {redissonClient.getKeys().count();System.out.println("Redisson connected to Redis successfully.");} catch (Exception e) {System.err.println("Failed to connect to Redis: " + e.getMessage());}return redissonClient;}
}

需要修改setAddress("redis://192.168.57.111:6379")中的地址为自己的redis地址

2.2、DistributedLock.java

java">import java.lang.annotation.ElementType;
import java.lang.annotation.Retention;
import java.lang.annotation.RetentionPolicy;
import java.lang.annotation.Target;@Target(ElementType.METHOD) // 作用于方法
@Retention(RetentionPolicy.RUNTIME) // 注解在运行时生效
public @interface DistributedLock {String prefix() default "lock:"; // 锁前缀,默认为 "lock:"String key() default ""; // 锁的Key,支持SpEL表达式long leaseTime() default 30; // 锁的默认持有时间,单位秒long waitTime() default 10; // 获取锁的等待时间,单位秒
}

2.3、DistributedLockAspect.java

java">import org.aspectj.lang.ProceedingJoinPoint;
import org.aspectj.lang.annotation.Around;
import org.aspectj.lang.annotation.Aspect;
import org.aspectj.lang.annotation.Pointcut;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;
import org.redisson.api.RLock;
import org.redisson.api.RedissonClient;
import java.security.MessageDigest;
import java.security.NoSuchAlgorithmException;
import java.util.concurrent.TimeUnit;@Aspect
@Component
public class DistributedLockAspect {@Autowiredprivate RedissonClient redissonClient; // 注入 Redisson 客户端/*** 定义切入点,匹配所有使用 @DistributedLock 注解的方法* @param distributedLock 分布式锁注解对象*/@Pointcut("@annotation(distributedLock)")public void distributedLockPointcut(DistributedLock distributedLock) {}/*** 环绕通知:在目标方法执行前后处理分布式锁逻辑** @param joinPoint 切点,表示目标方法的执行点* @param distributedLock 注解,用于获取注解属性值* @return 目标方法的返回值* @throws Throwable 当目标方法抛出异常时向上抛出*/@Around("distributedLockPointcut(distributedLock)")public Object around(ProceedingJoinPoint joinPoint, DistributedLock distributedLock) throws Throwable {String lockKey = buildLockKey(distributedLock.prefix(),distributedLock.key(), joinPoint); // 生成锁KeyRLock lock = redissonClient.getLock(lockKey);boolean isLocked = false;try {while (true) {// 尝试获取锁isLocked = lock.tryLock(distributedLock.waitTime(), distributedLock.leaseTime(), TimeUnit.SECONDS);if (isLocked) {System.out.println(Thread.currentThread().getName() + " 成功获取锁,key: " + lockKey);return joinPoint.proceed(); // 执行目标方法} else {System.out.println(Thread.currentThread().getName() + " 未获取到锁,key: " + lockKey + ",重试中...");// 等待一段时间后再重试Thread.sleep(500);}}} finally {if (isLocked && lock.isHeldByCurrentThread()) {lock.unlock(); // 释放锁System.out.println(Thread.currentThread().getName() + " 已释放锁,key: " + lockKey);}}}private String buildLockKey(String prefix, String key, ProceedingJoinPoint joinPoint) {if (key.isEmpty()) {throw new IllegalArgumentException("Lock key cannot be empty");}// 拼接锁前缀和原始键String rawKey = prefix + key;// 将拼接后的键通过 MD5 生成唯一的锁键return generateMD5(rawKey);}/*** 使用 MD5 生成锁键* @param input 原始键* @return MD5 生成的锁键*/private String generateMD5(String input) {try {MessageDigest md = MessageDigest.getInstance("MD5");byte[] digest = md.digest(input.getBytes());// 转换为 32 位十六进制字符串StringBuilder hexString = new StringBuilder();for (byte b : digest) {String hex = Integer.toHexString(0xff & b);if (hex.length() == 1) {hexString.append('0');}hexString.append(hex);}return hexString.toString();} catch (NoSuchAlgorithmException e) {throw new RuntimeException("Failed to generate MD5 hash", e);}}}

这三步做完之后,该注解就能用了。

3、业务模拟测试

根据以下创建文件,并写入代码

3.1、InventoryService.java

这里测试可以itemId为键加锁

java">import com.pshao.charplatform.utils.distributedLock.DistributedLock;
import org.springframework.stereotype.Service;@Service
public class InventoryService {@DistributedLock(prefix = "stock",key = "#itemId", leaseTime = 10, waitTime = 1)public void reduceStock(Long itemId, int quantity) {System.out.println(Thread.currentThread().getName() + " 正在处理库存扣减: itemId=" + itemId + ", quantity=" + quantity);try {Thread.sleep(2000); // 模拟耗时操作,修改为两秒} catch (InterruptedException e) {e.printStackTrace();}System.out.println(Thread.currentThread().getName() + " 完成库存扣减: itemId=" + itemId);}}

这里可以输入以下多个参数,key是必须的,其他可以不输入

String prefix() default "lock:"; // 锁前缀,默认为 "lock:"
String key() default ""; // 锁的Key
long leaseTime() default 30; // 锁的默认持有时间,单位秒
long waitTime() default 10; // 获取锁的等待时间,单位秒

3.2、DistributedLockApplication.java

java">import org.springframework.boot.SpringApplication;
import org.springframework.boot.autoconfigure.SpringBootApplication;
import org.springframework.context.ApplicationContext;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;@SpringBootApplication(scanBasePackages = "com.pshao.charplatform")
public class DistributedLockApplication {public static void main(String[] args) {ApplicationContext context = SpringApplication.run(DistributedLockApplication.class, args);InventoryService inventoryService = context.getBean(InventoryService.class);ExecutorService executor = Executors.newFixedThreadPool(3); // 创建3个线程// 模拟多个线程竞争同一资源for (int i = 0; i < 3; i++) {executor.submit(() -> inventoryService.reduceStock(123L, 10));}executor.shutdown();}
}

3.3、运行查看测试结果

测试成功! 

如果运行后出现,类似以下日志

Action: Correct the classpath of your application so that it contains compatible versions of the classes org.springframework.aop.framework.autoproxy.AbstractAutoProxyCreator and org.springframework.util.ClassUtils

考虑是版本冲突,可以参考Correct the classpath of your application so that it contains compatible versions......版本不兼容解决方法-CSDN博客


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

相关文章

mysql的my.cnf配置文件参数说明

mysql的my.cnf配置文件参数说明 mysql的my.cnf配置文件参数说明&#xff0c;对于my.cnf的相关详细配置的参数说明和参数的常规配置 [mysqld] # ------------主要配置----------------- # 端口 port 3306# 数据地址 datadir/var/lib/mysql socket/var/lib/mysql/mysql.sock# …

【Python · PyTorch】卷积神经网络 CNN(LeNet-5网络)

【Python PyTorch】卷积神经网络 CNN&#xff08;LeNet-5网络&#xff09; 1. LeNet-5网络※ LeNet-5网络结构 2. 读取数据2.1 Torchvision读取数据2.2 MNIST & FashionMNIST 下载解包读取数据 2. Mnist※ 训练 LeNet5 预测分类 3. EMnist※ 训练 LeNet5 预测分类 4. Fash…

一文了解js 的正则

文章目录 正则基础正则应用正则性能优化特殊含义?.? 正则基础 一、正则表达式基础知识 什么是正则表达式&#xff1f; 正则表达式是一种用于匹配字符串中字符组合的模式。在JavaScript中&#xff0c;正则表达式是对象。它就像一个模板&#xff0c;可以帮助你在文本中查找、替…

Python设计模式详解之5 —— 原型模式

Prototype 设计模式是一种创建型设计模式&#xff0c;它通过复制已有的实例来创建新对象&#xff0c;而不是通过从头实例化。这种模式非常适合对象的创建成本较高或者需要避免复杂的构造过程时使用。Prototype 模式提供了一种通过克隆来快速创建对象的方式。 1. Prototype 模式…

java 可以跨平台的原因是什么?

我们对比一个东西就可以了&#xff0c;那就是chrome浏览器。 MacOS/Linux/Windows上的Chrome浏览器&#xff0c;那么对于HTML/CSS/JS的渲染效果都一样的。 我们就可以认为ChromeHTML/CSS/JS是跨平台的。 这里面&#xff0c;HTML/CSS/JS是不变的的&#xff0c;对于一个网页&a…

OSG开发笔记(三十三):同时观察物体不同角度的多视图从相机技术

​若该文为原创文章&#xff0c;未经允许不得转载 本文章博客地址&#xff1a;https://blog.csdn.net/qq21497936/article/details/143932273 各位读者&#xff0c;知识无穷而人力有穷&#xff0c;要么改需求&#xff0c;要么找专业人士&#xff0c;要么自己研究 长沙红胖子Qt…

SAM阅读

文章内容&#xff1a; 介绍了 Segment Anything &#xff08;SA&#xff09; 项目&#xff1a;用于图像分割的新任务、模型和数据集。 我们构建了迄今为止&#xff08;迄今为止&#xff09;最大的分割数据集&#xff0c;在 11M 许可和尊重隐私的图像上拥有超过 10 亿个掩码。该…

038集——quadtree(CAD—C#二次开发入门)

效果如下&#xff1a; using Autodesk.AutoCAD.ApplicationServices; using Autodesk.AutoCAD.DatabaseServices; using Autodesk.AutoCAD.EditorInput; using Autodesk.AutoCAD.Geometry; using System; using System.Collections.Generic; using System.Linq; using System.T…