当前位置: 首页 > article >正文

redisson 使用与分析

一、前言

        上一期小永哥简单介绍了基于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 {

    @Bean
    public 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分布式锁解决方案来解决并发问题。

    @Test
    public 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;"

五、结语

        这次分享源码对小永哥来说挑战非常大,我也不确定我是否说清楚了这个事情,大家感兴趣可以也尝试着打断点去跟一下源码,由于能力有限,有不完美的地方还望大家多多指教,希望可以和小伙伴们共同进步


http://www.kler.cn/a/598644.html

相关文章:

  • 【微前端】使用微前端解决碎片化的用户界面问题实践
  • Android开发layer-list
  • Blender导出fbx到Unity找不到贴图的问题
  • Linux NFS、自动挂载与系统启动管理指南
  • 【深度学习】【目标检测】【Ultralytics-YOLO系列】Windows11下YOLOV3人脸检测
  • spring-security原理与应用系列:建造者
  • vue3之写一个aichat---实现聊天逻辑
  • OpenRAND可重复的随机数生成库
  • git 合并多次提交 commit
  • 【xiaozhi赎回之路-2:语音可以自己配置就是用GPT本地API】
  • display: contens的使用
  • python爬虫Redis数据库
  • 【MyDB】5-索引管理之4-单元测试
  • Jupyter Notebook 常用命令(自用)
  • AI安全学习(刚开始,未完版)
  • git 命令回退版本
  • Kafka 八股文
  • python 游戏开发cocos2d库安装与使用
  • 【推荐项目】056-高校学籍管理系统
  • 回归——数学公式推导全过程