Redis 如何实现分布式锁？解决了什么问题？

齐朋
158 阅读5分钟

Redis 实现分布式锁的几种方式及演变

在分布式系统中，锁是保证并发安全的关键机制。Redis 作为一个高性能的内存数据库，在分布式锁的实现中扮演了重要角色。从最初的简单实现，到如今基于框架的高级解决方案，Redis 的分布式锁技术经历了多次改进。本文将详细介绍不同的 Redis 分布式锁实现方式及其演变，并给出相关的 Java 代码示例，帮助你更好地理解和使用 Redis 实现分布式锁。

1. SET NX（原始实现）

原理：

最初的 Redis 分布式锁实现方式是通过 SETNX 命令（SET if Not Exists）来创建锁。SETNX 的意思是：如果给定的键不存在，则设置该键值并返回 1；如果该键已存在，返回 0。在分布式锁场景中，通常会将一个标志位作为键，将线程 ID 或 UUID 作为值。如果这个键值已经存在，表示其他线程已经持有锁，当前线程需要等待。

Jedis jedis = new Jedis("localhost");
String lockKey = "myLock";
String lockValue = UUID.randomUUID().toString();
boolean lockAcquired = jedis.setnx(lockKey, lockValue) == 1;
if (lockAcquired) {
    // 成功获取锁，执行任务
} else {
    // 锁被占用，等待或重试
}

缺点：

  • 死锁问题：由于 SETNX 命令没有设置过期时间，如果持锁的客户端崩溃，锁不会自动释放，可能导致死锁。
  • 误删问题：如果锁的持有者在执行任务期间崩溃或失联，其他线程可能错误地删除锁。
  • 不可重入问题：如果一个线程已获得锁，它不能再次获得同一个锁，导致线程间竞争。
  • 锁不能续期：锁过期后，不能自动延续，可能会导致锁失效。

2. SETNX + 过期时间（Redis 2.6后改进）

改进：

随着 Redis 2.6 的推出，SET 命令新增了设置过期时间的参数，可以通过 EX（秒）或 PX（毫秒）选项设置锁的有效期。这样即使持锁的客户端崩溃，锁也会在过期后自动释放，从而避免死锁问题。

Jedis jedis = new Jedis("localhost");
String lockKey = "myLock";
String lockValue = UUID.randomUUID().toString();
boolean lockAcquired = jedis.set(lockKey, lockValue, "NX", "EX", 30) != null;
if (lockAcquired) {
    // 获取锁成功，执行任务
} else {
    // 锁被占用，等待或重试
}

缺点：

  • 误删问题：如果设置的过期时间不合理，可能会导致锁提前过期，进而导致任务的竞争。
  • 不可重入问题：仍然无法支持同一线程多次加锁。
  • 锁不能续期问题：如果任务执行时间较长，锁的过期时间内任务未完成，锁就会失效，导致其他线程获取锁。

3. 使用 Lua 脚本

解决方案：

为了避免误删问题，可以使用 Lua 脚本来保证操作的原子性。通过 Lua 脚本，在同一个事务中可以同时检查锁的持有者是否是当前线程，如果是，则释放锁；否则，不执行释放操作，从而避免误删。此外，Lua 脚本也可以解决不可重入的问题。

Jedis jedis = new Jedis("localhost");
String lockKey = "myLock";
String lockValue = UUID.randomUUID().toString();
String luaScript = 
    "if redis.call('get', KEYS[1]) == ARGV[1] then " +
    "    return redis.call('del', KEYS[1]) " +
    "else " +
    "    return 0 " +
    "end";
Object result = jedis.eval(luaScript, 1, lockKey, lockValue);
if (result.equals(1L)) {
    // 成功释放锁
} else {
    // 锁未被当前线程持有
}

优点：

  • 原子性：Lua 脚本的原子性可以确保获取和释放锁的过程不被其他线程打断。
  • 可重入：通过加上唯一标识符，可以确保同一线程能多次获得同一个锁。

新问题：

  • 实现复杂：编写 Lua 脚本需要小心，错误的脚本可能会导致逻辑问题。
  • 无法自动续期：Lua 脚本本身没有内建机制来自动延长锁的有效期。

4. Redis 框架（基于 Redis 实现的分布式锁框架）

解决方案：

为了更好地处理 Redis 分布式锁的各种问题，可以使用基于 Redis 的专用框架，如 Redisson。这些框架封装了分布式锁的实现，解决了死锁、误删、可重入和锁续期等问题。

Redisson 是一个开源的 Java 框架，它为分布式锁提供了更为完整的解决方案，并实现了自动续期的机制。它通过使用看门狗机制（watchdog）来定期延长锁的有效期，从而防止锁过期。

Config config = new Config();
config.useSingleServer().setAddress("redis://localhost:6379");
RedissonClient redisson = Redisson.create(config);
RLock lock = redisson.getLock("myLock");
lock.lock();
try {
    // 执行任务
} finally {
    lock.unlock();
}

优点：

  • 自动续期：通过看门狗机制，Redisson 可以自动延长锁的有效期，防止锁过期。
  • 可重入：同一线程可以多次获取锁，不会发生死锁。
  • 简化实现：Redisson 封装了很多细节，开发者无需关心底层的实现，能够专注于业务逻辑。

总结：

Redis 本身提供了基本的分布式锁功能，但原生的 SETNX 和 SET 命令存在多种问题（如死锁、误删、不可重入、不能续期）。通过使用 Lua 脚本，可以较好地解决误删和可重入问题，但它无法自动续期且实现复杂。基于 Redis 的框架，如 Redisson，则提供了更加完善的分布式锁实现，能够解决所有的上述问题，并且提供了自动续期等高级特性，适合生产环境中的使用。

在实际项目中，如果你的系统对高可用性和高可靠性要求较高，建议使用像 Redisson 这样的框架，它不仅能解决锁管理中的所有问题，还能极大地简化代码的复杂度，减少出错的可能性。

avatar
文章
阅读
粉丝