经验首页 前端设计 程序设计 Java相关 移动开发 数据库/运维 软件/图像 大数据/云计算 其他经验
当前位置:技术经验 » 数据库/运维 » Redis » 查看文章
Redis从入门到放弃(10):分布式锁
来源:cnblogs  作者:伊力编程  时间:2023/8/9 9:15:22  对本文有异议

在分布式系统中,实现对共享资源的安全访问是一个关键问题。Redis作为一种高性能的内存数据库,提供了多种方式来实现分布式锁,以解决多个节点之间对共享资源的并发访问问题。

本文将介绍五种Redis分布式锁的解决方案及其原理、应用场景以及Java代码的实现步骤。

1、SETNX

SETNX(SET if Not eXists)命令是Redis提供的一种原子操作,用于设置一个键值对,当键不存在时才设置成功。利用该命令可以实现基本的分布式锁。

  1. SETNX lock_name true

1.1、原理

  • 当节点需要获取锁时,它尝试通过 SETNX 命令设置一个特定的键作为锁。
  • 如果设置成功,表示获取到了锁,可以执行访问共享资源的操作。
  • 用完共享资源使用 DEL 命令释放锁。

1.2、代码实现

  1. Jedis jedis = new Jedis("localhost", 6379);
  2. String lockKey = "my_distributed_lock";
  3. String identifier = UUID.randomUUID().toString();
  4. boolean lockAcquired = jedis.setnx(lockKey, identifier) == 1;
  5. if (lockAcquired) {
  6. // 成功获取锁
  7. // 执行访问共享资源的操作
  8. // ...
  9. jedis.del(lockKey); // 释放锁
  10. }

1.3、缺点

死锁:可能因网络等原因 DEL 命令执行失败而造成锁无法释放。

2、SET(NX EX)

避免死锁解决方案是为锁设置一个TTL(Time To Live)。

SET 命令支持选项:

NX:表示只在键不存在时才设置。

EX:表示设置键的过期时间(秒)。

  1. SET lock_name arbitrary_lock_value NX EX 10

在上面的命令中,NX 与 SETNX 中的含义相同,而 EX 10 表示 TTL 为 10 秒。

2.1、原理

  • 当节点需要获取锁时,它通过SET命令的(NX EX)选项尝试设置一个特定的键作为锁,并设置锁的过期时间。

  • 如果设置成功,表示获取到了锁,可以执行访问共享资源的操作。

2.2、代码实现

  1. import redis.clients.jedis.Jedis;
  2. public class DistributedLockExample {
  3. public static void main(String[] args) {
  4. Jedis jedis = new Jedis("localhost", 6379);
  5. String lockKey = "my_distributed_lock";
  6. String identifier = "unique_id"; // Replace with a unique identifier for your node
  7. int lockTimeout = 5000; // Lock expiration time in milliseconds
  8. // Try to acquire the lock
  9. String result = jedis.set(lockKey, identifier, "NX", "EX", lockTimeout / 1000);
  10. if (result != null && result.equals("OK")) {
  11. try {
  12. // Successfully acquired the lock
  13. System.out.println("Lock acquired, performing the task.");
  14. } finally {
  15. // Release the lock
  16. jedis.del(lockKey);
  17. }
  18. } else {
  19. // Failed to acquire the lock
  20. System.out.println("Failed to acquire the lock.");
  21. }
  22. jedis.close();
  23. }
  24. }

2.3、缺点

线程A获得锁,在执行业务还未完成时TTL过期,线程A锁被释放,线程B获得锁。线程A后续为无锁执行业务,在线程A完成业务后,执行DE删除锁,因为KEY相同则会导致线程B的锁被删除,线程B后续操作则为无锁执行业务

如下图所示:

3、SET(NX EX) + 唯一ID

在设置key时,客户端应将唯一的 ID 添加到 kv 对。在删除key之前,检查这个 ID 以确定它是否仍然持有锁。如果 ID 不匹配,则表示该锁被其他客户端持有,当前客户端不应删除该key。

  1. SET lock_name client_id NX EX 10

3.1、原理

  • 当节点需要获取锁时,它通过 SET 命令的 (NX EX) 选项设置一个特定的键作为锁,并设置锁的过期时间。
  • 在设置锁的同时,还设置一个唯一的标识(如uuid)为锁的值。
  • 当节点释放锁时,只有当锁的值与自己的标识相匹配时才释放锁。

3.2、代码实现

  1. import redis.clients.jedis.Jedis;
  2. public class DistributedLockExample {
  3. public static void main(String[] args) {
  4. Jedis jedis = new Jedis("localhost", 6379);
  5. String lockKey = "my_distributed_lock";
  6. String identifier = UUID.randomUUID().toString();
  7. int lockTimeout = 30000; // Lock expiration time in milliseconds
  8. // Try to acquire the lock
  9. String result = jedis.set(lockKey, identifier, "NX", "EX", lockTimeout / 1000);
  10. if (result != null && result.equals("OK")) {
  11. try {
  12. // Successfully acquired the lock
  13. System.out.println("Lock acquired, performing the task.");
  14. } finally {
  15. // Release the lock (if acquired by the same identifier)
  16. String currentValue = jedis.get(lockKey);
  17. if (currentValue != null && currentValue.equals(identifier)) {
  18. jedis.del(lockKey);
  19. }
  20. }
  21. } else {
  22. // Failed to acquire the lock
  23. System.out.println("Failed to acquire the lock.");
  24. }
  25. jedis.close();
  26. }
  27. }

3.3、缺点

  1. if (currentValue != null && currentValue.equals(identifier)) {
  2. jedis.del(lockKey);
  3. }

IF 条件和 DEL 是两个独立的操作。极端情况下,IF 条件判定可以释放锁,在执行删除锁操作前刚好TTL过期,其他线程获取锁执行,前者线程删除锁删除的依然是别的线程的锁。

4、SET(NX EX) + 唯一ID + Lua脚本

为了确保释放锁的原子性,可以使用Lua脚本来执行判断+删除操作(lua脚本的执行的原子的)。

lua脚本:

  1. // 如果来自 Redis GET 操作的值等于传入的值,则删除键
  2. if redis.call("get", "lock_name") == ARGV[1]
  3. then
  4. return redis.call("del", "lock_name")
  5. else
  6. return 0
  7. end

4.1、代码实现

  1. String lockKey = "my_distributed_lock";
  2. String identifier = UUID.randomUUID().toString();
  3. String script = "if redis.call('get', KEYS[1]) == ARGV[1] then return redis.call('del', KEYS[1]) else return 0 end";
  4. Long result = (Long) jedis.eval(script, Collections.singletonList(lockKey), Collections.singletonList(identifier));
  5. if (result == 1) {
  6. // 成功释放锁
  7. } else {
  8. // 锁的标识不匹配,可能已被其他节点获取
  9. }

5、Redisson

以上的方案还存在一个问题:不优雅。

若执行业务完成时间不固定,可能小于过期时间也可能大于过期时间,在代码中只能设置可能完成的预估最长时间。

5.1、开源解决方案: Redisson

一旦客户端持有锁,它就会启动一个WatchDog守护线程来定期检查锁是否存在。如果存在,守护线程将重置 TTL 以防止锁自动释放。这种策略被称为租赁策略。防止先于业务完成就释放锁。

5.2、代码实现

  1. Config config = new Config();
  2. config.useSingleServer().setAddress("redis://127.0.0.1:6379");
  3. RedissonClient redisson = Redisson.create(config);
  4. RLock lock = redisson.getLock("my_distributed_lock");
  5. try {
  6. boolean lockAcquired = lock.tryLock(acquireTimeout, lockTimeout, TimeUnit.MILLISECONDS);
  7. if (lockAcquired) {
  8. // 成功获取锁
  9. // 执行访问共享资源的操作
  10. // ...
  11. } else {
  12. // 获取锁失败,处理重试逻辑
  13. }
  14. } finally {
  15. lock.unlock(); // 释放锁
  16. }

6、总结

Redis分布式锁提供了多种解决方案,可以根据具体的应用场景和需求选择合适的方案。无论选择哪种方案,都需要考虑锁的安全性、原子性、续约和释放等问题,以确保在分布式环境中共享资源的安全访问。

原文链接:https://www.cnblogs.com/myshare/p/17614618.html

 友情链接:直通硅谷  点职佳  北美留学生论坛

本站QQ群:前端 618073944 | Java 606181507 | Python 626812652 | C/C++ 612253063 | 微信 634508462 | 苹果 692586424 | C#/.net 182808419 | PHP 305140648 | 运维 608723728

W3xue 的所有内容仅供测试,对任何法律问题及风险不承担任何责任。通过使用本站内容随之而来的风险与本站无关。
关于我们  |  意见建议  |  捐助我们  |  报错有奖  |  广告合作、友情链接(目前9元/月)请联系QQ:27243702 沸活量
皖ICP备17017327号-2 皖公网安备34020702000426号