【Redis】Redis 分布式锁

2023-10-26 14:53:24 浏览数 (2)

一、分布式锁概念

随着业务发展的需要，原单机部署的系统被演化成分布式集群系统后，由于分布式系统多线程、多进程并且分布在不同机器上，这将使原单机部署情况下的并发控制锁策略失效，单纯的 Java API 并不能提供分布式锁的能力。为了解决这个问题就需要一种跨 JVM 的互斥机制来控制共享资源的访问，这就是分布式锁要解决的问题！

说得通俗些，集群中上了锁后，无论当前操作在哪台机器，所有的机器都会识别并且等待，锁释放后其他操作才能进行，这就是分布式锁，对所有集群里都有效

分布式锁主流的实现方案：

  1. 基于数据库实现分布式锁
  2. 基于缓存（Redis 等）
  3. 基于 Zookeeper

每一种分布式锁解决方案都有各自的优缺点，其中redis性能最高zookeeper可靠性最高

二、使用setnx实现锁

代码语言：javascript复制
set stu:1:info “OK” nx px 10000
  • EX second ：设置键的过期时间为 second 秒,，SET key value EX second 效果等同于 SETEX key second value
  • PX millisecond ：设置键的过期时间为 millisecond 毫秒，SET key value PX millisecond 效果等同于 PSETEX key millisecond value
  • NX ：只在键不存在时，才对键进行设置操作，SET key value NX 效果等同于 SETNX key value
  • XX ：只在键已经存在时，才对键进行设置操作
  • 多个客户端同时获取锁（setnx）
  • 获取成功，执行业务逻辑（从 db 获取数据，放入缓存），执行完成释放锁（del）
  • 获取失败的客户端则等待重试

用setnx和del添加以及释放锁

一般地，我们需要给锁设置过期时间防止锁被长期占用

这里有个问题：加锁和设置过期时间是两个操作，而不是同时进行操作的，如果上锁后发生异常情况，就无法设置过期时间了。我们可以上锁的同时设置过期时间

三、编写代码测试分布式锁

1. 使用Java代码测试分布式锁

首先在redis中设置num的值为0，编写Java代码进行测试

下方代码做的就是：获取到锁则num ，并释放锁；没获取到则0.1秒后重新获取

重启，服务集群，通过网关压力测试：ab -n 5000 -c 100 http://192.168.140.1:8080/test/testLock

查看 redis 中 num 的值

问题：  setnx 刚好获取到锁，业务逻辑出现异常，导致锁无法释放 解决：  设置过期时间，自动释放锁

2. 优化之设置锁的过期时间

设置过期时间有两种方式：

  • 首先想到通过 expire 设置过期时间（缺乏原子性：如果在 setnx 和 expire 之 间出现异常，锁也无法释放）
  • 在 set 的同时指定过期时间（推荐）

代码中设置过期时间：

问题：  可能会释放其他服务器的锁

如果业务逻辑的执行时间是 7s，执行流程如下：

  • index1 业务逻辑没执行完，3 秒后锁被自动释放
  • index2 获取到锁，执行业务逻辑，3 秒后锁被自动释放
  • index3 获取到锁，执行业务逻辑
  • index1 业务逻辑执行完成，开始调用 del 释放锁，这时释放的是 index3 的锁， 导致 index3 的业务只执行 1s 就被别人释放。 最终等于没锁的情况

a在操作时卡顿了，导致锁超时后自动释放；释放后，b抢到锁进行操作；此时a操作完成，手动释放锁，这就把b的锁给释放了，b再释放锁则会报错

解决：  setnx 获取锁时，设置一个指定的唯一值（例如：uuid）；释放前获取这 个值，判断是否自己的锁

四、优化之给lock设置UUID防误删

五、使用LUA脚本保证删除的原子性

使用lock的uuid可以一定程度上缓解线程释放其他锁，但并不能完全解决这种情况。因为比较uuid和删除lock并不是原子性的

问题：  a比较uuid通过后，锁到期了自动释放，b重新加锁，a此时会手动释放b的锁，这还是出现问题

解决：  使用LUA 脚本保证删除的原子性

LUA脚本：

  • 将复杂的或者多步的 redis 操作，写为一个脚本，一次提交给 redis 执行，减少反复连接 redis 的次数，提升性能
  • LUA 脚本是类似 redis 事务，有一定的原子性，不会被其他命令插队，可以完成一些redis 事务性的
代码语言：javascript复制
@GetMapping("testLockLua")
public void testLockLua() {
    //1 声明一个 uuid ,将做为一个 value 放入我们的 key 所对应的值中
    String uuid = UUID.randomUUID().toString();
    //2 定义一个锁：lua 脚本可以使用同一把锁，来实现删除！
    String skuId = "25"; // 访问 skuId 为 25 号的商品 100008348542
    String locKey = "lock:"   skuId; // 锁住的是每个商品的数据
    // 3 获取锁
    Boolean lock = redisTemplate.opsForValue().setIfAbsent(locKey, uuid, 3, TimeUnit.SECONDS);
    // 第一种： lock 与过期时间中间不写任何的代码。
    // redisTemplate.expire("lock",10, TimeUnit.SECONDS);//设置过期时间
    // 如果 true
    if (lock) {
        // 执行的业务逻辑开始
        // 获取缓存中的 num 数据
        Object value = redisTemplate.opsForValue().get("num");
        // 如果是空直接返回
        if (StringUtils.isEmpty(value)) {
            return;
        }
        // 不是空 如果说在这出现了异常！ 那么 delete 就删除失败！ 也就是说锁永远存在！
        int num = Integer.parseInt(value   "");
        // 使 num 每次 1 放入缓存
        redisTemplate.opsForValue().set("num", String.valueOf(  num));
        /*使用 lua 脚本来锁*/
        // 定义 lua 脚本：将判断和删除操作同时进行
        String script = "if redis.call('get', KEYS[1]) == ARGV[1] then return redis.call('del', KEYS[1]) else return 0 end";
        // 使用 redis 执行 lua 执行
        DefaultRedisScript<Long> redisScript = new DefaultRedisScript<>();
        redisScript.setScriptText(script);
        // 设置一下返回值类型 为 Long
        // 因为删除判断的时候，返回的 0,给其封装为数据类型。如果不封装那么默认返回 String 类型，
        // 那么返回字符串与 0 会有发生错误。
        redisScript.setResultType(Long.class);
        // 第一个是执行的 script 脚本 ，第二个需要判断的 key，第三个就是 key 所对应的值。
        redisTemplate.execute(redisScript, Arrays.asList(locKey), uuid);
        } else {
        // 其他线程等待
        try {
            // 睡眠
            Thread.sleep(1000);
            // 睡醒了之后，调用方法。
            testLockLua();
        } catch (InterruptedException e) {
            e.printStackTrace();
        }
    }
}

为了确保分布式锁可用，我们至少要确保锁的实现同时满足以下四个条件：

  • 互斥性；在任意时刻，只有一个客户端能持有锁
  • 不会发生死锁；即使有一个客户端在持有锁的期间崩溃而没有主动解锁，也能保证后续其他客户端能加锁（设置lock的过期时间）
  • 解铃还须系铃人；加锁和解锁必须是同一个客户端，客户端自己不能把别人加的锁给解了（使用LUA脚本和uuid）
  • 加锁和解锁必须具有原子性（使用LUA脚本）
redis 分布式锁 集群 脚本 客户端

0 人点赞