redis实现分布式锁(基础版)

使用redis实现分布式锁的方法有多种,基础版本是基于setnx命令,即如果不存在则设置。这个命令可以保证只有一个客户端能够成功设置一个key,从而获得锁。设置key的时候需要设置一个过期时间,以防止死锁。释放锁的时候需要删除key,或者使用lua脚本来保证原子性。

//导入jedis依赖
import redis.clients.jedis.Jedis;

//定义一个分布式锁的类
class RedisLock {
    //定义一个jedis对象,用于连接redis
    private Jedis jedis;
    //定义一个锁的key
    private String lockKey;
    //定义一个锁的过期时间,单位是毫秒
    private long expireTime;

    //构造方法,传入jedis对象,锁的key和过期时间
    public RedisLock(Jedis jedis, String lockKey, long expireTime) {
        this.jedis = jedis;
        this.lockKey = lockKey;
        this.expireTime = expireTime;
    }

    //尝试获取锁的方法,返回一个布尔值,表示是否成功获取锁
    public boolean tryLock() {
        //使用setnx命令,如果成功设置key,返回1,否则返回0
        long result = jedis.setnx(lockKey, "1");
        //如果返回1,表示获取锁成功
        if (result == 1) {
            //设置key的过期时间,防止死锁
            jedis.pexpire(lockKey, expireTime);
            //返回true
            return true;
        }
        //如果返回0,表示获取锁失败
        else {
            //返回false
            return false;
        }
    }

    //释放锁的方法
    public void unlock() {
        //删除key,释放锁
        jedis.del(lockKey);
    }
}

基础代码优缺点:

分析一下这个代码的优缺点。这个代码的优点是简单易懂,使用setnx命令可以保证锁的互斥性,使用过期时间可以防止死锁。这个代码的缺点是不够健壮,有以下几个问题:

  • 如果在设置key的过期时间之前,客户端崩溃或者网络中断,那么key可能永远不会过期,导致其他客户端无法获取锁。
  • 如果在释放锁之前,客户端崩溃或者网络中断,那么key可能没有被删除,导致其他客户端无法获取锁。
  • 如果在释放锁的时候,key已经过期,那么可能会误删其他客户端设置的key,导致锁的安全性被破坏。
  • 如果锁的过期时间太短,那么可能会导致客户端在执行任务的过程中,锁被其他客户端抢占,导致任务的一致性被破坏。
  • 如果锁的过期时间太长,那么可能会导致客户端在获取锁失败的情况下,等待的时间过长,导致性能下降。

redis实现分布式锁(进阶版)

为了解决这些问题,可以使用一些更复杂的逻辑,如使用lua脚本来保证设置key和过期时间的原子性,使用唯一的随机值来标识锁的持有者,使用续租机制来延长锁的过期时间等。

//导入jedis依赖
import redis.clients.jedis.Jedis;

//定义一个分布式锁的类
class RedisLock {
    //定义一个jedis对象,用于连接redis
    private Jedis jedis;
    //定义一个锁的key
    private String lockKey;
    //定义一个锁的过期时间,单位是毫秒
    private long expireTime;
    //定义一个锁的唯一值,用于标识锁的持有者
    private String lockValue;
    //定义一个续租线程,用于延长锁的过期时间
    private Thread renewThread;
    //定义一个lua脚本,用于原子性地设置key和过期时间
    private String setScript = "if redis.call('setnx', KEYS[1], ARGV[1]) == 1 then return redis.call('pexpire', KEYS[1], ARGV[2]) else return 0 end";
    //定义一个lua脚本,用于原子性地删除key
    private String delScript = "if redis.call('get', KEYS[1]) == ARGV[1] then return redis.call('del', KEYS[1]) else return 0 end";

    //构造方法,传入jedis对象,锁的key和过期时间
    public RedisLock(Jedis jedis, String lockKey, long expireTime) {
        this.jedis = jedis;
        this.lockKey = lockKey;
        this.expireTime = expireTime;
    }

    //尝试获取锁的方法,返回一个布尔值,表示是否成功获取锁
    public boolean tryLock() {
        //生成一个唯一的随机值,作为锁的值
        lockValue = UUID.randomUUID().toString();
        //使用lua脚本,原子性地设置key和过期时间,如果成功返回1,否则返回0
        long result = (long) jedis.eval(setScript, 1, lockKey, lockValue, String.valueOf(expireTime));
        //如果返回1,表示获取锁成功
        if (result == 1) {
            //创建一个续租线程,每隔一半的过期时间,就延长锁的过期时间
            renewThread = new Thread(() -> {
                while (true) {
                    try {
                        //休眠一半的过期时间
                        Thread.sleep(expireTime / 2);
                        //延长锁的过期时间
                        jedis.pexpire(lockKey, expireTime);
                    } catch (InterruptedException e) {
                        //如果线程被中断,退出循环
                        break;
                    }
                }
            });
            //启动续租线程
            renewThread.start();
            //返回true
            return true;
        }
        //如果返回0,表示获取锁失败
        else {
            //返回false
            return false;
        }
    }

    //释放锁的方法
    public void unlock() {
        //使用lua脚本,原子性地删除key,只有当key的值和锁的值相等时,才会删除
        jedis.eval(delScript, 1, lockKey, lockValue);
        //中断续租线程
        renewThread.interrupt();
    }
}

进阶代码优缺点:

分析一下这个代码的优缺点。这个代码的优点是比之前的代码更健壮,解决了以下几个问题:

  • 使用lua脚本可以保证设置key和过期时间的原子性,避免了客户端崩溃或者网络中断导致的死锁。
  • 使用唯一的随机值可以标识锁的持有者,避免了误删其他客户端设置的key的情况。
  • 使用续租机制可以延长锁的过期时间,避免了锁被其他客户端抢占的情况。
  • 使用lua脚本可以保证删除key的原子性,避免了客户端崩溃或者网络中断导致的锁未释放的情况。

这个代码的缺点是还是有一些问题,如:

  • 如果续租线程出现异常或者延迟,那么锁可能会过期,导致锁的安全性被破坏。
  • 如果锁的过期时间太长,那么可能会导致客户端在获取锁失败的情况下,等待的时间过长,导致性能下降。
  • 如果锁的过期时间太短,那么可能会导致续租线程频繁地延长锁的过期时间,导致网络开销增加。
  • 如果redis服务器出现故障或者主从切换,那么锁的状态可能会丢失,导致锁的一致性被破坏。

为了解决这些问题,可以使用一些更复杂的逻辑,如使用watchdog机制来监控续租线程的状态,使用自旋锁或者阻塞锁来优化锁的等待策略,使用集群或者哨兵模式来提高redis的可用性等。或者可以使用redisson框架,它已经实现了这些逻辑,而且提供了更多的分布式锁的功能和选项。
但是进阶版代码已经能cover大部分的场景,没有技术能实现万无一失,只是在出现问题的时候进行有效的补救,代价在承受范围内就行。也没有什么技术是永恒最好的,抛开业务谈方案就像空中楼阁。
最后,再一次感谢大家的阅读!

02-21 01:50