redis分布式锁+自定义注解+锁续期+动态获取锁名称

研究一周终于出成果啦!
注解

package com.it.annotation;

import java.lang.annotation.*;

@Target({ElementType.TYPE, ElementType.METHOD})
@Retention(RetentionPolicy.RUNTIME)
@Documented
@Inherited
public @interface JwLock {
    /**
     * 锁的名称
     */
    String lockName() default "jwLock";

    /**
     * 锁的后缀 动态锁名称
     * 适用能直接获取的参数 或者对象属性
     */
    String lockSuffix() default "";

    /**
     * 参数索引位置 第一位0
     */
    int lockIndex() default 0;

    /**
     * 锁的有效时间
     */
    long expireTime() default -1;

}

AOP 动态获取锁名称 也可以固定锁名称

package com.it.config;

import cn.hutool.core.util.StrUtil;
import cn.hutool.json.JSONObject;
import cn.hutool.json.JSONUtil;
import com.it.annotation.JwLock;
import com.it.util.RedisUtil;
import lombok.extern.slf4j.Slf4j;
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.aspectj.lang.reflect.MethodSignature;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;

import java.lang.reflect.Method;
import java.util.concurrent.TimeUnit;

/**
 * @description: 分布式互斥锁
 */
@Slf4j
@Aspect
@Component
public class JwLockAspect {

    @Autowired
    RedisUtil redisUtil;

    @Pointcut("@annotation(com.it.annotation.JwLock)")
    public void lockPointCut() {
    }

    @Around("lockPointCut()")
    public void around(ProceedingJoinPoint joinPoint) {
        Object[] args = joinPoint.getArgs();
        MethodSignature methodSignature = (MethodSignature) joinPoint.getSignature();
        Method method = methodSignature.getMethod();
        JwLock redisLock = method.getAnnotation(JwLock.class);

        //获取锁名称
        String lockName = redisLock.lockName();
        //获取超时时间
        long expireTime = redisLock.expireTime();
        String lockSuffix = redisLock.lockSuffix();
        int lockIndex = redisLock.lockIndex();

        if (args.length > 0 && StrUtil.isNotBlank(lockSuffix)) {
            if (lockIndex < 0 || lockIndex >= args.length) {
                throw new RuntimeException("lockIndex不能小于0或者大于等于参数个数");
            }
            Object arg = args[lockIndex];
            String suffix;
            //拼接锁后缀
            if (arg instanceof Integer || arg instanceof String || arg instanceof Long) {
                suffix = String.valueOf(arg);
            } else {
                JSONObject jsonObject = JSONUtil.parseObj(JSONUtil.toJsonStr(arg));
                suffix = String.valueOf(jsonObject.get(lockSuffix));
            }
            lockName = lockName + suffix;
        }

        boolean isLock = redisUtil.tryLock(lockName, expireTime);
        try {
            if (isLock) {
                log.info(Thread.currentThread().getName() + ",锁名称" + lockName + "加锁成功,开始执行业务...");
                joinPoint.proceed();
                log.info(Thread.currentThread().getName() + ",锁名称" + lockName + "业务执行完成...");
            } else {
                log.info(Thread.currentThread().getName() + ",锁名称" + lockName + "尝试获取锁失败, 稍后重试!");
                throw new RuntimeException("咦 ,姿势不对 ,手速慢了!");
            }
        } catch (Throwable throwable) {
            log.info(Thread.currentThread().getName() + ",锁名称" + lockName + "加锁失败", throwable);
        } finally {
            //如果该线程还持有该锁,那么释放该锁。如果该线程不持有该锁,说明该线程的锁已到过期时间,自动释放锁
            redisUtil.deleteLock(lockName);
        }
    }
}
package com.it.util;

import cn.hutool.core.collection.CollUtil;
import cn.hutool.core.util.IdUtil;
import cn.hutool.core.util.StrUtil;
import lombok.Data;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.data.redis.core.StringRedisTemplate;
import org.springframework.data.redis.core.script.DefaultRedisScript;
import org.springframework.stereotype.Component;

import java.util.*;
import java.util.concurrent.TimeUnit;

@Slf4j
@Data
@Component
public class RedisUtil {

    @Autowired
    RedisTemplate<String, Object> redisTemplate;

    @Autowired
    StringRedisTemplate stringRedisTemplate;

    ThreadLocal<String> threadLocal = new ThreadLocal<>();
    /**
     * lua 脚本
     */
    public static final String Script = "if redis.call('get', KEYS[1]) == ARGV[1] then return redis.call('del', KEYS[1]) else return 0 end";

    /**
     * 分布式锁 timeout为-1时锁自动续期
     *
     * @param lockName 锁名称
     * @return Boolean
     */
    public Boolean setNx(String lockName) {
        return setNx(lockName, -1L);
    }

    /**
     * 分布式锁 自定义过期时间
     *
     * @param lockName 锁名称
     * @return Boolean
     */
    public Boolean setNx(String lockName, long timeout) {
        String uuid = IdUtil.simpleUUID();
        if (timeout == -1) {
            timeout = 30L;
            Boolean isLock = setNx(lockName, timeout);
            if (isLock) {
                threadLocal.set(uuid);
                //锁续期
                renewExpireTime(lockName, uuid, timeout);
            }
            return isLock;
        } else {
            timeout = timeout <= 0 ? 120 : timeout;
            Boolean isLock = stringRedisTemplate.opsForValue().setIfAbsent(lockName, uuid, timeout, TimeUnit.SECONDS);
            if (Boolean.TRUE.equals(isLock)) {
                threadLocal.set(uuid);
                log.info(Thread.currentThread().getName() + "获取锁{}成功,UUID:{}", lockName, uuid);
            }
            return isLock;
        }
    }

    private void renewExpireTime(String lockName, String uuid, long timeout) {
        //自动续期脚本
        String script = " if redis.call('exists',KEYS[1],ARGV[1]) == 1 then " +
                " return redis.call('expire',KEYS[1],ARGV[2]) " +
                " else" +
                " return 0 " +
                " end ";
        log.info(Thread.currentThread().getName() + "锁名称:{},renewExpireTime.uuid:{}", lockName, uuid);
        new Timer().schedule(new TimerTask() {
            @Override
            public void run() {
                if (Boolean.TRUE.equals(stringRedisTemplate.execute(new DefaultRedisScript<>(script, Boolean.class), Collections.singletonList(lockName), uuid, String.valueOf(timeout)))) {
                    Long expire = stringRedisTemplate.getExpire(lockName, TimeUnit.SECONDS);
                    log.info(Thread.currentThread().getName() + "锁名称:{}续命成功,还有{}秒过期", lockName, expire);
                    renewExpireTime(lockName, uuid, timeout);
                }
                ;
            }
        }, (timeout * 1000) / 3);

    }

    /**
     * 释放分布式锁
     *
     * @param lockName 锁名称
     */
    public void deleteLock(String lockName) {
        String uuid = threadLocal.get();
        if (StrUtil.isNotEmpty(uuid)) {
            log.info(Thread.currentThread().getName() + ",锁名称:{},uuid:{}", lockName, uuid);
            String curLock = stringRedisTemplate.opsForValue().get(lockName);
            if (StrUtil.isNotEmpty(curLock) && uuid.equals(curLock)) {
                redisTemplate.delete(lockName);
                redisTemplate.execute(new DefaultRedisScript<>(Script, Long.TYPE), Collections.singletonList(lockName), uuid);
                threadLocal.remove();
                log.info(Thread.currentThread().getName() + ",锁名称" + lockName + "释放锁成功...");
            }
        }
    }

    /**
     * 分布式锁
     *
     * @param lockName 锁名称
     * @return uuid
     */
    public String setNxStr(String lockName, int timeout) {
        String uuid = UUID.randomUUID().toString().replace("-", "");
        timeout = timeout <= 0 ? 30 : timeout;
        Boolean flag = stringRedisTemplate.opsForValue().setIfAbsent(lockName, uuid, timeout, TimeUnit.SECONDS);
        if (Boolean.TRUE.equals(flag)) {
            log.info(Thread.currentThread().getName() + "获取锁{}成功,UUID:{}", lockName, uuid);
            return uuid;
        }
        return null;
    }

    /**
     * 释放分布式锁
     *
     * @param lockMap 锁名称:uuid
     */
    public void deleteLock(Map<String, Object> lockMap) {
        if (CollUtil.isNotEmpty(lockMap)) {
            lockMap.forEach((key, uuid) -> {
                log.info(Thread.currentThread().getName() + ",锁名称:{},uuid:{}", key, uuid);
                String curLock = stringRedisTemplate.opsForValue().get(key);
                if (StrUtil.isNotEmpty(curLock) && uuid.equals(curLock)) {
                    redisTemplate.delete(key);
                    redisTemplate.execute(new DefaultRedisScript<Long>(Script, Long.TYPE), Collections.singletonList(key), uuid);
                    log.info(Thread.currentThread().getName() + ",锁名称" + key + "释放锁成功...");
                }
            });
        }
    }
}


pom

<?xml version="1.0" encoding="UTF-8"?>
<project xmlns="http://maven.apache.org/POM/4.0.0"
         xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
         xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd">
    <parent>
        <artifactId>item-master</artifactId>
        <groupId>org.example</groupId>
        <version>1.0-SNAPSHOT</version>
    </parent>
    <modelVersion>4.0.0</modelVersion>

    <artifactId>redis-lock</artifactId>

    <properties>
        <maven.compiler.source>8</maven.compiler.source>
        <maven.compiler.target>8</maven.compiler.target>
    </properties>

    <dependencies>
        <dependency>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-starter</artifactId>
        </dependency>
        <dependency>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-starter-web</artifactId>
        </dependency>
        <dependency>
            <groupId>org.projectlombok</groupId>
            <artifactId>lombok</artifactId>
        </dependency>
        <dependency>
            <groupId>org.redisson</groupId>
            <artifactId>redisson</artifactId>
            <version>3.19.3</version>
        </dependency>
        <dependency>
            <groupId>com.alibaba</groupId>
            <artifactId>fastjson</artifactId>
        </dependency>
        <!-- Redis依赖 -->
        <dependency>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-starter-data-redis</artifactId>
            <version>3.0.4</version>
        </dependency>

        <dependency>
            <groupId>cn.hutool</groupId>
            <artifactId>hutool-all</artifactId>
        </dependency>
        <dependency>
            <groupId>org.aspectj</groupId>
            <artifactId>aspectjweaver</artifactId>
            <version>1.9.7</version>
        </dependency>

    </dependencies>

</project>

演示
Controller

@GetMapping("/tryLock2")
    public String tryLock2() {
        boolean isLock = redisUtil.setNx(maoTai,30);
        try {
            if (isLock) {
                log.info(Thread.currentThread().getName() + "获取锁 开始执行");
                Thread.sleep(3000);
                log.info(Thread.currentThread().getName() + "执行结束");
            } else {
                log.info(Thread.currentThread().getName() + "获取锁失败");
                return "fail";
            }
        } catch (InterruptedException e) {
            log.error("获取锁失败" + e.getMessage());
        } finally {
            redisUtil.deleteLock(maoTai);
        }
        return "success";
    }


    @GetMapping("/tryLock3")
    public String tryLock3() {
        log.info(Thread.currentThread().getName() + "来了...");
        testService.tryLock3();
        return "success";
    }

    @PostMapping("/tryLock4")
    public String tryLock4(@RequestBody TaskDTO taskDTO) {
        testService.tryLock4(taskDTO);
        return "success";
    }

    @GetMapping("/tryLock5")
    public String tryLock5() {
        Boolean isLock = redisUtil.tryLock(maoTai, -1);
        try {
            if (isLock) {
                log.info(Thread.currentThread().getName() + "获取锁 开始执行");
                Thread.sleep(25000);
                log.info(Thread.currentThread().getName() + "执行结束");
            } else {
                log.info(Thread.currentThread().getName() + "获取锁失败");
                return "fail";
            }
        } catch (InterruptedException e) {
            log.error(e.getMessage());
        } finally {
            redisUtil.deleteLock(maoTai);
        }
        return "success";
    }

    @GetMapping("/tryLock6")
    public String tryLock6() {
        testService.tryLock6();
        return "success";
    }

service

package com.it.service;

import com.it.annotation.JwLock;
import com.it.annotation.RedisLock;
import com.it.entity.TaskDTO;
import lombok.extern.slf4j.Slf4j;
import org.springframework.context.annotation.EnableAspectJAutoProxy;
import org.springframework.stereotype.Service;

@Slf4j
@Service
public class TestService {

    @RedisLock(lockName = "redis-lock")
    public void tryLock3() {
        log.info(Thread.currentThread().getName() + "获取锁 开始执行");
        try {
            Thread.sleep(3000);
        } catch (InterruptedException e) {
            log.info(e.getMessage());
        }
        log.info(Thread.currentThread().getName() + "执行结束");

    }
	/**
	*动态获取锁名称
	*/
    @JwLock(lockName = "JwLock",lockSuffix = "dataPart")
    public void tryLock4(TaskDTO taskDTO) {
        log.info(Thread.currentThread().getName() + "获取锁 开始执行");
        try {
            Thread.sleep(3000);
        } catch (InterruptedException e) {
            log.info(e.getMessage());
        }
        log.info(Thread.currentThread().getName() + "执行结束");
    }

    @JwLock(lockName = "JwLock")
    public void tryLock6() {
        try {
            Thread.sleep(35000);
        } catch (InterruptedException e) {
            log.info(e.getMessage());
        }
    }
}

package com.it.entity;

import lombok.Data;

@Data
public class TaskDTO {
    String taskId;
    Integer dataPart;
}

用jmeter测试 5个线程同时调用/tryLock4
{
“taskId”:“1”,
“dataPart”:“20230501”
}

动态获取锁名称演示
在这里插入图片描述

用jmeter测试 5个线程同时调用/tryLock6
锁续命演示
在这里插入图片描述


  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论
下面是一个使用Lua脚本实现Redis分布式锁的代码示例: ```lua -- Lua脚本实现Redis分布式锁 local lockKey = 'lock' local uuid = ARGV\[1\] if redis.call('get', lockKey) == uuid then redis.call('del', lockKey) return 1 else return 0 end ``` 这段代码首先定义了一个的键名为`lockKey`,然后通过传入的参数`ARGV\[1\]`获取到要删除的的UUID。接下来,它会通过`redis.call('get', lockKey)`来获取当前的值,如果与传入的UUID相等,则说明当前是由该UUID持有的,此时会使用`redis.call('del', lockKey)`来删除,并返回1表示删除成功。如果的值与传入的UUID不相等,则说明当前不是由该UUID持有的,此时直接返回0表示删除失败。 这段代码可以用于实现Redis分布式锁的原子性删除操作,确保只有持有的客户端才能删除,避免误删的问题。同时,使用Lua脚本可以保证删除的操作是原子性的,避免并发情况下的竞争问题。 #### 引用[.reference_title] - *1* *2* [Redis 实现分布式锁+执行lua脚本](https://blog.csdn.net/qq_34285557/article/details/129700808)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v91^control_2,239^v3^insert_chatgpt"}} ] [.reference_item] - *3* [Redis分布式锁问题(九)Redis + Lua 脚本实现分布式锁](https://blog.csdn.net/weixin_43715214/article/details/127982757)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v91^control_2,239^v3^insert_chatgpt"}} ] [.reference_item] [ .reference_list ]

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

橘右今

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值