郑州做网站推北京公司注册地址要求
一、前言
上一期小永哥简单介绍了基于redis实现自定义分布式锁,通过三步优化将漏洞完善。本期要分享的是一个基于redis的分布式锁的成熟框架-redisson,该框架简单易用,功能强大,对于上一期我们担心的问题,redisson框架内部也都已实现,下面简单介绍一下redisson的使用
二、redisson配置
package com.relation.web.core.config;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;/*** @author huhy* @version 1.0* @Description:* @ClassName Date:2025/3/9 22:31*/
@Configuration
public class RedissonConfig {@Beanpublic RedissonClient redissonClient() {Config config = new Config();config.useSingleServer().setAddress("redis://127.0.0.1:6379").setRetryInterval(5000).setTimeout(10000).setConnectTimeout(10000);return Redisson.create(config);}
}
三、redisson基本使用
本次我们使用redisson来改造一下上一期测试代码,用redisson分布式锁解决方案来解决并发问题。
@Testpublic void redissonTest() throws Exception{CountDownLatch countDownLatch = new CountDownLatch(10);for (int i = 0; i < 10; i++) {new Thread(()->{String uuid = UUID.randomUUID().toString().replaceAll("-","");String key = "test:createCode";//执行业务逻辑之前加锁RLock lock = redissonClient.getLock(key);try {//假设业务需要执行7秒,已经超过了设置的超时时间TimeUnit.SECONDS.sleep(8);}catch (Exception e){}try {lock.lock();createCodeTest(1L);}finally {//判断当前要释放的锁是否是当前线程加的锁if(lock.isHeldByCurrentThread()){//业务逻辑执行后解锁lock.unlock();}}countDownLatch.countDown();},"测试线程"+i).start();}countDownLatch.await();}/*** 功能描述:测试生成编码* @param: 编码规则id* @Author:huhy* @Date: 2025/3/11 23:03*/private void createCodeTest(Long id){//通过编码id获取编码规则TSCodeRule tsCodeRule = codeRuleService.selectTSCodeRuleById(id);if(ObjectUtil.isEmpty(tsCodeRule)){System.out.println(id+":编码规则不存在");}//获取最新的流水码值String serialCode = "";List<TSSerialCode> tsSerialCodes = serialCodeService.selectTSSerialCodeList(null);if(tsSerialCodes.isEmpty()){serialCode = String.valueOf(1);}else {tsSerialCodes.sort((t1,t2)->{Integer code1 = Integer.parseInt(t1.getSerialCode());Integer code2 = Integer.parseInt(t2.getSerialCode());return code2.compareTo(code1);});serialCode = tsSerialCodes.get(0).getSerialCode();}//获取当前线程名称String threadName = Thread.currentThread().getName();//计算出本次需要的流水码Integer newSerialCode = Integer.parseInt(serialCode)+tsCodeRule.getStepLength();//生成业务编码String codeResult = tsCodeRule.getCodePrefix()+newSerialCode;System.out.println(threadName+" 获取编码:"+codeResult);//将最新的流水码保存TSSerialCode tsSerialCode = new TSSerialCode();tsSerialCode.setSerialCode(newSerialCode+"");tsSerialCode.setCreateTime(new Date());serialCodeService.insertTSSerialCode(tsSerialCode);}
运行测试代码,在并发场景下编码生成正确,说明redisson分布式锁框架简单、易用、可靠。
四、redisson核心实现简介
4.1、我们在锁创建、加锁、解锁处打上断点,通过断点来窥探一下redisson的实现原理。
进入redissonClient的getLock方法,该方法调用了RedissonLock的构造方法,构造方法中对RedissonLock的相关信息进行一些初始化操作,比如说commandExecutor(执行器)等信息的初始化,此步骤只是做一些初始化的事情,并没有真正开始加锁。
4.2、通过断点一步一步往下走,我们主要是找核心的加锁逻辑,就是最后截图中的tryLockInnerAsync方法,该方法的核心逻辑其实也是通过redis命令进行加锁,只不过它用的是lua脚本进行的加锁。
我们把lua脚本复制出来看一下。大概意思就是先判断是否存在,如果不存在就进行创建,并且设置过期时间,如果已存在,判断是否是当前线程自己的锁,如果是的话,则进行一个累加操作,类似于锁的重入。
//如果key不存在
if (redis.call('exists', KEYS[1]) == 0) then //创建元素redis.call('hset', KEYS[1], ARGV[2], 1); //设置过期时间redis.call('pexpire', KEYS[1], ARGV[1]);return nil; end;
//如果key存在并且是自己加的锁(也就是锁重入)
if (redis.call('hexists', KEYS[1], ARGV[2]) == 1) then //则给原来的可以进行累加redis.call('hincrby', KEYS[1], ARGV[2], 1); redis.call('pexpire', KEYS[1], ARGV[1]); return nil; end;
return redis.call('pttl', KEYS[1]);
如果加锁成功,那么另起线程进行锁的续期操作,具体是每隔过期时间的三分之一进行续期,就是说如果设置的过期时间的三分之一时间内没有执行完业务,那么就再进行续期操作。续期的核心操作也是通过lua脚本来执行的。
//如果key存在
"if (redis.call('hexists', KEYS[1], ARGV[2]) == 1) then //将key再设置一遍过期时间redis.call('pexpire', KEYS[1], ARGV[1]); //返回成功标识return 1;//结束end;
//如果key不存在,则返回失败标识,则续期线程退出
return 0;"
4.3、解锁时,我们还是一步一步去找核心方法unlockInnerAsync,在这个方法内,核心的其实还是lua脚本,通过lua脚本对锁进行释放。释放完锁以后,最后再关闭续期线程,这样完成了一个完整的流程。
"if (redis.call('hexists', KEYS[1], ARGV[3]) == 0) then return nil;end; local counter = redis.call('hincrby', KEYS[1], ARGV[3], -1);
if (counter > 0) then redis.call('pexpire', KEYS[1], ARGV[2]); return 0; else redis.call('del', KEYS[1]); redis.call('publish', KEYS[2], ARGV[1]); return 1; end;
return nil;"
五、结语
这次分享源码对小永哥来说挑战非常大,我也不确定我是否说清楚了这个事情,大家感兴趣可以也尝试着打断点去跟一下源码,由于能力有限,有不完美的地方还望大家多多指教,希望可以和小伙伴们共同进步